UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   7 L+ s  ~% ~/ F" j2 \
6. typedef   unsigned   long   ulong;
   9 J, t( C- P& o5 \5 {, q; K
7. 
   7 d' f9 I5 p5 s/ F& y
8. class   MyUPnP
   , `' l  n% r8 \# a
9. {
   ' _' P* X; h7 o3 {
10. public:
    ! w6 f, Z" @* \1 u& [* M1 V
11. typedef   enum{
    4 \0 P! J. Y5 c9 ^# d$ |0 Q0 U/ M; s
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    0 h# F: o/ m4 U2 Q6 E
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
    - q7 I, ]$ a6 m8 T
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    * I0 Z5 D1 i5 O2 t) A8 R
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    . l7 U+ h* X; k% J7 H/ |+ ?" w
22. }   UPNPNAT_PROTOCOL;
    # s: b3 u9 ~9 h: C" o$ w) w7 D
23. 
      n5 G5 m% y: J% g- e' o+ Y* U5 z8 ^
24. typedef   struct{
    ! A7 m; o8 j  G4 H9 j: M( k1 L8 a
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)
    " P- E4 X$ f( G# x& H( n3 w
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    + }* }2 F9 p+ w( H$ {
30. 
    $ U, }& I: X. h% c* M
31. MyUPnP();
    1 `/ [0 Z6 j: z; P+ }* O: ]
32. ~MyUPnP();
    ) y3 C3 Y6 J1 Q1 x% e' `0 l+ _" }( d# g
33. 
    5 Z7 d8 w- ^$ y; t. Q. i
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    $ ]1 j2 J& V# b( z5 M9 m* j
37. 
    
38. CString GetLastError();
    2 J( K/ v. G) n2 o2 g* \
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    / Q) S; S3 B+ b. |% E
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    2 T! j& L+ n' C# V( h  T
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);
    
50. 
    1 I* t2 w% b& t. w7 V
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ' d& Y3 ^1 `0 U( [- Z
52. 
    6 q- V* T" l1 @
53. bool Search(int   version=1);
    / t: S7 ~. `' H+ l
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    $ {/ f, {( w* b8 F5 I  d+ [7 t0 \
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    $ s% j2 r. \' P7 B2 C% Z
61. CString m_name;
    ) G! H$ O: V1 k7 R9 g3 k9 W
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    / i4 z& R. ^" H! p- I
65. CString m_friendlyname;
    - l- p: E$ _$ }
66. CString m_modelname;
    . {/ Q' T. @9 W: l* C
67. int m_version;
    
68. 
    ! u; B5 w; ~( y& H5 K% d* G
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    # q/ |8 X( Z- s$ Y/ m
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    8 U1 N+ W! V9 j- ]
75. 
    ) [* ^2 J. C2 i7 _# d
76. bool isSearched;
    
77. };
    % \5 V5 k' w: v' C9 {( p/ S! d
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ) m9 ]8 l( I+ @. E1 T# @& u
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   ; F4 E; ?1 L6 U$ L0 D& _
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   8 `. s# a" T' ^6 A8 p9 e9 G) Q
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   + y  e" k0 I+ e3 z, g
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   , u2 p6 }. ]# K4 }
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    5 f/ f9 D2 l/ C* t% F
11. 
    9 a% ?  Y0 L% E- v9 o3 k  _
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    * V/ _3 N# X# ^/ s! E. u
13. static   const   int UPNPPORT   =   1900;
    3 g3 v+ O4 ^4 ~6 ]) K
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    . x8 v$ }, i1 f3 F5 w2 e
15. 
    
16. const   CString   getString(int   i)
    4 X& J5 v3 }! |3 D
17. {
    ; V; G) j1 }8 W) y
18. CString   s;
    : g0 m4 A; M  v8 e# U
19. 
    
20. s.Format(_T( "%d "),   i);
    * X7 N3 y8 S+ j% [% W2 q
21. 
    - D4 }" h7 [0 U3 F) t% L, y
22. return   s;
    7 v; t  }7 T& I8 O7 r
23. }
    
24. 
    " j3 W' ]. e9 `' ?- m
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    5 J# l$ \" w5 Q. \- }
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    # B9 @/ T2 w: H! ?- q
28. }
    
29. 
    . I" q4 \9 g+ J2 h% r! k+ ?
