UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   + P1 d: k3 a, Y+ s( R9 N
3. 
   5 J1 l7 [+ w& S: C
4. #pragma   once
   
5. 
   ) z' s) M/ m3 T" ~  u$ B
6. typedef   unsigned   long   ulong;
   3 [* j) d/ R, L; i4 n1 [- g
7. 
   
8. class   MyUPnP
   1 B1 }9 r3 y( P, {4 l4 j  ]
9. {
   1 w5 R* f2 R9 }0 U8 \# a$ C. n' C
10. public:
    
11. typedef   enum{
    % q" Z2 R/ t7 b& l9 [
12. UNAT_OK, //   Successfull
    . R( @: o) V9 V
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    ) y5 ]3 ?) |! K7 n9 d0 y
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    3 ?; b2 X* K) y! @$ e$ e
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    5 U5 M5 T& v( w7 y( a3 R4 K* ?
22. }   UPNPNAT_PROTOCOL;
    ) m. ]" s9 W  ^- g# @
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ' r1 _, S4 J0 W" m  s
26. WORD   externalPort; //   Port   mapping   external   port
    ) u9 r/ p9 G8 \$ N1 a$ N. u7 I
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    . h* |  L; g) q4 O. i9 k  D* n
28. CString   description; //   Port   mapping   description
    - w) o: F8 o! k2 `1 [' F
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    $ H. r& R; u0 v: {! D9 l) B
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    9 ~- O- c7 T$ u' S0 p" w( y$ C
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 Q1 L) W+ p- r' z( n- [1 X
36. void   clearNATPortMapping();
    
37. 
    . f- h  l2 t1 g7 A3 M) w4 N
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
      q9 d  D3 f* Z/ D& B6 ~4 h5 @. t
42. 
    , u4 s: M# ^' D! j7 Y" Q
43. protected:
    8 m7 ^- l$ l) y; Z: H+ h2 ~
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    3 b2 x& J' d8 g( g: a" G
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / g3 d# s7 j: L7 r
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    : `7 K. I( l; W- x( o( c
52. 
    ; c3 \1 a) O6 X% z
53. bool Search(int   version=1);
    " a' R4 p: \6 t- ]! p) ]
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    - l7 m7 s+ X) }, |
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    . s6 P; ]  S7 H2 o) ]
59. bool InternalSearch(int   version);
    * }/ }% W  k9 Q* n) u; I0 P' o
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    3 R' M! v' z) P. n) M) Y. D
63. CString m_baseurl;
    / t. J7 u. h& {5 F0 y7 N) C7 v6 B
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ; r/ d1 M+ c3 e% s7 y, \
66. CString m_modelname;
    
67. int m_version;
    ( y$ O$ u9 E+ l( F- i: ]
68. 
    
69. private:
    / C* C7 m8 p% g
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    6 x! c& H( G; x5 F" \
73. CString m_slastError;
    4 Z' Y: L% A; X0 r+ R* M' p4 M$ V
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   2 p' ]* ~& K. f, b/ }
2. #include   "stdafx.h "
   ; W9 E7 `' U5 l) b3 f" [# r' w. O
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   4 n, t' B0 s7 ?) ~* i" L2 B4 k5 T
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    7 E" s  ]# G- E( ?) o8 ^5 @' J
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    5 J2 @& }/ u  J  {) p5 p
13. static   const   int UPNPPORT   =   1900;
    " o0 e6 \& V  p' F7 g7 I. n
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    1 E. e/ l, u; A
17. {
    
18. CString   s;
    
19. 
    ) A+ J$ b* q+ r, B
20. s.Format(_T( "%d "),   i);
    
21. 
    , B2 C' b' ~8 [* r+ j, p# D0 h" Y1 }
22. return   s;
    
23. }
    
24. 
    9 |" J/ p% n- _9 g7 M. C5 n
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    2 q5 j  r, j) U: o6 Y9 C' z4 j8 n$ i
28. }
    # [+ c; C9 h* E- V6 b/ P
29. 
    ! a7 {6 L2 a2 [2 A& s/ Z
30. const   CString   GetArgString(const   CString&   name,   int   value)
    7 I3 k4 P+ B/ x) U. h$ d7 c
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    ; K, _) h  r" V5 a+ t
39. const   CStringA   sa(request);
    ; v) V2 @7 }% G2 J  x' @1 K
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    # I3 f6 \2 b) R
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    0 g9 O! Q  G. `6 @8 Y% s
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));
    3 a/ H" K' O: j5 C+ x* u" Y' r. N
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ {4 t$ A0 E+ J6 b  ~- b$ ]* p
57. closesocket(s);
    3 |8 k% q" B% w
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    " w. T3 v8 `9 s
59. if   (!rlen)   return   false;
    , K# F: [: q) Q  c, h
