UPnP

天云上人

/*uPnP.h*/

1. 
   5 e' Q& l' k" v( r- q
2. #ifndef   MYUPNP_H_
   
3. 
   4 K' C" H, h1 J4 P- \1 M1 u
4. #pragma   once
   4 f3 {+ a- @  |5 G# O8 T: `
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   0 K) r8 g. B) w3 q6 ]% \
8. class   MyUPnP
   , U2 t+ t( B2 S) ~+ o8 @
9. {
   " B* ]6 e$ W' T1 t
10. public:
    + x. x( K  p. E9 G3 H
11. typedef   enum{
    + x& `0 B+ {" [- K4 `# t6 R+ k3 D# U
12. UNAT_OK, //   Successfull
    ' x0 F8 n; Z1 T! a% e& L
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    5 k% _8 z5 V5 d
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
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    # \1 L, c% m3 |$ [  I
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    # k- E  Z0 K* W& t
21. UNAT_UDP //   UDP   Protocol
    + z# f/ a+ }& N6 d. h
22. }   UPNPNAT_PROTOCOL;
    
23. 
    3 K! f% l5 Z2 k
24. typedef   struct{
    ! C* F9 i! P! d# Z
25. WORD   internalPort; //   Port   mapping   internal   port
    8 R8 j* M* E4 i( ?
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    8 C5 ]/ Q6 D% ~$ w8 Z
28. CString   description; //   Port   mapping   description
    " p0 p# z/ e0 s
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    $ B, h- \' D3 G5 [, t7 g
32. ~MyUPnP();
    
33. 
    ! W# ]; `, }3 t; w3 c" R. {; ~
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    # _8 h8 d0 C9 {: n+ Q
36. void   clearNATPortMapping();
    1 a3 G; n' v  r) @' n$ `
37. 
    
38. CString GetLastError();
    8 V) m- s7 g9 l& ~  m
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    : k- U8 U* D) p* t
41. bool IsLANIP(WORD   nIP);
    
42. 
    9 T$ M6 U9 E! P# j- ~% r9 m& v8 S
43. protected:
    
44. void InitLocalIP();
    + C7 d* m* e% U6 e7 n% ^5 A% B- S
45. void SetLastError(CString   error);
    
46. 
    8 L- q$ Y4 R/ o, @& T% A
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    . n& J+ w1 N; D: B: V2 v  M# z5 I$ V( n2 R
48.       const   CString&   descri,   const   CString&   type);
    $ G2 \3 r; t, q4 C
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    1 Y! q7 Q2 @% T) P. s8 [' y
54. bool GetDescription();
    4 \& X0 P' ?2 S' S0 q
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ; d% B" K6 [0 T1 t. p+ A
57. 
    " s% _& W9 N, P% z
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 R& E; O! s( t+ `
59. bool InternalSearch(int   version);
    $ S* S5 Z' B( e# z7 ?
60. CString m_devicename;
    9 ~$ H9 `" R" U) u
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    & R; j: o  V  i: R5 p9 i
64. CString m_controlurl;
    
65. CString m_friendlyname;
    / @& u# e2 |/ Y6 p6 Z
66. CString m_modelname;
    . E  y. S3 `2 w  ]" a& t+ u3 b
67. int m_version;
    * v8 ^2 _# T( T) H& E7 s+ x0 c7 S. x
68. 
    
69. private:
      z8 R8 X- |: P& g, R, g& _
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    1 D8 _' S& ]5 `/ [1 A
72. CString m_slocalIP;
    
73. CString m_slastError;
    2 r$ T& {; S+ x7 W" _" N
74. WORD m_uLocalIP;
    
75. 
    " ?1 x. l5 Z: |4 S! g: X
76. bool isSearched;
    $ d( f! m1 G' |$ e
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
     h. K# \% o/ s0 U6 `  c0 m
3. 
   $ H( ~' J) s5 B7 C' B0 t8 t, _
4. #include   "upnp.h "
   ; u+ b0 a. P9 b0 i+ q, N
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   4 n6 _: z) {* g6 }: I! K- ?
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   " \" A% L& B) x
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   7 X$ K1 P+ {# _
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ; ]7 O, |4 R( I% p
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # F& N$ d+ m) u# I! K
13. static   const   int UPNPPORT   =   1900;
    4 L6 {/ C: X! E+ l+ h( I% Q
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    ( x% _0 `! T. d2 R1 A' N$ P
19. 
    / W: Z- e1 |& \- m  ~; l
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    0 S; i' M* y7 \9 ?" j7 x# M
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    9 I5 f0 ]# g; v
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    5 F0 _) d) }) ]) @+ _. u
33. }
    . @1 m7 y5 P' U4 {- T
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    3 s# I4 A8 `6 F
37. char   buffer[10240];
    : `3 I7 X! Z$ ^# z* W' W# q
38. 
    
39. const   CStringA   sa(request);
    6 N9 E* Z: L; j9 ~# z* ], l$ @5 e6 V
40. int   length   =   sa.GetLength();
    : d% D- Q% M# I6 w) ?
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    : p+ m  G& @& c$ F# e
43. uint32   ip   =   inet_addr(CStringA(addr));
    5 \2 [( D& k8 L# m8 ~9 c3 e
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    4 e7 A1 C  ~1 w$ A. [
46. sockaddr.sin_family   =   AF_INET;
    0 j# _+ G2 p1 e2 w! b: g
47. sockaddr.sin_port   =   htons(port);
    & U$ Z/ H7 e$ c
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    4 U2 ]- h, F' M" j
51. ioctlsocket(s,   FIONBIO,   &lv);
    # ]9 K  i4 f5 i$ |
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    / S3 }$ D* C7 s! f. i6 @
54. int   n   =   send(s,   buffer,   length,   0);
    - w0 O) e/ _5 U
55. Sleep(100);
    0 ~6 @( {" p$ r. t
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    $ }& S8 ?# g+ l9 i# `+ ]% y
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    " b2 D% G$ R) d* K5 l
60. 
    . U% r+ M, \* E. V+ K& i5 g1 f
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    1 l- J7 ]: D, r- }: P* i
63. return   true;
    2 O2 j' N7 o- g% O
64. }
    3 o3 J3 C/ H7 l( y% _
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    5 @) P+ V  U$ M$ p
67. {
    + |; V' v! |5 `9 E3 u; t( j
68. char   buffer[10240];
    
69. 
    , v$ T- Q0 B9 _+ Y$ P1 T
70. const   CStringA   sa(request);
    - _: D( I8 w7 a# k( {/ A) L
71. int   length   =   sa.GetLength();
      ]5 w: @! v8 }. c1 J: T: O
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    & D' N2 k) ~" X) Z* P7 h
74. struct   sockaddr_in   sockaddr;
    ) v, m9 H6 ?, i5 u  Q
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ( O! j+ Y& F  Q! F1 t% d. L# K
77. sockaddr.sin_port   =   htons(port);
    5 _3 Q0 ^3 D) I4 ~3 \6 ?9 I- e
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 t  |# M& ]1 |7 ?' E# L4 h: r
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 F$ p) c9 d# A9 T6 r9 V/ O/ n
81. }
    % m' S3 ?7 P; o8 l+ f! Z' X) {; D
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ) v4 G0 ~- j2 ]" D9 ^9 J2 l
84. {
      j1 W) S- O% s& M6 k
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    * Q: P6 M2 x( A) H% b  h. y- x
92. pos   =   0;
    3 E  O3 f! H; }, t, p+ n3 p, w8 f
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    $ Q) n% S! ^1 O: o3 {  U4 c) j6 w
96. return   true;
    + i5 ~2 `! r: `4 i" C& h
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     8 t+ z0 }/ Z- O- z5 h
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     0 N% R; g0 l, I/ m) Q2 a" Y+ l3 o
104. 
     4 k4 p9 u8 f* t7 k4 d2 z: z
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     8 k; T  p  J/ I/ R/ v
107. 
     8 Z4 Z: d* U6 S- r6 k
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     / q( s' o5 t! _( I
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     # L. ?; u# I; q2 i
112. }
     7 k( i$ i, P8 x1 g
113. 
       {8 s7 y/ t# r+ \' J% z
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     ( Q3 O' r  R. z7 n
116. {
     / p# @8 O$ S; e9 U3 j
117. m_uLocalIP   =   0;
     * Z9 K, d$ m0 r2 @% }' z$ i3 [
118. isSearched   =   false;
     0 j2 ~; a5 v$ W7 Z6 u
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     / z8 S/ S) F2 ~# t
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     , v6 [# a: l0 v2 X* }$ E# h; v
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     . x/ f. r( U6 h9 Z, B! k
131. }
     
132. 
     
133. 
     9 o7 i0 E8 z" b  g- b
134. bool   MyUPnP::InternalSearch(int   version)
     , I$ v  E1 ]( E8 G+ @! S
135. {
     8 f4 X" v+ ^5 q& t% }4 D  g
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     4 ~$ V# _$ b' t- [
138. 
     , G9 x: a+ l7 u
139. #define   NUMBEROFDEVICES 2
     4 r$ `( H- n2 D9 e
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     2 r" B" k/ l7 m: @, R
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     1 Q  d' g- b. x  L) ~" ^* ^
144. };
     
145. 
     ) B0 z; ?2 p' E! u
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ) I3 q' e- I# N. K1 K! e
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     & s' Z, r8 D8 P! {/ E& u
150. int   rlen   =   0;
     , s3 G/ C; ]% R7 ~# d
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     $ l' k) \+ l7 ]2 S3 W
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);
     5 U* a" N! z& i# l7 b
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);
     2 L7 Q, Q! d2 k5 O, G' A4 ~
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     - o0 D; Q: U! U- v
162. Sleep(10);
       S5 N% r- _! |8 O
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ) M$ p" G% T2 q2 y
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     8 f& f9 h0 X8 Q& x
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     . L& q. t% t, f0 ?) P7 r
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     0 a+ }* C" U8 m9 I  t3 W
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 ~3 U& z  y: f. M# n; P
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     ( O: B5 P  f6 ?7 ~! h- {
185. return   GetDescription();
     
186. }
     + f& ?; @" C8 I3 a6 g
187. }
     8 @" u7 E( Z) Z
188. }
     
189. }
     
190. }
     9 j  }) X5 y* w5 z
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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

$ ^% L1 M- i; A4 X
" D8 F  [& R9 ^8 b///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>
3 w3 K" y5 ^1 K, V$ y

  enum TRISTATE{
        TRIS_FALSE,
9 S5 D( a1 k+ A# E( q        TRIS_UNKNOWN,
- Y5 r4 H# |2 _4 Q' j        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
3 E/ d% \& u" ]' L. e3 L        UPNP_IMPL_NONE /*last*/
7 O$ o. u1 a' w0 u/ @};

  T; T( G4 u% c  B( k3 r" {
* s" S& m0 P! d3 v* T! I

class CUPnPImpl
% g' U% L9 \  q2 A2 M{
; ^  Y5 H% A  n4 \public:
: e+ w$ v% N( v! y        CUPnPImpl();
. i2 P. I7 G4 ^. w. x        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
4 d( c( h8 K3 v4 w                UPNP_FAILED,
                UPNP_TIMEOUT
6 ?6 x# s# h7 A        };
8 ]3 K, F' y) c% |! k
. H% `, N4 h" E
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
3 b9 ~2 P% n' G& B& y/ I3 o) U  \        virtual void        DeletePorts() = 0;
6 P8 a# N+ O% w. p- B1 }5 C        virtual bool        IsReady() = 0;
1 P+ J' p( L% a. @( B        virtual int                GetImplementationID() = 0;
" m. F9 p3 N) Y& g       
1 }4 W/ u4 g- b4 G# v2 S5 `' @        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

. K+ {/ b* j8 F: t5 O' H! e! b
% `& ^0 Q% v/ r        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ K8 K) H- }" \* v" [3 s        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
, Q( ^1 N. J5 ~% M6 N3 z
7 {! m1 m* D/ F, N' x% v/ S
4 U5 e- e( T" N" e+ f- @# @// Implementation
- a6 i! W  H1 I5 ^protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
/ P8 m# a3 F# I1 A( F( n) o' W        void                                SendResultMessage();
        uint16                                m_nUDPPort;
$ ~9 N, L- z8 k0 H        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
% F# m' ]) r: Q9 _) ]        HWND        m_hResultMessageWindow;
5 C6 r; Q; ^5 @, `8 [+ y' S        UINT        m_nResultMessageID;

/ ]6 V  H! O3 V, f' ?
};


$ U1 J& O/ R( |$ ]' Y// Dummy Implementation to be used when no other implementation is available
, Y3 D3 H1 `- [! g! @+ C9 Qclass CUPnPImplNone: public CUPnPImpl
+ V: ~& q  p9 S* c{
3 J$ S  |; ^! cpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
4 U6 ~; z: n$ B* m* m        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
! l& X7 S3 |. w8 o# {2 q! U
4 C! o4 j% c& u' w
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

* J! `$ m: d- Q
#pragma once
: `# \8 ~! F5 D0 u1 ^  s! b#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
4 J2 n8 H/ f7 R2 R#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>
2 {, Q9 b( ]! e3 t
/ T1 A4 }. {7 K* l9 W# l8 E

9 H( Y( S, A- P, \' R# a. A! }8 H
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
; k- Q, n/ i- U" I: H0 u9 j- v( Dtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
+ @6 G7 E8 G) C: E+ ?. dtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 m- C# L: i6 \typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
/ Q% W( e/ Q) C

) ~) h* m( j9 Z+ h' _. u9 Ptypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
, u' e* D2 f/ y  z- t6 H/ H/ r, g  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
1 R. R6 M; y: u5 ~7 i- @( f  PMIB_IPADDRTABLE pIpAddrTable,
. Z( o; i$ f1 q1 Y7 h* i& U  PULONG pdwSize,
  BOOL bOrder
' ^4 {! k# p: W);
6 r5 u/ W5 r. M8 q5 w& w2 a" U

typedef DWORD (WINAPI* TGetIfEntry) (
1 ?6 ?2 I6 [2 L- Y  PMIB_IFROW pIfRow
' N- n! d1 i7 U, l' D/ b5 z6 A);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
' H! P6 n2 U3 W3 u- \
- p; k! }8 R+ R# c# D* {2 m
/ t. i+ b( }+ s, oclass CUPnPImplWinServ: public CUPnPImpl
; _: p0 Y9 n% q! }' q0 I0 V{
        friend class CDeviceFinderCallback;
; G& D4 x/ h* t/ V! @; {! e2 e        friend class CServiceCallback;
- Q/ `  f. y: q7 b0 c% T// Construction
, s3 a8 a' n  }0 |" qpublic:
8 T6 d! b) S6 v" U; W. p        virtual ~CUPnPImplWinServ();
5 s+ |' k7 Y; H) Z3 f6 e        CUPnPImplWinServ();
# r+ w0 l5 R  y& ?3 b$ d

7 d1 M1 Z5 C  l/ Z2 Y        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
# l1 i. R. L) X        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
7 O' A7 F' \& I: n- h
7 _0 C' u+ \! Y- {0 ~" W% z5 p5 z9 T
( Y2 ]7 w0 X3 w# G        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
( s, b# n- L  y7 P! X        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
5 o) V3 p) a$ a

$ z3 J7 J$ v* j# V3 Eprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
# B( z/ r0 W! g; ]        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
9 V6 z! B( J: C8 }: r/ q& f        void        RemoveDevice(CComBSTR bsUDN);
  ~! {9 B! i' l9 `        bool        OnSearchComplete();
- e# t* p- ]7 y        void        Init();
" {# x! u/ {+ S5 I" l
! ~8 U" ~) s0 D8 H; A, D4 W' w
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
+ D5 j% R4 E/ b                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
5 x# p; |" B4 e: }7 w  }                        m_bAsyncFindRunning = false;
/ A, a0 ~5 x! X& `                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
) G' F# w9 ?' P6 k" _. k* b                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
' U3 N) E3 _# b) e9 Z; x! N% b( i$ w

        TRISTATE                        m_bUPnPDeviceConnected;
) v6 }/ \9 b+ t: s

// Implementation
5 S0 K+ w' N9 ]! [+ b        // API functions
6 i; k: \; `! C4 n8 t) T0 }" C- a        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
* m: S6 |" `: n8 m! l4 N# G        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
. o% C& ^# ^8 B# A1 q) ~        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
1 U! e& J  I4 p) r7 Z8 s2 y( W1 C        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
+ @+ V+ U8 ?" r* q        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
! [5 A+ E9 L$ b& F

8 m. Y0 C9 s+ H% a3 U" S2 _  s0 w        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

- p3 G* y3 c! n' b7 m$ v; _
" o  K, r$ g! S! |* b        static FinderPointer CreateFinderInstance();
/ g, S- |  T1 u% C. M" h        struct FindDevice : std::unary_function< DevicePointer, bool >
7 j0 N' x* c8 x" F- b        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
3 Y% l# @% T/ M$ `; [! B' n3 `                result_type operator()(argument_type device) const
                {
. B1 x0 u8 o* a2 l                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

; C" X& |: L( x+ @
2 s5 H% f8 e$ }8 W9 ~0 N! S' a                        if ( FAILED( hr ) )
  w% j4 \6 o! @3 H% w3 @+ i                                return UPnPMessage( hr ), false;

) X3 W8 E; ^6 K4 U2 {( t: R
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
5 ]6 q" `+ l) p                }
9 q9 {) {2 y" }2 ~                CComBSTR m_udn;
        };
1 Q$ h" Z4 }3 N5 V/ {: J       
2 l  l( J# ?' U2 P, N9 w        void        ProcessAsyncFind(CComBSTR bsSearchType);
2 U9 o0 K: g4 g; J% p        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
. \: V$ r* i" P! z3 k        void        DeleteExistingPortMappings(ServicePointer pService);
; S' _) K/ S( @" D& C) G9 ]        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
; S3 a' L8 N$ p& r8 b) Y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
9 b4 l# b1 X7 b* L( j( u0 Q                LPCTSTR pszInArgString, CString& strResult);
: b7 N' k% d/ i        void        StopUPnPService();


' D1 V! n1 w+ i" _        // Utility functions
3 O5 n, c# T3 F0 g  ?9 `3 X        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
! K" M& U7 R5 ~+ y6 X0 \        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
1 q: F( |$ E7 i, h        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);
        CString        GetLocalRoutableIP(ServicePointer pService);

( ~7 f# Z! y# Y1 b& E$ I& g# t5 [- u2 }
// Private members
+ r. R) N1 w1 K; I5 w# B( ~5 z/ O+ Gprivate:
2 M. p  A; u  T' m' e        DWORD        m_tLastEvent;        // When the last event was received?
! s- Z! Z- n5 X        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
6 m' S( F! N8 [0 v        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
8 U) N3 K  t) \4 Z4 a* ^6 e: A8 g( k        ServiceCallback                        m_pServiceCallback;
. A% b, F$ L+ j" U" ~

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
; W3 W$ q: V. n! x        bool        m_bPortIsFree;
7 ]/ H: r% v, U% S9 s+ I3 q& }% |) g        CString m_sLocalIP;
, r4 v- I: h  C- L        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
# h# F+ q, k, v+ S7 m        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
) n/ i8 P; |+ L( N/ i( d* H( v        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
$ S  [% w' M- k: b& B" o        bool        m_bDisableWANPPPSetup;
( w5 ~# ~+ j% J: z& k0 T. T
0 c' c, @0 r8 d9 g5 d7 c& i5 W+ s
! }) B& F$ X! h0 v8 Y0 X};
) j& h/ `$ M6 b; @0 J9 B3 p( d

* U( f2 o& T1 [& W0 q// DeviceFinder Callback
class CDeviceFinderCallback
- z6 ^' x" @; I% L        : public IUPnPDeviceFinderCallback
& d- B/ |1 s9 V5 k8 B5 c{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


9 G7 a# G4 X% [* E" g% u   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
/ u  e9 D  }% @+ g% U! d   STDMETHODIMP_(ULONG) Release();
! i  a& t0 H% G- X
% h0 O. t# z- }) ?: ?8 L5 e
// implementation
4 Z; V" _5 B5 n  G! a2 pprivate:
7 r6 M' ~! ]- u: ~& M/ {: H        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
6 i  \6 m$ p8 Y$ X6 O  w% T! K/ x        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

: i, a4 M1 [, n' H# r/ p6 F  G* Y
# q% A; n! Q3 R" Q' ?" g/ T6 vprivate:
1 R  m* |. R/ Y& O) R! @+ x# f; G        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
! _6 q+ C! Q! c9 A};

6 j# ~! ?! j" v' }+ W
// Service Callback
; t# ~9 \0 L+ p: u% Uclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
! d* g& [9 T% g        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
! p8 E: i: o$ b7 j; r9 N        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

' p2 Z% k$ @; P. a! Z4 r2 [( J2 ^% o
6 m7 n' \# |$ b; X" c// implementation
private:
9 v) ?2 \% R1 I' r& ]% K8 _( G        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
& |/ n% H: P* }4 y

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
9 j9 a0 }# r2 H  A$ j+ g2 X};

. D, a! j- j5 [! R
/////////////////////////////////////////////////
3 f9 s, x' I9 m1 _6 t+ R9 V3 r
0 \: [. I) `3 X5 \4 H, K2 P0 g
使用时只需要使用抽象类的接口。
$ R8 X0 h9 m0 n" Z8 |+ Q8 a% NCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
7 H" V8 K+ |3 Y( W" NCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
% B8 s+ W2 B- KCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.