30. const   CString   GetArgString(const   CString&   name,   int   value)
    6 s$ N+ Q& q' \& y. ~$ j* q% f- N
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    , T- V) @! E! ~, X5 X$ P
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    & a( U2 |4 w0 \4 Q- t" a3 f
40. int   length   =   sa.GetLength();
    & L/ G$ k; m6 ^  W
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    . V! L6 [$ J* {: D# b
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    * a. M& U) [, A4 r' z
46. sockaddr.sin_family   =   AF_INET;
    + P4 Y( n  ]% g
47. sockaddr.sin_port   =   htons(port);
    ) U: q/ t/ R5 w. B* `
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);
    3 B/ j% o4 o3 J1 t) Z1 D/ `, p3 p
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ! V! k2 t5 z8 N0 R3 b( V& t
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    & d4 C: W$ p8 W( b1 ]+ p, C
57. closesocket(s);
    , _1 U9 f8 z$ r1 a  {7 R
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 U5 e; h6 H0 C  U  Z
59. if   (!rlen)   return   false;
    * v/ k' b, \  F2 {0 `3 r
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    " ~3 ^& f5 n+ u+ N1 I" N8 X
62. 
    
63. return   true;
    
64. }
    # |/ f0 y# Q9 A) ?, M
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    # l  t; w- d' P0 Q9 I! F/ l! |& z
67. {
    
68. char   buffer[10240];
      c' u" g# @' Q- P
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    8 [8 m# A; z8 S% Z/ E
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    $ C: \& r0 |8 r2 {
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ! R* M/ a, i0 k
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 R  o5 r9 D! C9 v; ?- j8 j& f6 G4 J4 n
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . ]+ |' B+ X; Y- b
81. }
    ; K/ ~$ K- V/ r$ C3 X. n0 g& b
82. 
    0 c& w" v; Z9 K( H) e3 l
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ( B1 o$ L  W1 b- g$ R
84. {
    
85. int   pos   =   0;
    : \% e; j. T/ w1 |4 b3 ], J+ n# H
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ( w' C6 H2 P) c  [
91. 
      X8 j  v7 J% N9 f9 z
92. pos   =   0;
    7 R+ H& V  s4 x0 f
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    6 J) c" C$ d" ]! ~& z
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ( L! |1 |! m, S: p6 z2 D3 t. i/ z
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     7 C' P  J3 T+ r# q$ @
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     5 G1 c9 ?8 G  ?/ s% k. U
105. int   posStart   =   all.Find(startTag);
     - Y- f1 {. P# c. n3 `
106. if   (posStart <0)   return   CString();
     3 N/ z) d! j: o+ v- Q
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     2 j* B7 Q5 O- O: P
109. if   (posStart> =posEnd)   return   CString();
     8 L: n& P) v: B6 s2 E$ I5 m% O
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     ; S/ W+ q2 h$ A
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     9 t! B4 i% Z0 i
116. {
     % n1 ~" h. P  \( `; d
117. m_uLocalIP   =   0;
     3 |6 z$ j. M5 ?8 w/ U' Y
118. isSearched   =   false;
     6 e9 i0 H* o4 S6 {. M
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     1 U8 U4 G/ g) ?' H4 Q1 n
123. UPNPNAT_MAPPING   search;
     # T* j! }. ]- N+ r: R- y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ) M5 m9 s7 G( D1 O2 j* s/ g
126. search   =   m_Mappings.GetNext(pos);
     4 A" @- L+ l$ k) V5 f4 x; F
127. RemoveNATPortMapping(search,   false);
     1 @1 W$ v2 E8 a1 p) V
128. }
     ; ~4 t0 T# Y9 u/ k* u
129. 
     . C3 w" \1 J# I; U& w# a6 }3 \
130. m_Mappings.RemoveAll();
     2 g' k" S; Y6 P& j5 I
131. }
     " \" ^! W8 g( D% y0 N
132. 
     7 \! {8 F- i% s, }9 x  g
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     7 k0 Y+ l$ U( E3 p2 B* p
135. {
     
136. if(version <=0)version   =   1;
     . S& b& [. p9 v; Y# p/ u: }4 R
137. m_version   =   version;
     , N0 U2 T; q, P8 o' M
138. 
     2 h) c# _& ~+ ~4 v- I& k
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     & \$ z. w. ?2 A2 d5 o
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     : I* w5 }% Q0 W6 P1 _' |
145. 
     : R- t# t' n* J* L& I
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     . Q% W# v/ B- a6 S4 m- V
148. ioctlsocket(s,   FIONBIO,   &lv);
     4 e: r$ k* t% }  p1 P
149. 
     1 k/ r2 R6 v, l( u$ _9 `
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     1 |/ i: p  n, p% K0 |
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     9 J5 O- [# I- F+ C% o, S# ?
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     8 F) `! b$ r5 E7 F
155. CString   request;
     4 `3 Q" O0 W2 s7 O1 {
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 {* G! u) F4 B0 ?& a+ h6 d% j! z, {
159. }
     
160. }
     9 C% L- i8 K0 O# Y0 M5 j
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     , @* ?9 D# W% M, ^5 H5 G7 g
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     8 I) ~" A% d+ O5 V) ]: J
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     6 v. @" `" s2 n" i8 {9 N
171. if   (!parseHTTPResponse(response,   result))   return   false;
     6 V7 |: L" d% K/ `+ z. G
172. 
     ; _0 j, L/ u. B+ s  y
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);
       i/ u' |& X- o& H; x1 c8 `
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     2 J$ Z+ N9 Q7 p% g5 i8 v
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     * v- J* O- Z& J( K. p
178. if   (line.IsEmpty())   return   false;
     + V  j% T" D# x9 `* F- z
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     7 {, c" O- l# {
183. m_description   =   line;
     
184. m_description.Trim();
     . ?9 Q! Z6 w6 P) c  B0 Z
185. return   GetDescription();
     
186. }
     . }& F$ _. n& z/ h7 w) ]' I/ t: J
