UPnP

天云上人

/*uPnP.h*/

1. 
   # {. d6 o( _/ |9 k7 A
2. #ifndef   MYUPNP_H_
   ( {$ {& V7 ^1 X2 _8 y& v8 _4 `
3. 
   
4. #pragma   once
   + Y6 J2 n5 }% D. y+ |+ X
5. 
   
6. typedef   unsigned   long   ulong;
   & {* B9 `$ q, O! `
7. 
   6 ~# n5 A  J5 q; Z1 e
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    6 @6 v: x5 {& ]2 ?  a& F
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ! i6 ~. E/ `  n. _" P: T
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;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    ! N$ K; ~9 k  d% L6 f1 L2 g
22. }   UPNPNAT_PROTOCOL;
    % W9 Y  C; K; b0 f& v
23. 
    
24. typedef   struct{
    . [; _, w" Q' F0 f
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)
    ( M# [/ g5 c: L3 Y$ M
28. CString   description; //   Port   mapping   description
    , w5 a! u0 @" l+ o1 G
29. }   UPNPNAT_MAPPING;
    
30. 
    ( T4 M) @0 R9 |
31. MyUPnP();
    
32. ~MyUPnP();
    1 K2 g+ F6 t( O2 @# @
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    2 U7 q' \6 s0 e$ {  H$ J* p' j/ }
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    * U7 m( w# E. n+ I
40. WORD GetLocalIP();
    : \- C8 B; ~" X8 f0 z9 c0 @
41. bool IsLANIP(WORD   nIP);
    
42. 
    . m9 f) I8 M# F4 ?
43. protected:
    
44. void InitLocalIP();
    6 e1 x+ @' Y/ f' c# k9 ^4 w; J
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
      p9 X) c; q! _* h
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ; O+ ]3 p. `0 O  z9 h
52. 
    
53. bool Search(int   version=1);
    9 U+ R, F* q8 i) }2 p) r
54. bool GetDescription();
    $ f# O) G4 o/ |" U4 y4 L( ^
55. CString GetProperty(const   CString&   name,   CString&   response);
    6 N+ i0 P( j: T+ n) n# i; F( q
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    1 J! ]0 }- E$ A
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    1 I4 j% w/ i8 \2 ?/ S& p
60. CString m_devicename;
    ( y4 k# N9 B) s  ^
61. CString m_name;
    , ^% n( C6 A9 _
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
      U! c. d: a1 ]% K$ g
65. CString m_friendlyname;
    
66. CString m_modelname;
    8 p3 Q, v- ^/ W' w, Z
67. int m_version;
    4 ^: d' f  ^% B! ~9 v1 ?7 G
68. 
    7 Z* J& q( }! s% A# x8 s$ W# O
69. private:
    % X' _9 [, U' ~. g$ n
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    9 v+ U+ P" i  ~$ O* X0 G
71. 
    ) d7 \% B5 _* C: L7 |! j% R
72. CString m_slocalIP;
    
73. CString m_slastError;
    2 Q" P7 W% r# I+ s3 Q. T8 H$ l
74. WORD m_uLocalIP;
    
75. 
    , i2 @3 N2 s8 ^8 B
76. bool isSearched;
    
77. };
    - W) u* H- G* B
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ' t  o8 E3 T3 F9 O8 I/ V* c) [( J8 \, G
2. #include   "stdafx.h "
   
3. 
   - `3 Z( X0 b. A& Y$ j' o
4. #include   "upnp.h "
   
5. 
   1 C9 H" i( d+ e) k7 b  h
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   1 v  U2 V; u, K3 N
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + ~" t) K8 t# N1 u" @9 @. p! F+ F
15. 
    
16. const   CString   getString(int   i)
    " Q4 P4 |0 y1 r" q, P# B
17. {
    
18. CString   s;
    % d( X, P8 L3 u% g0 `
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    ! l) j/ G9 F! X3 S3 s# M9 W' }
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    + b2 {1 W; v) i& L- k: N) J* Z
26. {
    : z; f4 o( y$ [1 M8 a
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    9 T. ?- U; N0 U1 f- d
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    1 f* `4 t( a  c; D  ~- g
33. }
    
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);
    5 E- G+ P  _; {: a
40. int   length   =   sa.GetLength();
    3 V* J' i* k( \6 c* S/ d
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ) H$ O0 ^/ f" X8 Z. x3 e/ u! C
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
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;
    & y6 s. J5 V1 u3 V* S1 L1 p8 }5 l
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    * X  R! v/ W( T3 A1 J
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    & j* J, `/ ^! |0 S
54. int   n   =   send(s,   buffer,   length,   0);
    % h% f; |" A9 H+ t! m% a) {: p
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ) F6 y1 y1 p) V/ {. Y' k
57. closesocket(s);
    2 P( g. O% N  K! }  Y/ t% P4 a+ Y
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ' `- }" S$ k! F/ y9 v
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    ! d5 ]% {5 R& c3 B3 K' D
70. const   CStringA   sa(request);
    : A4 `. d3 g, X/ z
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    # \( a6 K0 G# ~. B6 y! x( v% [
73. 
    " Q! |# R# t$ R5 n4 j% h
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    4 @! L, L% W  ^' f% H
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
      X& x3 @6 X( X
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 A* f7 g# q3 |( s8 w2 Q
84. {
      W, k/ L. A6 `* E
85. int   pos   =   0;
    ) c& ^& |: d! F" W+ _# q6 T
86. 
    " h2 D4 z8 G" R( M" e+ F- ~
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    , K& Y: |/ R- \4 z9 N! N1 Q$ ?
88. 
    7 `* V; D9 \9 m" w$ F
89. result   =   response;
    
90. result.Delete(0,   pos);
    5 r- o. ]$ m! G
91. 
    " J6 b/ V+ @) T
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    4 K/ w# F: d) G$ }$ O% A
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     8 m/ O. I9 l$ [- z# w( R& _4 s
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # \1 }% n+ R0 ]9 a4 F- c# l( u  f
103. CString   property;
     : q! K+ N( x% z* t8 s
104. 
     7 B. m2 V6 E4 }( X9 U" [
105. int   posStart   =   all.Find(startTag);
     6 I  _0 D1 _, V
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     1 s1 }0 ]1 e  m9 i
109. if   (posStart> =posEnd)   return   CString();
     3 p; C+ F! q/ ^. \2 n
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     & I* L% N6 Q( i4 o2 @9 P
116. {
     ' P$ B- V- C5 H
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     0 e0 h+ d4 s3 A* S0 k
121. MyUPnP::~MyUPnP()
     ) E- p$ R& I" u6 s' g- q4 K
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     - D5 N# e  i2 y  a
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ' r1 h. ^+ x( K+ r$ H; e
127. RemoveNATPortMapping(search,   false);
     ' T% Z" c; A: `" M
128. }
     9 o  F3 d- L" a. j
129. 
     " u  O6 B6 t( ?* X- ~# R
130. m_Mappings.RemoveAll();
     2 W% ]/ C* B( R3 U+ p
131. }
     $ q1 `. F7 ?9 w! n5 @% H* b9 _
132. 
     6 p+ D! W7 C* y1 E9 n
133. 
     ( w4 @" H# d  G
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     " i9 |# x2 y' K" x9 F! v
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     8 n% N% |8 e2 ?$ T% F  K" a6 l
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ) x8 H9 T: t. u8 |4 ~) U, i
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     . ]7 z! k: l  P9 q) s9 b
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++)   {
     : Z3 T+ Y6 f" `. I
