UPnP

天云上人

/*uPnP.h*/

1. 
   # k' E3 r/ a/ }1 Z, d
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   7 B! |: a; W3 g
7. 
   
8. class   MyUPnP
   " |" k; }( E$ |
9. {
   ( C; {4 v' `" `3 a
10. public:
    & }- t8 S- i" y$ t' d* g
11. typedef   enum{
    ) u1 m4 U# m! l  y. e' I4 @: B2 [
12. UNAT_OK, //   Successfull
    0 p9 D6 U- \+ t1 U
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
    4 q2 z5 N- Z$ `1 R) w- U
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. 
    ) Z: l- a) j) P0 C$ Q) h
19. typedef   enum{
    8 \. O# |1 e1 d/ ?. H3 r& L
20. UNAT_TCP, //   TCP   Protocol
    0 d* o/ |, B/ Q  V$ L" S1 i0 k
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    + j& m. e( U" Q) s
28. CString   description; //   Port   mapping   description
    ( x( V% I! O! R( d8 {; ^: r
29. }   UPNPNAT_MAPPING;
    5 U5 E' e, y  j: T3 B7 z2 [  u
30. 
    
31. MyUPnP();
    + I1 F; Y* ^/ ]; w, J
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 ~7 I+ ~5 [$ b! p
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    6 p0 T8 }" m: E7 F7 V
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ! \% V- w" ^5 m0 l% N* C
41. bool IsLANIP(WORD   nIP);
    
42. 
    4 T: I* K6 {3 ^9 h
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    $ u: v  n; R( x7 K
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      z- M7 _! O, ^% E2 R% K1 s# G+ S7 R/ J
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    4 t3 ~$ T- j5 V: D
55. CString GetProperty(const   CString&   name,   CString&   response);
    1 k8 y7 G- q& K/ ^3 ?" k! p
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    % t: w6 j; W0 F; K, P+ C4 ?
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 q. h6 {+ F0 [. j' |: Y
59. bool InternalSearch(int   version);
      z7 m- e- R: }, B6 J3 K3 A$ J
60. CString m_devicename;
    3 G) n+ e7 \5 K) k! O' ?7 u
61. CString m_name;
    $ f7 Q7 c, [0 w$ `' }
62. CString m_description;
      U' A1 {3 d: @2 [4 y# c) B
63. CString m_baseurl;
    
64. CString m_controlurl;
    % ~; Q. n! G$ Q5 W3 R, c! {; V2 @
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    ( s$ z2 Q- b0 \) M1 h0 R
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    & N7 o. G# _  I1 {" ^
71. 
    & r. x% f* c! A2 a, ~9 O9 K6 f9 [& d
72. CString m_slocalIP;
    1 B) N9 ~/ D* C, n2 T
73. CString m_slastError;
    ; S% M4 s  {' C! [. ?/ v( u
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    - g/ t; m# Z+ S4 D' y
77. };
    - X" |" e3 X4 ^" v1 O" y
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * f" Y7 w, u# G+ P* ~7 `; @) `1 `
2. #include   "stdafx.h "
   4 ]* x- X' Y/ f+ ]4 a5 k
3. 
   8 X; m. r& ?6 U; w& {) Q$ H; v
4. #include   "upnp.h "
   
5. 
   7 o! M, D+ Y5 F- R
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   # Q% X) d  }; K. z4 S- r
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   3 R& l  s. k' v2 i4 g
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    " |; u- k( ]9 m& A* W( q  _( g# N% i
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + {8 Y% A5 h& ?0 D: g6 l0 A' b
15. 
    3 R; A1 B8 z; c4 z. _/ L8 F0 A
16. const   CString   getString(int   i)
    0 l& b" P4 ~' S" S+ H  V. ~
17. {
    / M: n+ ^# f  Q& x
18. CString   s;
    
19. 
    9 i4 J- l7 ~' j3 B2 J8 E
20. s.Format(_T( "%d "),   i);
    / N/ @  z% Y5 W0 b% H
21. 
    
22. return   s;
    
23. }
    ) h$ w# L2 d7 n7 Y1 ?
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    , h9 \+ w. S, U+ W  |8 E
26. {
    " O" L0 m) g- Y
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    : X) g$ K1 C4 I. `' `# I+ t1 \
28. }
    ) p; b, z3 @3 X) x- `9 J. u
29. 
    % o7 n2 j' ]9 I  V
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    - ~: v5 j' M7 m. l; S
33. }
    ! w) ]# z& o% |7 Z
34. 
    * D8 h' X0 q9 g) k: p1 ~& B, ?
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    / U1 j' e; a: s% K/ T+ f
37. char   buffer[10240];
    $ P( l* G* t3 x  }! A4 t
38. 
    % {, U2 D$ F% M- A( s% X' {5 J
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    3 W" F& ?+ e" [9 N
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    5 X: W9 \- I) U) W5 \0 k6 @6 m
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    6 v) h5 \  D; t+ r3 ~
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      G6 n) v5 z. A* ]2 X' r
46. sockaddr.sin_family   =   AF_INET;
    + ^6 m% C2 `7 F0 q/ `8 Y3 ?+ ~/ m4 t
47. sockaddr.sin_port   =   htons(port);
    5 o1 G6 |4 g. W/ h8 t  d
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
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));
    . r+ ^9 Y$ }, h4 n
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    8 ?+ B7 e# u( V! U
57. closesocket(s);
    + x" q3 m$ Q4 e& B" z7 ?1 g
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ; o% _  K  P/ V/ m9 @2 Y
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ' z/ j( c$ R4 @6 S7 D# ?
64. }
    
65. 
    * P! O( K" e7 F4 x- r2 v. R
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + B* Q1 Y% y& [& i. ~* K5 S
67. {
    
68. char   buffer[10240];
    9 ?+ N6 m: B2 V2 z$ ?
69. 
    
70. const   CStringA   sa(request);
    8 a' y; q' Z+ y/ V2 A% f5 ^
71. int   length   =   sa.GetLength();
    9 j+ o- t% V9 [1 A8 H! K5 p
72. strcpy(buffer,   (const   char*)sa);
    9 c+ A2 [7 d* i
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ' F$ }" H9 y! S" T1 @8 c5 O1 u
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));
    ( \/ i" W: K7 K: d6 h
81. }
    
82. 
    - u  u4 B1 K: }
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
      m4 c2 P6 c' s+ z( k
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    $ L, I; R4 s2 h$ _
90. result.Delete(0,   pos);
    & {  R$ j1 D& I2 q
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    3 u7 B" Q; L7 O7 g+ I( P# h2 i
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ! F7 O$ N. N! b9 b
97. }
    
98. 
    7 H& v" ]$ ~5 T* Z8 x7 u6 O
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ! A/ l+ `" z* l. U" a! v, j
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     - |# Y/ d4 X1 P, o# D
106. if   (posStart <0)   return   CString();
     * P2 c# C( ]8 L9 b6 H6 R
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     * W. j1 X2 @/ L+ Y
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     8 P$ I/ U7 d) F5 f4 j  w
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ) r7 b. d8 b/ d2 W1 W# L
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     # M  @, B5 |1 A9 U( p9 c
120. 
     