187. }
     
188. }
     
189. }
     
190. }
     * L. q5 G) w( ?8 j9 U: n
191. closesocket(s);
     ' R: L" y+ |; Z6 J
192. 
     - o2 x3 k' E5 M4 j* `6 |( e" J6 A& u: ~) L
193. return   false;
     
194. }
       c3 l  G: a4 r

复制代码

天云上人

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


' E) z( P9 E6 C: B///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
) J5 K$ y5 @0 Y% ]5 g8 g3 I( h#include <exception>


  enum TRISTATE{
6 ]- B. j1 P$ E  o* f+ Q        TRIS_FALSE,
        TRIS_UNKNOWN,
6 c$ d9 `# N7 R$ \        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
5 x' [$ N6 i0 I- z$ u0 I        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};




class CUPnPImpl
{
$ w/ F' f( X( U6 ipublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
& @! H1 Y( }; g# c0 Y        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
9 \% Z% a4 h% f7 w! h/ L; m                UPNP_FAILED,
) K& D# F' b; V8 }$ j4 r9 G                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
9 v2 j. X2 h& A- h& \; y1 z        virtual void        DeletePorts() = 0;
8 S3 @: s+ O! F1 K5 O/ k+ B: s" Y        virtual bool        IsReady() = 0;
! r- E% d$ K0 J# \        virtual int                GetImplementationID() = 0;
       
2 ?# b. L( j. i* }% H7 g/ L        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

3 r. W% H. q# G/ H% R+ L
2 V  j# c7 y6 b' L        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
6 g1 @+ [" m: p( D  C6 B        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

( z  L7 x5 C" p
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
* A( i" U' I. i# {0 P        void                                SendResultMessage();
        uint16                                m_nUDPPort;
5 m  V/ t( C2 c( t+ v6 U  Z( X( f        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

  W3 j( \# I5 P" o
  n8 K& I8 ^: l& |3 {private:
7 M& k3 \# l  H; \5 E6 {- g6 l% Q! r        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};

. }' S  _' o3 K  U) s
0 b0 }0 \, U2 c, _, k* M& k// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
# p6 w' R0 B6 F0 U! t+ k{
  ^5 P# Y# f. \; z0 Z9 z9 E/ T* W/ K- Mpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
% H2 `/ H7 u5 |  p( J1 Y- c+ E- F        virtual bool        CheckAndRefresh()                                                                                { return false; }
* b. v4 t3 O# [3 `        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
8 Q2 L! c- v! N7 W% @- \3 _        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
- a$ Y/ a2 z8 {2 U/ [& a};

, ]% p- M- G+ u
- [2 R+ D# c0 z4 l0 Z* W, q/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

1 x5 H9 V. R/ |4 B( }8 e
#pragma once
. s8 F9 @+ Z. E  Q+ a#pragma warning( disable: 4355 )
" x# t* h( d, \0 h; j7 q' Q
3 F5 y. W8 @, m7 Z
/ I, c9 O, ?) d1 @: c; L" [#include "UPnPImpl.h"
+ c5 U9 j! s+ U#include <upnp.h>
' y$ a' I  p9 {" B# v9 H3 K#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
1 Z+ \+ j( y9 V( S
$ b$ L3 g9 W) z6 y9 A
#include <vector>
9 l2 T( \9 _4 y+ R#include <exception>
4 N" t* p) s( }$ U8 A#include <functional>



5 _! j4 j& A* l9 ]9 A5 a
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
. h1 U$ ?# z- o' E; htypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
" R6 F8 Z0 _8 X2 s+ \$ O; u+ I0 N: ltypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
" b$ |; g: l+ Jtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
: I9 f0 B1 ^4 `/ Ktypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
* O9 j) W# f: ?7 e1 V9 H  k8 b' C

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
* c4 _4 i0 R# T0 P: a);
( o4 r; d: j5 J8 D
% V" x+ U) t" J% a7 X
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  p4 N# y$ d4 @& u  PMIB_IFROW pIfRow
);
# Z' Y( G$ Y' i; L! U$ {9 O

CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
7 i' W; D% R  h' T' M; j' o. a! @  `1 D

( v$ a5 D% P2 E, b% G3 @& m- Rclass CUPnPImplWinServ: public CUPnPImpl
8 v- n  I) O3 r4 v{
# g# \1 o; J. j) M        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
% d- g& M* Z# z+ G2 I) Q( m// Construction
! K0 A! w$ ^0 c( ~) @8 _- Opublic:
# E" q4 D* B( |        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

) J  l3 c5 i( w! t5 a
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
4 d1 n7 V0 N8 a2 {- e  _        virtual void        StopAsyncFind();
4 R1 ~: U+ v# x5 g, a. @, w6 _        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
* \" P3 n  \" s
' A& w- t6 P0 O5 s2 e3 W
        // 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
4 a# z2 L. _0 L" i9 L$ z2 o2 d. s        virtual bool        CheckAndRefresh()                                                                                { return false; };
+ l; T! m* V( [  M# }8 X4 E6 M8 V, [

protected:
9 q( ?0 Y5 @3 a0 n9 D' i1 q        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
% J  D" D3 D% J8 b: J" ~6 r# {        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
/ h- X* `- ~6 y& T        void        RemoveDevice(CComBSTR bsUDN);
3 j+ q$ h1 ?3 C. M        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
8 L8 y: E) E! }. A! F7 r) {9 D        {
: c& y! s* r; p; u$ ~" v$ Y8 l                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
  d- g- ]8 Q7 d: ^7 ^" W                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
$ c; G0 Z6 i. s5 T                        m_bAsyncFindRunning = false;
* ?; ^% y2 K* y' I4 {0 L                }
                MSG msg;
6 C5 a) l9 C- Y, U7 x6 G3 n                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" I- r3 k0 V- l7 |                {
                        TranslateMessage( &msg );
0 ?+ F6 C% a1 \% a1 O: e  l                        DispatchMessage( &msg );
" g8 T: {& B  ?4 I                }
, L& q; t  ~, T2 ]% E                return m_bAsyncFindRunning;
        }
) H# _9 [" k: g7 k

# Y3 R7 @. t9 [+ ?        TRISTATE                        m_bUPnPDeviceConnected;

0 T3 o9 J# X1 T; h; K" N& y& \+ O
" d% p4 E. h, g8 E( ^3 V# @) R// Implementation
2 J; m/ d& K9 N        // API functions
        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);
9 {5 T# l% H" p, K        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
) C- l" d7 }) ?6 k9 s, I        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
$ w" M+ _8 m1 Z& W0 D. ]

        TGetBestInterface                m_pfGetBestInterface;
5 `2 T6 a9 ^) b' P& {        TGetIpAddrTable                        m_pfGetIpAddrTable;
4 P3 ]5 L- ?3 _* h: ]* o        TGetIfEntry                                m_pfGetIfEntry;

1 H7 {' l" X; }* f$ I9 a
        static FinderPointer CreateFinderInstance();
( }8 k& I, u% I; n        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
5 v% p' ?% T7 w% d' ]& m$ Z1 Y                {
) y/ x% K; o  M9 |/ n                        CComBSTR deviceName;
% g6 z, }% x  Y; A! Q/ c  s                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& Z0 U: ?0 V$ S
9 \3 V+ i& k# s9 `' X* h5 c9 G
                        if ( FAILED( hr ) )
) p5 l- I2 a+ m7 {                                return UPnPMessage( hr ), false;
* c9 s( a$ w/ H, J! w' w' T7 E

+ B# l6 g' [3 r  ?                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
1 F7 j" {4 Q9 H/ @: d5 S                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
( }8 ^5 \; U& n( }        void        DeleteExistingPortMappings(ServicePointer pService);
% r) W0 s, m+ {7 J        void        CreatePortMappings(ServicePointer pService);
+ y) F0 d* E% B# i$ ?, J0 n+ @        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
" t7 C' P: N6 T                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

* ?5 r" k% g$ z* I8 X1 x# ~( W! D
# r! p  C/ X% j0 }+ D) l/ ^        // Utility functions
* }3 a) {. G) v3 P3 c$ i        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
6 X5 {; x/ i/ b; F9 V+ U9 ?$ ^( f        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
; g7 }5 K8 }1 j! {. V        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
7 X* t4 m4 W  L! j3 ?% e        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
( M/ G$ a$ w9 i% ]& \        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
# ~' n9 v" E1 b7 X1 e, |% g' N& _        CString        GetLocalRoutableIP(ServicePointer pService);
& q7 B7 e$ j4 N. V% Y0 Y
% X$ ]  i5 P* L6 n0 _! ~
// Private members
- u; r! ^+ V+ o- ^6 Fprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
) u  Z! ^5 S0 _( c8 X$ s) a; I        std::vector< DevicePointer >  m_pDevices;
* I* r# H3 i% _: ~9 m        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
6 ^3 \/ t5 p9 ~+ }" M3 [  R        ServiceCallback                        m_pServiceCallback;
4 `' Y, o* p1 [$ c& n2 i% }5 s
* I' J- `* v( E& {
. ^+ z+ N9 |+ D! e! \        LONG        m_nAsyncFindHandle;
% R) a( X; z8 v) c* K# O7 w) `        bool        m_bCOM;
% u2 T$ R( D7 ^# H' w7 b4 M        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
8 p0 r2 K  U3 X, l        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
9 K$ i! m" b' B        bool        m_bInited;
        bool        m_bAsyncFindRunning;
2 A' @1 }6 e/ a0 i        HMODULE m_hADVAPI32_DLL;
% G2 M* P+ p. q% V3 A6 t* D        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
' l$ `' c: ~6 A3 L1 B; v9 m; t9 o        bool        m_bServiceStartedByEmule;
, x- y6 h2 W. @* C- H        bool        m_bDisableWANIPSetup;
0 W# p: w& k0 ^6 g( w" ~5 u        bool        m_bDisableWANPPPSetup;
4 ]% b6 w4 [5 y0 ~

2 Q  G2 F, ~/ r( e7 ]  u0 i};


// DeviceFinder Callback
  H3 p, n5 }3 q5 P0 J" Yclass CDeviceFinderCallback
* R' Z( y3 f" D1 V4 H! k        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

; A1 r) P# B9 f% W, P1 S
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" v( L7 j6 g/ M# U) N0 i9 h   STDMETHODIMP_(ULONG) AddRef();
9 w& d1 C$ u# {   STDMETHODIMP_(ULONG) Release();