152. if   (!(i%100))   {
     / x) N9 p- s3 t# A2 Q! ?
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
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);
     7 f( R( [( D/ ?3 R4 Y7 ]
159. }
     
160. }
     " Y; u8 o" |8 F7 I+ y- s  U
161. 
     
162. Sleep(10);
     ( p& A7 _* L7 X9 h7 k
163. 
     
164. char   buffer[10240];
     . L- @8 O* d0 Q- @2 ^; `& m
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     % ~+ O1 Q3 a8 m0 P, x* C
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     3 z1 j  |$ y+ b) M5 R
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);
     8 K( b. L% N* I
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ( k9 a4 d+ ~& c
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     * Y# b/ @6 ]. T5 t
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     ) W/ r9 g2 M) s( ^% E
183. m_description   =   line;
     
184. m_description.Trim();
     1 r% r  ?$ S: F  c, J, P6 M3 S3 o+ }
185. return   GetDescription();
     
186. }
     , Y9 ]9 y9 J( d# G
187. }
     ) v  m( a0 [- ~# N
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     ! P. N$ C% e* j8 W- k+ b0 `1 N
192. 
     ) z1 ?1 _8 ~/ y4 M, x0 d- k( b/ H
193. return   false;
     ( S4 p/ H/ y% Y. v5 r7 Y) ~8 @0 A
194. }
     & z  l2 r( N( S: I$ \

复制代码

天云上人

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


; b  D! H7 ^! ~1 s/ ~, G///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
8 ]3 e4 k& j$ l" }$ j- V1 j

#pragma once
& F$ b( A/ ~7 a, Y; I; s$ p#include <exception>
  I# x7 c+ r8 H. ]* u' x. ?' v; M

  enum TRISTATE{
        TRIS_FALSE,
% s4 \4 ?5 G7 |" B3 H        TRIS_UNKNOWN,
        TRIS_TRUE
};

2 O, _3 }% F5 o& {9 E/ m3 W6 \* ~
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
; t, {$ R$ ^8 {% |  @# R1 u% P! e        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
! x" l4 o- V' J3 z. M};

; A: i1 n+ R, @


class CUPnPImpl
. O1 Z) ^  t" G; e0 r0 G{
" l2 H3 n5 b* G$ P0 d- k/ bpublic:
        CUPnPImpl();
. c, I# ~( j, _9 w; S& ^5 a) _. ]        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
* X/ \2 I* b7 I5 G5 [3 X* a

' f" A/ i: W2 F, f7 q, H        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
" i1 U& Y: n0 U  }; H" }  _2 A' [        virtual bool        IsReady() = 0;
2 d& ^5 g/ g  l1 m( c, h        virtual int                GetImplementationID() = 0;
7 ]9 b7 @5 e$ o) W       
& o' j& \$ X" s4 p" Y        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


0 U5 j$ z  f) a: h        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
  p; w9 G5 ^. f! x' |0 M) [1 Q        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
; y6 o: I  L) L4 A0 |        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
8 M% {- D) t. S2 A: ?% J! U        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
6 k; E5 D. ]: O7 j  `

# g. C2 u+ V6 M) V9 K# |3 i% z8 _// Implementation
8 n" V/ Y9 S1 P8 v! ~- l# c. G: e. Xprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
6 }: I- V. Y; |        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
8 `3 L4 |: l2 T- ~7 T5 f7 o        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