121. MyUPnP::~MyUPnP()
     1 ?% f  \- N5 A' E! G" f
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ! x% ]" ~6 g( X' a5 Y  w
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     - Y0 p& C1 O6 e) S( R3 B0 H
127. RemoveNATPortMapping(search,   false);
     4 \+ E' I9 D- n5 o9 D
128. }
     
129. 
     4 R6 |0 e) j' y
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     8 {- ~& f6 A  A" a7 n- N9 O+ f
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     / E8 y$ Z  m7 a8 x& h
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     7 n8 ^# P$ k; \. ?+ u+ b
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ' x5 l( ]) P; D( }. B; [: M- E
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     : k( d2 A: C, c: |7 s+ i/ h6 ~( f' j
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     , B2 C4 Z3 {7 k' K: Z5 N
144. };
     
145. 
     + r' \% c0 @$ k( n; q
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     , n& B0 l' c; `/ Z  q. B0 J. Y
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ' z9 u  D# ]5 c8 ~" P+ `& T
149. 
     6 I; A* l. L' E- f5 o. ^( k- P3 `
150. int   rlen   =   0;
     
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);
     1 p1 Z  _# @) J& t% X# S8 W8 n" Q
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);
     0 U$ k, x* \4 B1 |# }
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     ! M& u5 `9 @, j; J
161. 
     2 V' Z8 d/ B9 \# r" |; n9 [
162. Sleep(10);
     0 \: r! s7 E. E& C9 S! s$ S
163. 
     
164. char   buffer[10240];
     ; j0 z% c7 U, k6 B! M
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     6 p7 j. L( }* w' g1 r
168. 
     8 h* o7 _* z, l; v5 Z
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     $ S; Y. f2 ~* g. ~0 s+ W3 o
172. 
     2 }% }6 s, |& n: ^# ^
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     9 a3 {) {& O( ^1 C, R9 }
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     8 r" g/ M8 m- \) d
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     $ \& p3 o- q1 n4 _9 z( r( C
184. m_description.Trim();
     9 [  y8 E7 D7 _1 M7 u7 v; o5 `
185. return   GetDescription();
     ) ^5 c) B8 ?7 P- Z