( v+ Z% ~$ ~" O  a7 n8 H
// implementation
6 D/ O, y( f  v* d- `8 ?private:
& L: z: B8 m) ~. F; K) x        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
* P4 D) t, ?) f6 q        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
1 A- p. S: e( [9 e2 `        HRESULT __stdcall SearchComplete(LONG nFindData);
9 D7 B5 s9 i& U

private:
        CUPnPImplWinServ& m_instance;
$ i4 D) y! h1 m& {/ x2 v        LONG m_lRefCount;
};

* F( C1 c6 j) I3 ?8 c
// Service Callback
: l( \9 b0 `$ ^class CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
# d5 L: m5 k0 o$ h                : m_instance( instance )
* u% i" L/ M+ c5 l4 X3 p1 _        { m_lRefCount = 0; }
; P, t. O7 t8 o3 d+ a" E5 m6 N   
/ _5 ?! p7 Y) [6 C" X$ q: U' t& d   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
, h; _  J( o; G! L" q   STDMETHODIMP_(ULONG) Release();
1 H3 {  q( s8 j5 b- `/ |0 l0 h9 K
" w) }! D, U% X, P4 Z$ s
$ n4 ]& Q. q+ a% ]8 k// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
, r7 D( g+ a" V2 v$ a" X        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
1 A5 J+ _7 a% i  X: k7 r

% F7 W+ E  R5 N8 f5 Q7 Hprivate:
$ W4 y; r2 w2 z/ u; k. }$ M. ~        CUPnPImplWinServ& m_instance;
+ q8 B/ M) n+ B' r8 X# z1 a( h        LONG m_lRefCount;
};


/////////////////////////////////////////////////
8 F# r3 o3 [- y3 A- v3 J' U

# R% R' c8 ?' r3 ]使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.