6 J2 Y& u& N5 J) {0 A. N
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
: \0 }' Y5 Q: m. n: Z; A
7 d# _' y$ k! r8 O/ J' V( ?
. V. N# n5 ]& O* N8 p, R7 m};

4 @2 ]# V! W3 H- s
  P+ n. E- x7 v5 L; d// Dummy Implementation to be used when no other implementation is available
( p/ A2 m* Z( H# J3 b; Wclass CUPnPImplNone: public CUPnPImpl
8 N% j2 g; o* _{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
$ h% ~5 k. k& V        virtual void        StopAsyncFind()                                                                                        { }
) V: [3 R4 E. n8 m3 q0 F( k# D        virtual void        DeletePorts()                                                                                        { }
) X- l" t  i1 ~2 M4 }        virtual bool        IsReady()                                                                                                { return false; }
# c  o6 R/ @+ D- w. D        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
4 D. U* H: p& k+ U5 k};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
( x/ W6 p) `; M9 U! J2 m' V

#pragma once
" o5 a5 V/ l6 W3 P0 f* ], A#pragma warning( disable: 4355 )
* J- D! {  Y% C  B9 A" c- ^

+ }3 k  r  r) S7 @+ w+ c#include "UPnPImpl.h"
/ [/ Q! |' ~# {#include <upnp.h>
) J7 @" ?- e( m. C& a5 [#include <iphlpapi.h>
0 @  W5 D+ m! ^+ Z' I# w  }#include <comdef.h>
#include <winsvc.h>

; B- d& z% K  Z( o/ H* N" g
6 G5 K  \5 f/ S#include <vector>
#include <exception>
/ v. o  O  e+ |% Q3 y#include <functional>
. }! T3 K" D  e/ \



typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
9 l; ^  K4 Z  Y) y8 J' X# M3 ntypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
' S: M& E0 M! I* ]typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

4 v( p( W! \1 M3 c
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
9 |7 k4 H5 l, E' L! h" s  t' I% E);

0 |7 w$ Y% y; X  }2 l
+ q( r! e- w3 q8 p! ?typedef DWORD (WINAPI* TGetIpAddrTable) (
8 E8 G8 }/ Q4 O4 n$ F! K, i  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
: z3 ~3 _2 Z% W# o  A  BOOL bOrder
);

9 U' p% D+ C5 B9 c/ z! u' _# q
% V$ z. R+ M3 z3 W0 ktypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
8 q* |% `+ Y4 J+ ^5 r' Y6 s);


/ a" @5 V, y* n2 A9 O" `5 T% yCString translateUPnPResult(HRESULT hr);
+ G0 F9 L. v# N' I8 e& ]HRESULT UPnPMessage(HRESULT hr);


* F1 S& {3 ^4 z. V/ p: s* Qclass CUPnPImplWinServ: public CUPnPImpl
) S; q; @* O# L{
3 K0 Q3 C, p3 z. u        friend class CDeviceFinderCallback;
& ?, N- }) h7 n( X, r/ ]        friend class CServiceCallback;
( U/ o# a# t( g0 {5 k6 Q4 r" y* O5 D, Z// Construction
public:
        virtual ~CUPnPImplWinServ();
! T5 j8 M$ [. H1 [1 n7 U, |        CUPnPImplWinServ();


* m& p3 N- T0 x# i        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
! x! q8 y# l% R2 g8 C% a3 d$ M8 |        virtual void        StopAsyncFind();
3 S5 k# d& a6 _% Q        virtual void        DeletePorts();
: `% E, T3 {* d* j, Q: t, {1 d' m        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
1 y! b" M8 V5 M. g* I
. B  ]3 s2 T% W1 i7 L9 V0 Y- v
9 \, \7 h  g7 }. b4 Z/ K2 F        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
2 h8 s6 k. }$ `+ s3 R1 d- N. V7 q        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
: i+ g0 ^/ `2 v& s& k" Q$ B        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
' f- X! w  P+ J8 N/ b0 L        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
  e* u7 H+ q8 W        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
" A9 o' W5 A; p, F! Z' W        void        RemoveDevice(CComBSTR bsUDN);
3 s" h7 x9 U/ U        bool        OnSearchComplete();
  f5 p3 t$ X; F$ f        void        Init();
2 q1 @; S. k  F/ \5 v

6 F/ N, n  ^3 W; t0 m0 V4 Z        inline bool IsAsyncFindRunning()
, I! p0 c$ }( X$ q3 P        {
/ d/ o3 T/ u% l* ^, e, r& q! g, S                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
7 B. V/ U: s+ z- i  B4 H                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
) J' p& }. l& w" x, v$ l; t' f/ F3 U                }
5 F, p8 j$ A# f% m/ A' v/ ]/ z5 d* Q                MSG msg;
) N$ c$ I0 |) |7 p! h: y/ K                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
: V$ y: f) N- J! `* Z: k$ O                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
/ V1 Z- c: g. L  r2 E5 c0 S+ N: s7 }                }
" A# m& V$ |% P+ Q- i! b& ]                return m_bAsyncFindRunning;
0 a0 R3 ~# U5 c: e/ m: {        }
5 s( H- v5 z  o8 b( i* o$ ~
5 ]& p; R4 x8 C% K( n+ u7 R
        TRISTATE                        m_bUPnPDeviceConnected;

0 x# f" i% P" i- H
// Implementation
; j- x2 x( Z3 q# g+ |. U) B        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
. B+ }* t! C6 a7 K        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
* I5 b; f# k! `& W; X        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

