UPnP

天云上人

/*uPnP.h*/

1. 
   , P0 D% \  M1 ]& K4 Y& h! j
2. #ifndef   MYUPNP_H_
   
3. 
   & P- b( P2 @: t1 c  r
4. #pragma   once
   2 t* J* D$ X$ j% I- C
5. 
   # K" k  G" b. _
6. typedef   unsigned   long   ulong;
   0 P, }5 N8 s) _  S5 ]* Q- [% C
7. 
   7 y' N) G3 Q% |/ f
8. class   MyUPnP
   ) v* |. U) n; h+ C6 j; S: d$ `
9. {
     q* D& S1 q6 K
10. public:
    / k, r; y. v. y! i& f6 p. C" e$ B
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    4 w0 T+ a5 ?" `% A
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 e  k, v" K$ ]0 }8 _- n
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. 
    5 _& I3 ~0 Z; W3 ~# I
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    7 T" s: p; u5 h
22. }   UPNPNAT_PROTOCOL;
    ( i0 l' \' j  z* x1 o
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    # t+ H' A& Q/ K6 V9 |9 t* V( |
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    / N0 |* c2 x6 w  @& N
31. MyUPnP();
    " c( D: w4 I% ?% v1 q
32. ~MyUPnP();
    / a* P2 M2 e- F/ s' Q3 }, t0 \
33. 
    / ~! y3 ?- I; h
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    + Q" |2 ~( K3 _8 y/ O, O
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    + v/ t; g* f5 F+ e. e
42. 
    
43. protected:
    % h( _9 q" j7 h! w& `; Z. J  w
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    3 D3 l( ?4 ]( Q( K
46. 
    * ~: ]/ M- I- S0 f' D
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. 
    9 z5 n. t$ K( T% m0 h
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ! F" l, t# H2 L1 d; k
53. bool Search(int   version=1);
    6 T; o$ P( r% j0 s6 Y
54. bool GetDescription();
    8 f+ X5 ]- _( {1 V, F5 Z( O7 h2 J* u4 N
55. CString GetProperty(const   CString&   name,   CString&   response);
    / F1 _3 ?6 i) P% H9 D
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    5 Z4 A6 }" a) U' W) z- w: r
57. 
    ( R1 I( c8 L; M3 o* z6 f
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    3 U! w# x5 d; d0 G2 T- e* o1 K. _
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    ( W/ w) D9 T1 g6 |6 k8 c
63. CString m_baseurl;
    . K5 |, r% B) \3 Z# ^0 C; u
64. CString m_controlurl;
    $ P4 B, p1 r! W- l: W3 l3 i
65. CString m_friendlyname;
    ; a$ A3 ~5 X0 E+ t% P& y' T! O
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ) O# ]& M0 M. ^
71. 
    % Y! q4 r6 M/ g' l0 w
72. CString m_slocalIP;
    6 g+ A, o* n: d  x9 r" J
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    3 {" o) [$ f+ k7 ]
75. 
      _7 c# ?6 u8 ?5 H) O) Y( S
76. bool isSearched;
    
77. };
    * @4 M+ {" C- V
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   8 z2 C* j7 x& _. B
2. #include   "stdafx.h "
   : [1 @( g3 n1 \& O
3. 
   ! M3 C' N, R. W5 w
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ! [7 T* S- h1 c  Y8 \( O
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   9 x$ @+ b2 _! m' E* M5 H3 B, V
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    8 w. h  l; u7 E& t
11. 
    . [6 _' o3 I6 V
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    0 h8 Q5 e  i* ~" K% L* N
15. 
    
16. const   CString   getString(int   i)
    & k2 b8 ]* a' ?$ e/ F9 p. m
17. {
    ; L6 K8 U3 ^4 Y; e: t; T- q
18. CString   s;
    # E2 D, O- L! @% {1 @! W
19. 
    
20. s.Format(_T( "%d "),   i);
    ; p8 |7 A& E0 }) Q0 }$ Z
21. 
    
22. return   s;
    
23. }
    & O* c+ g; j8 m* [6 }9 \! U0 A
24. 
    3 N5 M" b, e7 F, z- g! b
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    $ p0 |( e/ w6 {8 G& \% I7 ?2 ~( v
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ! {5 I% N1 _3 M8 ~
28. }
    & J0 ?' H0 L( }8 @1 V( |
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    3 Z' \" B, T, j# W4 _- d7 v3 l
31. {
    4 |1 W/ R1 {- H: \, c, p) e0 I" z
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    8 P, Y  i! _2 |( Z% b9 F- P
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    4 W$ ]+ {1 k  C" D7 K2 F# t  s
36. {
    
37. char   buffer[10240];
    
38. 
    " r$ K/ j$ v0 p0 B
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ) ^& K6 z* h! x4 v* |/ ^
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ; R/ ~  W+ {; F3 C
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    9 u/ [0 E4 V& }. p5 b' }
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    9 n: V* c: W3 h+ P
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    # p& {& ~9 D* W2 b; Z
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    * L4 Y1 F% U# }6 J& Q* H
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    7 H8 u. W+ E! W
68. char   buffer[10240];
    8 R& O1 j8 D5 ^+ }' ^" U
69. 
    
70. const   CStringA   sa(request);
    ; N2 K1 Y. ?; I+ w2 b
71. int   length   =   sa.GetLength();
    9 f& q, d+ g0 f, |" @, l( S6 Q+ X! w+ \8 Q
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    - ]* k) ^! B1 j7 r
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    0 e7 d: O- D5 \
76. sockaddr.sin_family   =   AF_INET;
    $ G6 B1 O4 ^: F1 i
77. sockaddr.sin_port   =   htons(port);
    $ u  ?% B* m: e4 ?2 _$ I
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    1 x5 ?( F/ i7 z2 |& T, S8 S2 E% M
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    & H( o7 H0 `# d; n/ Z# `0 t
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    6 j& V8 H3 F, Y$ D6 l. F2 a/ A6 l
90. result.Delete(0,   pos);
    
91. 
    . V9 N- q' k; D" |* C
92. pos   =   0;
    ! G, y1 u. Y. p  v
93. status.Tokenize(_T( "   "),   pos);
    ! o4 s; m' N2 |4 Y2 Y+ k3 Y$ O: l
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    4 W0 `, F/ l+ q( w" B
97. }
    9 @$ Q1 y7 A# L# b1 g0 r$ }1 W/ m
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     7 H8 X$ u1 f/ |# x9 X
103. CString   property;
     * W: ?8 b9 C- I7 ]( U
104. 
     
105. int   posStart   =   all.Find(startTag);
     8 g' A" @/ I$ u# k
106. if   (posStart <0)   return   CString();
     7 M2 W7 A! [  a& y2 A
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ) @* b4 }; P+ {/ v
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     8 r; i( r' n+ Q( ?
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     0 Y0 L+ P4 q9 i1 V
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     $ s& f0 v. C" O
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
       |0 U* \3 [! c* L' ^
122. {
     : @" i* g7 A  q' C: `
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ! U5 M* m0 T! V# a1 y2 {/ g: B& W+ x
125. while(pos){
     / \, v# I" m2 l" x  k6 d
126. search   =   m_Mappings.GetNext(pos);
     6 c7 U: U7 Q) K* Y' p9 R
127. RemoveNATPortMapping(search,   false);
     , Q" p4 l' `; h" P8 a  a7 J% Q
128. }
     
129. 
     5 o* t. R7 D& D6 y* d
130. m_Mappings.RemoveAll();
     
131. }
     , U. `! Y; K7 ?' m/ `$ @
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     4 }; q. S  {  i3 r" |9 A
139. #define   NUMBEROFDEVICES 2
     4 T! G3 P' W8 ~
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     0 E9 P8 a7 E+ f7 r3 p) D
142. {UPNPPORTMAP0,   _T( "service ")},
     ' M; H' L) i' r+ u! s1 b* |# p
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     7 A/ W2 U% h- b- [7 Y! s' Q7 I
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     + U# K5 v' y4 q& r2 p+ z
147. u_long   lv   =   1;
     6 I9 J% y+ p* u  j8 I; k' j
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     , [8 ?- i3 \! @6 |
150. int   rlen   =   0;
     : B0 k( X' q; x3 V# Q, c; V
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     - F: \8 `" |9 T: i+ G, O8 Q
152. if   (!(i%100))   {
     9 M, u% N  v1 g
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     : C, A# ]7 ]7 D6 S
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     . O/ N# _3 \& q, ]
155. CString   request;
     ; \. J, m' o0 ~
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 "),
     5 N+ S3 g. S& B0 n0 O6 x
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     * a& y0 ~- e8 v% I. h) r
159. }
       D9 \1 J6 ], z; F0 Q' C+ @2 ?* ^; J
160. }
     
161. 
     
162. Sleep(10);
     * o8 q) o8 k; Z6 f6 {. g
163. 
     3 s4 v; K0 H2 |- [, L
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     + m) M- z1 B: b4 C' @
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ) J) m; @* C$ u+ t8 x- e: @( o
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ( x. a5 Y( v2 I
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);
     : ]$ j: w1 y1 V& C+ M: [  ~' @, C! l
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);
     & b# _5 r- p" q6 Q# t& A% W8 X5 t. B
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     4 e( U0 |0 X: D" A! _" i! a
183. m_description   =   line;
     % |7 d9 A# l: u( }
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     % {$ c: o: q3 O% u, O- L
187. }
     
188. }
     3 S- W. w" W! `
189. }
     
190. }
     . y- Z) T  N- W/ a. u
191. closesocket(s);
     , I, w4 T4 l0 ?& ~, s0 j  \
192. 
     6 g% g4 i  n6 ?7 v0 M0 i' q
193. return   false;
     ! C6 w+ ^! |8 j/ u3 [2 x0 Z) v
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
1 v3 P. D2 D- v; N  K
  _. s2 R2 O% n( i. W" \9 ~
///////////////////////////////////////////
" n8 n/ W( y# _9 W; d1 i//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


% \* {4 W/ C6 L' y: C#pragma once
0 p% e& `6 l# a' K) @#include <exception>


, d+ \% K7 X% i$ p! L+ E) g  enum TRISTATE{
        TRIS_FALSE,
% h+ c! y: U' F$ F/ `2 P, m        TRIS_UNKNOWN,
        TRIS_TRUE
};

& J& F# m) g: V0 [
0 c( y* N4 _: Q8 y, `# qenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
- A0 H5 L% j6 J& i* \0 f        UPNP_IMPL_NONE /*last*/
; L9 Z$ A/ s' e6 e! T};

5 d/ ]# F5 H: U  b& [! u

3 E, ~1 U( q2 g+ F' V/ t1 V
5 ]% n. w3 T  t- |# r$ sclass CUPnPImpl
& E  O5 \5 L8 y! v& _{
public:
        CUPnPImpl();
+ @6 K$ Z8 v% Z5 H7 J* e& E$ U        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
4 Y% B6 s0 ~! g9 Z        enum {
+ V3 y3 g. M% ]- m                UPNP_OK,
                UPNP_FAILED,
$ {7 ?0 l; L5 G) f  }9 Z                UPNP_TIMEOUT
        };
8 D- j+ C4 V! e7 e5 M7 @. f3 K

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ @, ?$ r) C$ `! l% a/ R        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
7 g7 j5 v& v9 U7 q        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
' L! L+ Q7 i+ Q7 h
, t* U+ J- o8 |# b$ R+ e
, U  o+ x+ E1 \) L        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
0 \$ I7 m, y3 j% i* S3 J        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
( O8 }" d0 t7 |6 ^1 K1 Uprotected:
( [% j1 S! {  v7 D; S+ S% V2 s3 M" D+ b- @        volatile TRISTATE        m_bUPnPPortsForwarded;
/ w0 g. f1 {) u3 `( \        void                                SendResultMessage();
* k* `/ F' U2 w" k        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
6 e# {3 W' B0 ~" _3 h! I        uint16                                m_nTCPWebPort;
. i5 E' {( J) p( ?# O        bool                                m_bCheckAndRefresh;
: Q0 H( @3 P- n
/ A( b8 W) I4 b0 j- Y- z" c" A) i
0 S( S$ E9 c5 T/ ~- n6 {private:
        HWND        m_hResultMessageWindow;
" M+ ^0 @9 g0 `        UINT        m_nResultMessageID;
4 |/ x3 R- i  n+ k, T
, o3 V. g- @# Z' v7 e) |
};
+ y5 S; ^( D' M0 ^- O
% \3 D- z; L& l2 J
2 o3 q& [) M/ o# z/ k2 Q// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
: O; k8 R+ P0 Lpublic:
( F! J% u0 K6 H, b, F# |- ^' G        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
8 L; T6 I% c# |        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
- _- G- {- L, d3 f        virtual bool        IsReady()                                                                                                { return false; }
9 _$ j" Q) u% B* h$ \6 Z2 R        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
& S& o8 g) j, i};
5 [2 d) ~+ v0 ~2 _. n/ y
. H% @5 g* I# s& e0 g) B. \
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
( L% ^  o+ V" T: f( E, f. l
# I  M0 q" D  i  L$ Z1 L
8 n! A. }: H& q+ m  }5 Y#pragma once
#pragma warning( disable: 4355 )
: ~, Q8 a! a9 P7 a) ~. H9 i
+ K5 \% c; Y- n# O! _
#include "UPnPImpl.h"
#include <upnp.h>
4 ~; S- U9 @& P4 d4 B, e' D9 I#include <iphlpapi.h>
#include <comdef.h>
" g  B! T) G0 e6 \. _& O# b#include <winsvc.h>


$ v" Z: c+ D; M' @' d. X#include <vector>
#include <exception>
#include <functional>


8 ?: E* Z8 {; e" ~0 p

2 T: f- g* V) v6 G7 Z# w1 Ntypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
) x* z: @' Z6 [- K3 n8 htypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
0 h# z- Q- u5 e; X0 Ktypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
/ Y. N( D% |" atypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
" O  k0 \! h4 m6 c9 T8 z2 @8 D8 Q- stypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

& S- r/ M, ^- p3 `
: M0 T' r) V. R' A9 itypedef DWORD (WINAPI* TGetBestInterface) (
: N( o/ g: F# x) i  C8 x  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
( h& {: E8 T2 @! f);
) B3 c  z' f: E4 o3 u

9 |8 e1 ^5 D9 n  ?- c  W3 X3 ktypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
& @- @% P1 }4 N0 U; ]) G  PULONG pdwSize,
  BOOL bOrder
);

+ R; O# U7 E( u# Y5 Z+ E
' P0 U/ Q' l2 X# A1 ctypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
, l: k* J. _) L) y9 G' {, _2 d8 }
- t6 T/ l$ @' d/ v0 i+ w( x
1 L5 a% ~4 W+ L8 dCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
& K2 H3 R. y/ Y" X; @3 v" T. @
$ s# w. ?0 d, G' k* D" y
class CUPnPImplWinServ: public CUPnPImpl
{
6 j  [, J* n# s; \' O, g* l        friend class CDeviceFinderCallback;
8 U8 }: q4 o, x& N2 A$ o        friend class CServiceCallback;
// Construction
public:
3 S$ L" m  y; U$ w        virtual ~CUPnPImplWinServ();
6 h) n! [! y6 E- ?( u        CUPnPImplWinServ();
2 z5 M  b  Q- K7 Y, [  A4 B1 b
& L! h5 {3 c8 S$ |" s+ J
) a; n# a4 @7 O. h        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
9 d/ L4 A/ Q$ C, H) ]! e6 [        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
$ @( V% L0 J2 x4 T- o5 J" @
, Q9 f9 a' X7 {0 W. m8 G
/ {$ U3 r* A/ c- j( p, p$ B6 \        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
, @7 a( ?$ T, M! ]        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
8 \8 u1 R) M3 v- q% ?) T* i! P        virtual bool        CheckAndRefresh()                                                                                { return false; };

, _* G  |( `/ P
protected:
* c4 B- q! n9 ?9 K: T8 e        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
; t( k9 i" O: i; w* O# [        void        RemoveDevice(CComBSTR bsUDN);
) p  @3 w! ~# m6 O4 r! _- D% U        bool        OnSearchComplete();
        void        Init();


% ]6 t3 {+ ]! P+ _; l9 ]        inline bool IsAsyncFindRunning()
! M" N- ~! J1 X( s6 F2 }$ w        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
2 c) q' P8 S8 M+ M& b. \) b                        m_bAsyncFindRunning = false;
                }
                MSG msg;
7 [" d" l5 E1 I* F/ ^' J                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
, x+ U4 B+ v& J$ u5 W; E$ Z- H+ f                        DispatchMessage( &msg );
                }
" z3 ?, e& H# n3 f$ Z; C4 t. t                return m_bAsyncFindRunning;
1 H! I9 f9 U  f- F1 r# {        }
9 B4 N) a2 N! B' \2 d" [5 l

. \2 C0 B7 p& i8 Y: D        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
  d/ L8 c$ N& f6 i; N        // API functions
$ x" r- R* J  u        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);
6 k8 _& _# L; g) s7 j- D9 G, `        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
4 S1 ^& [, S' ?        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
/ e4 J) }" `1 R& M. J        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 v" Q- a+ W( O0 N

        TGetBestInterface                m_pfGetBestInterface;
1 {2 s: x% T# s% \5 }% a! s! j, o/ f        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
5 I3 k8 L; x# s& K$ f3 O8 l
; f& m- ]+ C. f8 q% o7 H
# h% q9 ]2 ~" s) C' {- y        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
  h  W1 t# I8 ]: p+ m! A, \                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
* \( x8 G/ D. V- J* R: [) D& ^4 \; s: H                        CComBSTR deviceName;
# l( Q8 L4 t2 L$ q                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
8 @7 p$ |! U3 L# a$ `8 r

                        if ( FAILED( hr ) )
) U( E9 {8 g( p. e/ l" q  v2 X- @0 F                                return UPnPMessage( hr ), false;
# H3 l, f" r" P  _# F8 q& y0 A
6 X5 d4 X4 }- q0 j
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
. Q  b* Y5 C) q. _. _+ L& Y( S+ I$ z                }
. u! Y# _* p2 h6 q5 E2 m$ w$ U                CComBSTR m_udn;
        };
# C/ l. h3 ?0 N7 F* b2 e+ k' ~9 r       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
% F* B' W* n; |2 D- i. ~9 [+ c# _9 n        HRESULT        GetDeviceServices(DevicePointer pDevice);
% K0 n- P: R- Y: B/ ^9 E5 @- K5 h        void        StartPortMapping();
0 {$ R7 }- V! x0 m+ ^        HRESULT        MapPort(const ServicePointer& service);
/ ?7 M/ C; P5 y% C2 _        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
* L' Z4 C% a# N$ ^        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
4 U. f: w+ m* ^; h        void        StopUPnPService();

( G) M' Q( W- x& f/ U; t7 x( g# Z5 ^
0 r, v& i% X% }8 }5 G/ a        // Utility functions
3 R* f) S7 y" R" N        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ D/ V# A9 ^4 F- W  p- P        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
, I6 R& f) R& o4 A        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
0 N$ O* F% F6 Q# ], L        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
, p( W& Y7 z% x  X/ S        CString        GetLocalRoutableIP(ServicePointer pService);

* i" \% Z  h3 n4 X5 A' p7 k
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
+ `) B# A0 S$ {2 K) y" f        std::vector< DevicePointer >  m_pDevices;
  y# J2 ?/ r' H' V& G        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
+ ]9 u/ ?7 i5 ~: s' A8 U) ?        DeviceFinderCallback        m_pDeviceFinderCallback;
4 k& t* X! t- t; d        ServiceCallback                        m_pServiceCallback;
5 j1 G. [' E$ @3 _) i1 G
& x/ i9 g1 i& @
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
0 }. A$ a  b8 @/ G7 u2 r: @6 B        bool        m_bADSL;                // Is the device ADSL?
6 ?2 Y2 k! U! W5 ]' b, P5 |6 a/ V6 L        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
7 o- \$ ], F$ y( C5 u% ~4 t' D        bool        m_bInited;
4 ^* x7 i/ e4 H# I& Y2 c        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
- K/ ^7 {8 C0 R1 C8 ?        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
. O# t- @- R; l" E' |+ C        bool        m_bDisableWANPPPSetup;


2 v/ A/ J1 Z% x1 t- f' v: J- s};

- V2 M) ^: S' J# J; i! B/ q& @1 l
// DeviceFinder Callback
: q& O. d. ^" ^% X3 i% A8 I: ~class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
& f& q( @9 P% j( s9 V6 @* n& |public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
3 R2 J1 D6 g9 m% e7 E1 W                : m_instance( instance )
        { m_lRefCount = 0; }

$ z% D- [1 [+ j
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 V9 N# m9 U) o   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
/ k$ g7 @/ k( `2 g. O. z, `6 zprivate:
$ S( d) I8 w6 z7 [  y        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
5 v- Q! M6 ^5 J) ~        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
5 o9 v0 q" D- ~- p" `- G( m        HRESULT __stdcall SearchComplete(LONG nFindData);

$ T( b, r; d  s& l3 [! i' z$ g" Q9 n
private:
        CUPnPImplWinServ& m_instance;
3 R8 j  `: M5 k1 ^  y& j        LONG m_lRefCount;
1 ~4 r, ^+ Y# b- X. f: e! U};
; [3 @4 l4 C3 C; x& e- i! r

2 O! h0 L+ h/ Y/ |8 X; j// Service Callback
7 ?; Z; g( T* ^$ e5 Rclass CServiceCallback
        : public IUPnPServiceCallback
  E, U$ R5 `, D{
public:
1 k; O/ n( f& p9 u7 y: b        CServiceCallback(CUPnPImplWinServ& instance)
- I: L$ ~8 |& c& \3 ]6 f                : m_instance( instance )
; m6 m2 D8 c. N: p: l5 e        { m_lRefCount = 0; }
: k" o! q8 Q# H: U; N   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 r6 o% @: E5 R   STDMETHODIMP_(ULONG) AddRef();
+ c% n. P% b7 R4 H; O9 J3 P   STDMETHODIMP_(ULONG) Release();
& B# k3 N4 T& {5 z
+ C/ {: d" _6 z
// implementation
private:
0 P  D7 _; B/ }. u        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
' x! R2 W. R- ]$ B- O
- ^& z! m  Y/ O0 Z7 K# a1 l( C* m
private:
4 k( K" {. t: t. P# K8 m        CUPnPImplWinServ& m_instance;
9 }2 D+ ~, g! {  i! }8 P2 c        LONG m_lRefCount;
" X8 p# t( p; L# `};
( I7 L0 `3 D! P

/////////////////////////////////////////////////

! d0 k: v' _3 U5 e  L
使用时只需要使用抽象类的接口。
4 T0 M+ @- \- ^; [; t5 ICUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
! A+ g; R8 u( O, tCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
( G* T2 _3 @5 w: tCUPnPImpl::StopAsyncFind停止设备查找.
8 ]1 @2 |6 h0 M4 f) ?CUPnPImpl::DeletePorts删除端口映射.