186. }
     $ i7 Y% q, c; v. b' F8 z$ L
187. }
     0 D  h! t1 X& ~! y# m
188. }
     : ?4 G4 m- S% {9 s
189. }
     
190. }
     ; N# U" I# G! O  Y
191. closesocket(s);
     % M4 j3 G2 v+ @' D& F3 p
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


6 S( Y& d; p. w( `5 {4 V) [( r///////////////////////////////////////////
! I9 T; S8 [1 |/ b/ x//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>


  }0 _% K2 @. b- V  P) X) d$ ^  I  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
3 V1 U3 n5 y* j, R1 L, n" o! N};


8 _4 t1 Q0 [. C5 a+ henum UPNP_IMPLEMENTATION{
5 [- A& p6 V7 o& u1 G% u7 X- C% F        UPNP_IMPL_WINDOWSERVICE = 0,
0 Q$ r5 |/ [! R5 G/ L% r6 I" {        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
* }8 q6 s8 s  V2 H2 Z7 C};

$ w  T) a3 }$ w: p, n7 I# Q
9 Q% F2 H0 m) D. S5 q6 _  ~

class CUPnPImpl
& k* d6 C, _& f* ^5 Q* R( `{
public:
" n6 ~) Y7 X* X3 x        CUPnPImpl();
/ R6 X2 X* x# o3 n4 K4 O        virtual ~CUPnPImpl();
4 N4 c3 i/ e' Q1 s7 c        struct UPnPError : std::exception {};
0 i# P3 ~8 t' a        enum {
5 x# H8 c; ^& S7 l& ~) c; Z( D                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
6 Y1 h3 h' ?' F        };
9 q; _. }+ ~7 ^- X

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
& r' g1 z& P# _/ ]) N2 ?) v9 y        virtual void        StopAsyncFind() = 0;
: {1 b3 x' {9 Z) l( P1 ?1 o        virtual void        DeletePorts() = 0;
6 s, l' J  r. U. K& G, z# L' s) V        virtual bool        IsReady() = 0;
8 C$ t% k5 p* {5 S( E        virtual int                GetImplementationID() = 0;
/ D% K, h! ~4 m       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


0 S. r6 K5 J6 k$ O* R$ ?7 B        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
# F" x7 k- B# v- _, |9 Y- v( Y        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
" Q  H! H$ V( D2 h8 J        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
+ g' {0 ~# i% c5 |# p# S4 a        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


+ h, m3 f( l! ~# o. w1 _8 N// Implementation
# O/ n! a6 T( l& L. y) Z- M5 Uprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
, ~# n7 m- Q8 i' a! m+ B, M5 G        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
, N1 x& P8 ~7 V3 A; o" b        bool                                m_bCheckAndRefresh;
# z5 t1 j6 P) n, c. h5 X

+ ^6 j5 f0 `3 |2 pprivate:
: L6 U5 ?7 @( e% b  Q6 a% x9 I        HWND        m_hResultMessageWindow;
' n8 P' V% p/ ^' P; t+ y        UINT        m_nResultMessageID;

6 h- M" A, X7 X  P; E
* y; a4 o/ @; f};


// Dummy Implementation to be used when no other implementation is available
' g- V, S- B9 |( C$ y# E; i' xclass CUPnPImplNone: public CUPnPImpl
{
( {. M6 P' @3 d/ v6 J6 Spublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
5 O, J) v' k5 A" X        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
! w2 {$ \2 Y* b7 e- `        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
% h! J; [9 x4 d) [* t% j3 ?

#pragma once
& R7 i6 Q' h. b. p#pragma warning( disable: 4355 )


& m/ \4 {+ ^9 M  x, x- ?4 Q5 j#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>

  b! N% ]1 [6 a4 @" Y
#include <vector>
#include <exception>
* I# p* D. T3 Y4 P5 h1 q3 ?#include <functional>


" A% T* o5 K5 s, w6 F/ @- [% g( A

: G7 A/ @) _: ?8 M1 ^8 d) atypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
- D- S( T5 d" U; {# w! vtypedef _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;
+ @6 i1 ^( a6 _typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

7 _8 n. P0 b3 \6 [4 R/ \) S* T4 ^
typedef DWORD (WINAPI* TGetBestInterface) (
5 m5 I* M0 J3 q. `" V7 k. Y. n. @  IPAddr dwDestAddr,
1 r) ?: ^# k5 ^* [+ y  PDWORD pdwBestIfIndex
);
* P( `# i7 u. P% M7 ]5 |$ ]) ]9 |
0 g; Q' G. H2 J2 I
+ w* W3 Y6 R7 ?( \typedef DWORD (WINAPI* TGetIpAddrTable) (
* k! O! h$ m! i+ W2 q" k2 u  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
$ N5 |2 W3 T4 Y& U2 P7 d  BOOL bOrder
- @: g6 `3 E0 \3 v7 C+ C);
, z1 h  Z" H- v1 N/ |% p7 u3 r
3 ?2 P3 e  {! i+ t+ {
typedef DWORD (WINAPI* TGetIfEntry) (
/ P& ^/ H# P! \8 v/ ]. j  PMIB_IFROW pIfRow
);
2 }$ t2 x4 e. Q2 ?8 [& V
4 c/ l1 v& o8 G7 f- q4 _
CString translateUPnPResult(HRESULT hr);
2 w; E9 s5 d  r4 fHRESULT UPnPMessage(HRESULT hr);
0 O8 V6 _6 T( h2 ]

( q: `1 `, Q  x; ]$ Sclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
+ k8 x% f: h: m2 D        friend class CServiceCallback;
7 ~9 O# d* R% ^3 }) a  @4 M// Construction
* _& U9 D$ d6 d8 u- S  o9 bpublic:
" G1 K% l5 \1 B' h# [. m        virtual ~CUPnPImplWinServ();
8 m3 h7 Z3 P* W% @8 D        CUPnPImplWinServ();

" `& O0 ]4 Y+ w4 F: a9 N
5 b6 ]7 p6 V+ H7 v' P7 e% l. J        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
. X; E4 R. m5 d        virtual void        DeletePorts();
        virtual bool        IsReady();
/ `& a7 Y8 a; O0 n        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

- {9 B) H1 R8 C+ j3 @% M
! u% d! p: p( `1 U- z( r& H        // 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
        virtual bool        CheckAndRefresh()                                                                                { return false; };
% O/ \5 n0 L. C' V% m
; _% I$ \0 Y4 d! L& b& B
protected:
5 c& a+ @/ I! [0 Y+ u9 }        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
) \4 P- l+ z, R7 B- \        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
& P3 R; P1 [, V6 L, R: T3 u        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
5 e: f* ~. B# P0 S! v$ i- s        void        Init();
5 s6 y& ~* J! g5 H$ Q: e
7 s1 X9 t  B$ G1 g! m& @  x
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
! T% O2 E6 f. P1 c; {% v% b) x                }
                MSG msg;
1 P  U8 t# a6 }1 i! z$ n6 ~  h* l- C                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
0 H' r5 K1 N% |. I' X. D5 a2 n                {
" A6 g% Q/ B- i8 x! T                        TranslateMessage( &msg );
9 {( d* }8 x0 A' ~/ O: O. T                        DispatchMessage( &msg );
                }
' r8 `! e4 k5 {# _/ a: P                return m_bAsyncFindRunning;
        }

9 _; B" ]2 e4 j0 r
        TRISTATE                        m_bUPnPDeviceConnected;
# U8 N7 b1 ~. `" I" M! \' B

// Implementation
2 @' R* q! s4 d        // API functions
: M' o. w) ~8 ]; o        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);
/ e( o0 l6 i: e9 V8 O8 @7 s        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
1 _# B+ z, ~, O2 H& Q9 ?9 ?# u        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

* F7 K* I1 g! J6 L% q2 P) C: o
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
& P/ I8 q2 I7 c3 c% h7 Q8 A& O        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
1 G, J/ o; z* D+ Q& j" O                        CComBSTR deviceName;
4 L4 C( b+ S( f/ G: f2 O( `- u                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& a7 R. v5 o4 o! W6 ]7 F
/ ^. M4 ^9 g+ [5 Y3 e
2 l* O' z  R  T4 ^                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

6 _. c1 H2 S+ ~8 K2 h( o3 q* @
) e. @3 _0 A# F) U: U3 K! A                        return wcscmp( deviceName.m_str, m_udn ) == 0;
- M! S  t# W0 W                }
# g: ~5 I  k6 O( K                CComBSTR m_udn;
. I+ z, ]/ W3 E, Y        };
& j* u6 L, ]4 K: y# J+ n       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
! ?0 F; ?4 ^0 ]3 p8 \' s5 B        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
! w9 k& T9 q% M( s        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
( y5 }: U) ?' V0 h' ^. e        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
: N( b0 y# }/ ~# E/ K        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
7 E% ?! R. C' D( ]: N+ N+ b) O                LPCTSTR pszInArgString, CString& strResult);
) l; B# {( c1 a        void        StopUPnPService();


0 b% Z1 H+ a! F- u6 e        // Utility functions
5 ^- H3 X- I) z, s$ \1 j! l        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
% K. g% R" r9 O4 E8 c5 Y6 f$ o        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
! _* Z3 E5 l  q9 y0 v        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
, h0 X- G* b$ \; F, q1 r

// Private members
3 s' b* c  D+ u, X' [private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
: }( n/ B& J% O: _        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

; t+ b# l: e( u* P7 o; T
        LONG        m_nAsyncFindHandle;
- S0 C" h/ \: w& N! F        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
% [0 X1 C4 j+ A+ _' g* j6 F        CString m_sExternalIP;
* w6 g% s8 X4 ^# P9 c. `8 q        bool        m_bADSL;                // Is the device ADSL?
2 {5 w: z0 g8 I0 G        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
+ c! k' N& Y) O4 v        HMODULE m_hADVAPI32_DLL;
6 I. V' |' n3 R, ]  ?  d        HMODULE        m_hIPHLPAPI_DLL;
. O  c* g" @5 j4 `, x/ z$ [; c% |        bool        m_bSecondTry;
9 `5 G" Q; D9 q        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
& r; n) t# ~3 a        bool        m_bDisableWANPPPSetup;
. ^5 W4 d: e$ K1 M

/ |& r3 r$ y; M1 Q& A$ A};

; I2 [) A" Z+ ?
- [- ~3 k0 e; B// DeviceFinder Callback
2 }; A; q; @" W" Lclass CDeviceFinderCallback
3 R, u0 k4 X* `" w2 K) _6 m* Z' I        : public IUPnPDeviceFinderCallback
{
6 W/ @9 g/ g9 I- W" m+ Wpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
; ~) k. M, e( |" q) h& C6 t( E        { m_lRefCount = 0; }

' B# j4 o4 G3 A
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ t: a8 \: C5 M   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) t$ L6 X. k; w2 }) A

// implementation
private:
( B! N1 H5 F3 D. f# ], W2 H        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
: N: m0 w+ _" v: X& ?. f2 B) G        HRESULT __stdcall SearchComplete(LONG nFindData);
0 N( G) Z5 s2 @8 O4 j0 v

. P4 |4 J5 {9 p7 O0 dprivate:
3 ^# O- a) f- G$ ?$ p7 U3 ^4 N        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
: d" K( U  I+ W( E8 S, t};
1 q. |& [! g; L- h8 B  J$ n/ `

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
6 ?$ @% l: n$ {& \{
public:
9 P0 k& j9 E9 J0 l0 K        CServiceCallback(CUPnPImplWinServ& instance)
1 W" v- o0 c. f# |                : m_instance( instance )
/ i0 @" V/ ?! T7 ~( B; m3 t        { m_lRefCount = 0; }
$ F2 L' r& _0 ~( H, H   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  n1 d1 }  @0 p2 X, L* f( K1 D   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
: v. g. j, V' V9 r9 X& Y; y

8 z9 r" _7 l0 N5 d1 E. ?// implementation
private:
( |$ K! b5 i# @) J        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
  v( _& h; P- y8 `, x1 R
9 I1 x5 E) \" ~$ I
6 E, [) R- E6 _3 mprivate:
        CUPnPImplWinServ& m_instance;
! ]; R3 `' e2 {        LONG m_lRefCount;
};

. N( _& \5 s: H% \3 v' e  q6 `
1 a: c9 g$ p5 B- [5 x1 H! x/ i/////////////////////////////////////////////////
" Y: _7 t' U! G' X" T

4 b8 D# E4 Q% a' w& w1 ?3 `使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 c, q9 Q3 q& p5 p8 l- iCUPnPImpl::StopAsyncFind停止设备查找.
. w7 T' D  |8 s! ^8 m/ oCUPnPImpl::DeletePorts删除端口映射.