7 }: }9 \+ }+ ?
        TGetBestInterface                m_pfGetBestInterface;
: q, ?' A  {4 [7 d* d- N        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


* V4 k0 p) m* m' M  D0 p. n        static FinderPointer CreateFinderInstance();
) E3 o& Y8 Y# Z# \% w& R        struct FindDevice : std::unary_function< DevicePointer, bool >
. o% Z& k0 Y% T3 `        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
# a  k8 L) n8 F# G! G3 T                result_type operator()(argument_type device) const
0 [8 b0 f9 D% T: g5 p6 A5 k2 u% a                {
                        CComBSTR deviceName;
) |6 i1 p1 n- f                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
# P3 Y2 s* n3 z! h! I

% h  x- E/ Y# m5 ~1 q                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
+ c) X: U, i0 }/ y# [# v
& [# e0 c( O( Y% e, ]
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
1 B  C0 t; }9 Y                }
                CComBSTR m_udn;
        };
: a2 `4 G! V( Q       
' G. k( B4 o% y1 U! q        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
! t- t! b$ [  ^9 M1 X3 |% O        void        StartPortMapping();
# {9 A1 M# {0 g6 C+ |+ G        HRESULT        MapPort(const ServicePointer& service);
: M4 m! Z) |5 X& G% k        void        DeleteExistingPortMappings(ServicePointer pService);
  y- H# _$ d, E4 ?3 l        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
. a* Q5 q6 L+ ^) B- Z        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
1 D5 v: h4 q5 c3 E7 {        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
9 {: |* j6 U8 t. w9 \        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
+ G" {7 P& W  O' o) q' A        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
9 V- }0 J7 G, ]6 W  M        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
" K. r8 h* b* B2 {( U" m% s3 T        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
1 }7 b; @2 f3 {8 F+ v

- R( _5 W0 ?+ Q  I. ^5 G3 F// Private members
private:
# m5 r! G9 g9 ]+ F6 c9 j% R        DWORD        m_tLastEvent;        // When the last event was received?
* c! [. l; s9 i" u/ A        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
4 }6 D* t/ f/ J        FinderPointer                        m_pDeviceFinder;
0 S6 O+ P/ R0 M) [8 X$ X3 X        DeviceFinderCallback        m_pDeviceFinderCallback;
1 T: b0 H5 S4 _        ServiceCallback                        m_pServiceCallback;

* O" V' K: o7 N
0 D, G' v4 v1 o( y$ a1 i3 b        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
4 q& Y: `: F4 @3 ~" o0 o+ A        bool        m_bPortIsFree;
0 x; N* d# a# _8 a4 U4 q2 r; E        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
( h& @: r! A+ |        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
, S' T9 r& d+ w# H        HMODULE        m_hIPHLPAPI_DLL;
9 \2 Q3 S9 a& l2 r+ c+ \) }        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
7 {- \9 J0 k: C+ K+ Q$ g        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
( p* b, N6 a1 B3 B2 e% Z" y, X

// DeviceFinder Callback
class CDeviceFinderCallback
- q3 W* g$ k2 b* w) D6 H7 H1 {        : public IUPnPDeviceFinderCallback
/ k  q0 w5 R$ k7 j{
7 _( a' G+ ]" B7 A6 l9 o% Cpublic:
9 z" e8 t* E6 E( P8 S        CDeviceFinderCallback(CUPnPImplWinServ& instance)
" F* g$ w( F  w# u0 v                : m_instance( instance )
" J7 i  r6 O8 `1 p" R        { m_lRefCount = 0; }
& q" k1 n' N& |2 }$ y
) J% @+ u# t% K# t6 z
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  b! l1 o- s3 P7 W6 \   STDMETHODIMP_(ULONG) Release();


// implementation
$ M! |% D" c  ?$ G- N7 l7 j3 m- d5 jprivate:
" ^& q$ m5 S" f- ]6 s' K/ g: A        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
/ J1 k& I6 P8 A" O% C        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
, E" H. p5 y- }2 m$ K8 x

) w1 W/ [3 k8 jprivate:
        CUPnPImplWinServ& m_instance;
& D8 @9 A& w, X1 U        LONG m_lRefCount;
};
& S$ H( o1 |# \1 r: {8 _

// Service Callback
. i8 q7 E0 c8 k& F& y) d$ wclass CServiceCallback
        : public IUPnPServiceCallback
! H: r4 M5 C' T; @{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
' Q' z! J/ Y, V" d8 ?+ c   
5 n1 r% W/ y. h4 i   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- G7 o2 f+ ^) R: _& x$ a& o   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
% z- k6 O! o" F" [

* S# c0 p9 T  B1 j- l7 G// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
& K+ y* Y4 d4 ?6 `: q# X
2 ^2 O4 t! c1 a& o, p; H# _
+ `  m) {3 |" m  |private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
$ n3 M7 r6 t" O3 {: ~2 |& _};

2 W# V0 U* _. ^( r2 {8 Z/ b
- ?- Y; H' g4 k, C6 ^! H9 P/////////////////////////////////////////////////
0 h: S' C: j* l; \! M

; v* w; ?! K9 j. W( t使用时只需要使用抽象类的接口。
6 V+ b, h. y0 m; `0 x, g7 aCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
1 [8 z  s9 R+ `. M7 MCUPnPImpl::DeletePorts删除端口映射.