60. 
    + z& O; ?5 E  r& X/ Z6 E" G2 q
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    3 m: k) K8 Y1 O1 W0 b, d- V
63. return   true;
    
64. }
    / P" [8 |$ r6 z2 ]1 F
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    % J( T, Q1 e, u3 S( M. e9 P+ G! t
70. const   CStringA   sa(request);
    6 v6 {- Q) @4 _( a. W1 `3 w/ K
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ; c. P, T8 y6 U& p- o8 ^
74. struct   sockaddr_in   sockaddr;
    $ h$ e9 b2 R( U! u. g  L
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' O& t3 _6 a6 `
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    % E: {6 X* B' z2 @: @* K  c: g# W
84. {
    # ^  b, ~: B: B  L! z$ a; `* u/ r, p
85. int   pos   =   0;
    . f7 D) X/ ~$ D
86. 
    7 D2 X) [  C/ h( [
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    ! @* c( v2 e8 V8 ]6 v
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    # ~5 r' i  j  t
97. }
    6 M: m8 S8 v. H' y1 e
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    * ^- ]# s+ O% T5 [: x. A
100. {
     0 i7 {2 h; c( n% B
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     $ w7 D/ a  n( L* J" N
103. CString   property;
     2 d5 `( D% f* P
104. 
     + M( o% \$ E* O2 H/ J
105. int   posStart   =   all.Find(startTag);
     2 D, X9 o/ @+ o
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ) O4 T( B$ w0 M
109. if   (posStart> =posEnd)   return   CString();
     . ^$ B! N/ s3 U
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     , w3 Q& ~  e/ e( o
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     + M0 `7 `7 L8 @$ l
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     ) S, _6 N+ q6 y1 K4 O# P
119. }
     ( {3 H+ y( [9 O: [. n% [
120. 
     4 i- b# q) W9 I% f4 H+ X" G- N
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
       i! j6 n3 Q) q* _
125. while(pos){
     * m) b% c/ V9 n. A0 P2 I
126. search   =   m_Mappings.GetNext(pos);
     ; |  {' O; Z2 W7 E1 R8 Z
127. RemoveNATPortMapping(search,   false);
     
128. }
     - v" c* v; R/ J, D, l! O9 b
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     4 @: f7 a& W  |: k8 H
132. 
     
133. 
     1 W0 V* Z8 d1 V$ I- U. P2 R
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     3 s% U; ]+ B1 e' H- D+ g7 Z4 l% ^
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     8 Y  l7 L! M: f1 |- m+ _
142. {UPNPPORTMAP0,   _T( "service ")},
     8 P* l/ o/ F: M# S9 R- L5 G, f
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     . _5 p) Z6 J3 ?1 ?
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     , v; X- |4 N$ V$ s$ R
148. ioctlsocket(s,   FIONBIO,   &lv);
     4 B7 _4 X: `. Z% c
149. 
     - A' N8 d- }0 m! e* |: x* P
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     9 j, ^7 b, C& p( X; k3 |$ ]
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);
     
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);
     $ b4 K- N- \$ Q! U  q
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     / W  O( d" r( A' ]* E
160. }
     9 d& L! B* L% p  f1 d9 w8 j+ h
161. 
     
162. Sleep(10);
     
163. 
     * Z1 m. Y7 ]3 @! c1 u- L
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     . _0 S, _, a% w( G* J' _
167. closesocket(s);
     - o, x# K- Z, c) f7 F4 G
168. 
     ) z7 @- I4 s# q! G  O. H  R
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + B3 g" J% V% N! B) b
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     % g% D/ J: j( Y! _/ p; e
172. 
     
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);
     ; K4 H) c) e4 \7 d+ `8 `' X. A
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ( e- K: C, s6 a# _5 E
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     5 D) Z- g( g0 h: e8 B
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     : d, N" A# E1 ?. v: |
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ' S( _7 Y1 G, i5 Q( g- Z
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     - E* G" P- P# N  q0 M9 |0 }
192. 
     
193. return   false;
     1 _( O  e7 [4 C7 `# v  _% o
194. }
     0 J. @5 K' w" m1 O4 {  `/ `/ o; [

复制代码

天云上人

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


///////////////////////////////////////////
7 v0 E; L2 |! M% O2 Q- D//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

4 s! ^* l9 f6 _* `* q+ ~
" q& _9 V  b/ i3 S#pragma once
9 A7 f+ a$ z9 Z2 E: e( w#include <exception>


  enum TRISTATE{
- G" i0 D1 y* F. \        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
4 s' l% e, [, d, s

$ L. H* X" I8 O* b2 t# Venum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

  m) Z- L1 m# b. z9 `7 ~" I& S
  ^6 {, u: Z& m0 e. @
. n9 V% [2 Z& I' s) V  p
class CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
8 q9 q. Q, L  l& c        enum {
                UPNP_OK,
3 F  Y* k8 N( E                UPNP_FAILED,
# t3 R* ?8 Q0 L+ Q0 _# r% `                UPNP_TIMEOUT
        };
& y. C, s2 j& i4 F! ~0 A

* P: J  j" {) Q! [2 _2 ^6 l7 a& a        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
7 m+ i9 }# C( F( r" A. F        virtual void        StopAsyncFind() = 0;
2 t2 M+ q: j" i3 O        virtual void        DeletePorts() = 0;
8 a& i9 B( q- F. K        virtual bool        IsReady() = 0;
6 t  x4 H! o9 y7 \+ P7 i+ F# v        virtual int                GetImplementationID() = 0;
       
) t: K& t$ k+ Y& r1 ?4 ^) H        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
  U% l4 H1 J' Y2 S

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
1 k; j2 B- N8 D5 c8 R9 g        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
- o4 Y+ _( |+ {) I- P4 P$ U3 R% D! u8 `

: _  e) B0 h+ a( |/ |  @! W. b// Implementation
9 O9 T+ F1 M: Y6 `/ fprotected:
$ T) h; X4 p3 S4 G! E        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
: o4 Z( K  G& x5 J$ p        uint16                                m_nUDPPort;
8 S+ g9 q+ ~7 D* S/ `        uint16                                m_nTCPPort;
  |" J4 }* H: {0 I" A- Q$ Z        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
. ^) G9 j% O( L+ }( o( u0 P2 S$ q/ l
5 @9 ]% {: C0 J: r4 X2 Y
private:
! A7 l1 i' p  E  x1 O        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
6 ~3 r# ?, q* ?, U2 h5 S, u
6 s/ a9 q7 V, I4 X4 N% E/ C& ]
};

  g# N* L0 K2 E
. u4 _* K" o, o- ?// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
' t( m8 p. j' [{
public:
7 ^( n: c/ R& @  \1 T4 _        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
& ]3 G' Z, n- ^% R; E        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
7 M. G  R8 _& Y$ W: [+ x7 ^* ]        virtual void        DeletePorts()                                                                                        { }
/ ?- j1 N0 K( P1 Y: s        virtual bool        IsReady()                                                                                                { return false; }
. P. J" _3 |2 L* b        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 @0 ^8 v  z/ E* d" K2 Q: M};
: X2 e* w: t* i5 S

/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
' G0 M; n  m/ b* t7 y4 t- l& k

#pragma once
#pragma warning( disable: 4355 )


7 D# ~4 s/ |% J! ]( g# X& M#include "UPnPImpl.h"
2 n/ L) J! |' R#include <upnp.h>
9 ?) _5 Q$ T0 u6 {& q; T; R! R! y& u#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
% u0 A+ V; |/ m5 I' E1 R

#include <vector>
8 x1 t: K  f6 s! I! g#include <exception>
#include <functional>
9 o* E2 [8 G. U/ u
5 \* E. l) n) P, @, a0 S

& g! t% I4 Z! I7 ?- n. U4 X
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
% y3 b5 i- t$ V" y, \2 ^typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
# N- E( M3 Z* m5 d( T& ]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;
8 _3 k: X: P! Ytypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

* m* l' R, J) I! P3 w
! k3 K% ]) }$ J0 ~. Y! y2 ttypedef DWORD (WINAPI* TGetBestInterface) (
- p3 u. \$ J2 Y6 |8 j: ^  IPAddr dwDestAddr,
5 O1 `4 U& `, A8 X  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
! A; h2 b9 u1 O  PMIB_IPADDRTABLE pIpAddrTable,
8 S+ Y, k9 }4 T. K& R  {, y  PULONG pdwSize,
- C3 ^. C2 o' q: P6 P$ M5 Y9 n  BOOL bOrder
2 s% M7 I6 E  `1 ^# c' S);

5 f/ X0 D6 \1 k9 ?
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
6 \- |  F' Y) ^4 o$ U, Z1 ~
1 J2 ]! j+ e! z, ?" b4 o. C+ L. j$ w6 k
! h( D, F8 d; H- e& a$ @CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
/ c0 e& ]  f- k( i  b& M8 ]& ?1 Q
+ ^5 e1 t2 k. w# q* H
" g' E! X' x0 H3 \: ~( E5 r, fclass CUPnPImplWinServ: public CUPnPImpl
' q1 e" [+ [9 W3 r' q5 \' }# Z{
        friend class CDeviceFinderCallback;
7 [* b- q3 q! X2 `5 ^5 A8 l        friend class CServiceCallback;
& p3 {( K. y; b* E- N// Construction
' @% }- i) J0 I6 Ipublic:
        virtual ~CUPnPImplWinServ();
8 B- K  E8 N4 s        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
& z. k2 M# Y- q4 y        virtual void        DeletePorts();
9 b$ q" D3 Y) u, U$ D" `: c        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
9 c- t) m: E6 z* \& _
$ u+ x* w$ e3 Y
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
1 a0 J# H+ R$ o, ~        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
; i" f7 D$ D, l, N9 O2 M6 E6 m        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
: `' R( _* x( B# Q        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
8 t! S: G- R: R6 U% I. ^6 \: f        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
; s( H3 d# W6 D; y3 l' A) e2 U7 }% X        void        RemoveDevice(CComBSTR bsUDN);
7 _1 B, \1 f7 W: @- s% H% o6 T7 s& W+ K        bool        OnSearchComplete();
$ s5 l; y- q' \        void        Init();


/ b2 v$ R8 G  q        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
# A* D8 \1 P- |' L                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
. y2 w& }3 S% k7 h                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
, _% m% c0 q' ]+ X7 t) n* j                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
0 i2 j; r' R- C4 J        }


; {) [" C1 T3 q  w. J( ]% e        TRISTATE                        m_bUPnPDeviceConnected;
2 [5 Z0 p# N+ Q5 J7 r; \

// Implementation
. {  R2 r6 k8 q1 ^3 `8 i& N        // API functions
' P7 t5 U* h/ v! F6 t: J$ ~        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
# B, H4 A) N9 P  t1 R        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 X( W; \/ G6 T5 E6 d, r1 Y  t3 c        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
* O; d; W% j/ h) u        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
7 ^& ?0 T5 I. L+ F' `
' u. I* L9 O; E& _& y. a
- o# j( ^( H  a" H        TGetBestInterface                m_pfGetBestInterface;
  w) k) U3 ~( W( L        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
# b' B3 e6 H5 \2 q

( [( }) M' K" ?% S4 M  }        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
( c+ p/ N' A- W7 B6 D                result_type operator()(argument_type device) const
                {
. D) k/ T) d8 N7 R  h                        CComBSTR deviceName;
9 z* [0 d; M9 y0 ?8 [                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
7 y# o* x9 D) V- p6 l

+ ^$ V. `1 t$ V! M+ p. T9 x( I! D3 H                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
  @/ ?" \! d' k" L6 J
9 o+ i+ l( E8 S7 [
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% J: [8 w: n  A0 i6 T7 K- C' m                }
0 _+ C% i& M  d) F- n                CComBSTR m_udn;
        };
$ }% z5 ?3 v9 k. X5 O" Y! e       
+ j' q" W+ N9 C! k( L* ]        void        ProcessAsyncFind(CComBSTR bsSearchType);
; t# r8 E6 @8 }        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
1 z( R: E% Z1 X9 }        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
: a" E6 E3 O* T2 S, u) P        void        StopUPnPService();


        // Utility functions
  M5 y  q4 A) t9 e        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
1 \0 K/ H7 N' v9 k& R# _        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
" U3 _4 U; `  t4 E" ~( {/ Q5 r        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ z" F) m2 ]8 b; H5 D4 C8 X        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
( ^1 m/ q2 b2 ]  e        CString        GetLocalRoutableIP(ServicePointer pService);


! a" E% j, C1 S4 F6 O( {// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
. L. H, Y" r+ b% a  B4 a$ S+ \        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
- O' h+ T! l! R6 p- H- L        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
0 h/ r- X+ Z; V1 I: Z
, c% s2 Q8 J1 J4 \' O5 ~+ R$ O9 Q
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
- l' K3 [7 g- d9 |+ g        bool        m_bPortIsFree;
) l6 s( _0 p, n1 ^        CString m_sLocalIP;
) J# ^1 `* K1 C% v        CString m_sExternalIP;
( J4 Q- P- Y$ s( \1 E        bool        m_bADSL;                // Is the device ADSL?
- E) e1 S* O1 e5 M: T, b/ b1 A        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
1 F( R; ]0 F5 a& Z9 b* o        bool        m_bInited;
        bool        m_bAsyncFindRunning;
. Z+ M! Z; L/ B  k        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
% O' i0 I$ @* r) L4 `) P, a        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
1 \' }3 `4 O0 }+ B2 }& T1 t        bool        m_bDisableWANPPPSetup;

6 [; F' Y. k+ ?
};

' U' m* J5 y8 h. n9 ^6 ~
! u& n! v0 ~. E8 o  A* N// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
) f$ t: ^" w" l3 U7 O0 Y{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
2 A" u- M5 A) \5 s0 p) r        { m_lRefCount = 0; }
) {: p$ z& P' `9 @

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ r8 ^$ D! r6 m8 L, a" ~   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
6 T% P/ N" E; {& dprivate:
) d. d$ n" K6 J" _9 h& M        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
' p6 n( Z0 j0 U# Q, d  J5 _9 m        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
$ w# ^) @* S4 ^8 z
7 J4 j, o* K* S. d' R8 u
private:
        CUPnPImplWinServ& m_instance;
1 i% {1 ^) z) u2 g! m        LONG m_lRefCount;
" a! _- a8 J' G1 u7 p& L% u};
& `& h$ }2 ~2 I" b, c  f

# R6 i! v7 v( M' C+ K$ d// Service Callback
class CServiceCallback
  r+ e) N( j1 S9 ^        : public IUPnPServiceCallback
{
public:
8 ^& j5 k* U8 M' z6 Z        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
; `9 ]0 U( H8 B+ S) ]. z( \9 p4 n' M+ k   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! |7 a% ]1 a: A) p0 z: G   STDMETHODIMP_(ULONG) AddRef();
1 W' N- `. R( g; Y5 Q! O% H' l5 d   STDMETHODIMP_(ULONG) Release();

3 W2 H3 P! B3 ^6 _, \( J% \
2 C0 e# D0 W+ O  t$ T// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
3 K) [0 Y- b! S; G. [1 p        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
7 @5 \; p9 m, g6 o+ G
, l: T9 H, r9 R  Q9 o
  \6 @: x: a1 A$ `/ Jprivate:
& G* _+ x# w" F( _* ]. M        CUPnPImplWinServ& m_instance;
1 W0 W) Y% m# G2 P        LONG m_lRefCount;
};
' O7 p9 n. Z/ r. Y" v
, v! b- E/ L; F: c/ a
/////////////////////////////////////////////////
# G1 _! T1 e# m1 C( W

使用时只需要使用抽象类的接口。
/ \4 D0 u4 K; ]4 J, SCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( E% T% J2 e2 D. }4 B$ T% ^CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
. y! l- B' l; P. o- A2 O. [4 \; UCUPnPImpl::DeletePorts删除端口映射.