UPnP

天云上人

/*uPnP.h*/

1. 
   / l8 ]* P, B3 W4 a# L6 y# n& X
2. #ifndef   MYUPNP_H_
   
3. 
   6 J% p+ m- a& A) l; V- m
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   " Q3 o. V' X, V% j
8. class   MyUPnP
   # e( x% S) ~  G( W# Z+ e. p: j' K
9. {
   
10. public:
    
11. typedef   enum{
    . M: W8 h; l# N9 H/ k
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( G3 p6 V0 o! u
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
    ) C* n" V1 X3 G& D- ~) O
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    $ E0 d! q- l- M* T
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    7 J. |' i2 X  g& S% ^. `/ ?" Y8 [5 @
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    $ @" {" ^) [' \2 r
29. }   UPNPNAT_MAPPING;
    - c" X$ G0 [) u0 i
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    + j4 Y$ O$ T' |
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    : c7 _1 @# W  x3 R9 Y/ ]
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 {6 Y; ]) l: `, Z: }" w
36. void   clearNATPortMapping();
    5 m' B# f, _7 |7 O& B: Y" E1 c
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    % |5 m& x" a- F6 A
41. bool IsLANIP(WORD   nIP);
    0 _# o- T4 O: Q3 W: ?% l
42. 
    
43. protected:
    3 \2 z3 r5 w/ D1 b" H- z( n# B& W
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    9 V6 v6 \+ y( d/ i9 t$ Y4 ]
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    + }! S$ I& ]) y4 p8 K# ~
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    " p& j& ?  N3 U8 @
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    # x! d. @2 r' @9 E$ S7 s2 h
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    + J5 h2 J# z0 T
59. bool InternalSearch(int   version);
    9 z% a' a1 ^1 l* Z8 d
60. CString m_devicename;
    ) a2 n% \& M# z6 A
61. CString m_name;
    ' J: ]. B' c3 Y2 [9 C4 Q4 S
62. CString m_description;
    ! [/ ~6 @. y# Q; z
63. CString m_baseurl;
      B  S7 u: O8 x
64. CString m_controlurl;
    
65. CString m_friendlyname;
    2 L% X" m+ w. T
66. CString m_modelname;
    " X$ s& f& n) I' D3 H! i% t# d) Z
67. int m_version;
    
68. 
    . n: A& r9 ^; D5 J# A
69. private:
    & ?& j- ^; D7 h, j) _' r; i
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ) q# o6 t1 Q( z' `( P
71. 
      X3 w8 ]/ Z& B1 n9 c
72. CString m_slocalIP;
    
73. CString m_slastError;
    - B0 t7 d. J8 d1 v, H' L
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    " g1 t5 e# q* `+ ]. S
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ) [, {4 m2 w% [' R
4. #include   "upnp.h "
   : O0 n! ?9 e! I; _: W+ x
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   * m9 R4 ^, _& O9 l
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   3 A: S; e- H: y0 ^
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ; ?: n" |2 |, V/ s' L
13. static   const   int UPNPPORT   =   1900;
    8 k# t  k; S0 x3 {/ s9 Z
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    1 w% j, _: Y* W: p7 a. ^
15. 
    
16. const   CString   getString(int   i)
    $ m$ q, I: W) x, q0 H9 h
17. {
    2 }" U5 L! T0 C- u
18. CString   s;
    * d5 ~1 |: z2 i
19. 
    . J  ?4 Z4 Z# u9 G, ~
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    ; x5 u4 A3 K+ M
23. }
    5 o- t$ ?; m* b2 h
24. 
      Q- j* ~" P. s: f2 x. e5 Q
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ( C6 q4 B) H7 Q, K# Y
28. }
    , L. r& r4 {% G
29. 
    2 \$ d6 T( h; `; Q
30. const   CString   GetArgString(const   CString&   name,   int   value)
    9 _2 d8 U$ S) P) M7 h% K; ~
31. {
    1 }0 w- J; W9 }. R
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    : H* ~. L" z& I& j8 @- Y
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    & ~! Z  |) R9 k% T; z
37. char   buffer[10240];
    9 _" w2 h  P  v, t
38. 
    $ n5 e. D, D( N- z. E  x" ^
39. const   CStringA   sa(request);
    * X( K8 v, P9 D/ r- o1 q- q, o
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    7 R! V* U5 u! S, s/ V4 v! a4 {! e
42. 
    1 |0 N- O4 ?% F7 v- z/ |# w0 B
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;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    # o# {4 t( G2 d4 W6 ~$ \1 Y& d
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 o  o  r  f$ k/ E
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ! }8 |. ?* ]0 }9 H; \, a. Y$ ]
55. Sleep(100);
    $ F; ]8 [6 j9 ?& [# N7 q
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    . W: k  H" ~- l  G4 r! ~
57. closesocket(s);
    2 N9 c& K8 o& o
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
      G8 J" V0 h& l% i
59. if   (!rlen)   return   false;
    
60. 
    9 f! ]- y! X: R8 l1 m+ C
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    " `( \' I. y: I7 M
63. return   true;
    & T0 N/ T& A8 I1 P: }& a/ C, P
64. }
    
65. 
    6 _  Y2 H! f! v' I% Q, M& Y$ ]2 \
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & F* x  [% [1 W" A1 _# `3 ?
67. {
    % U! g6 f, o- F$ p: R
68. char   buffer[10240];
    
69. 
    ) z+ A  {* f, {( K. D9 j
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    . [9 \9 f# c5 F1 F# N" F( [
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    # L: l; o. o# F! O! l8 [
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    : c6 K) f8 k: D
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    6 ^- i& n: q2 ~8 {" n5 Y. w
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + ]3 C3 p: z3 S  J5 A6 w
81. }
    ' q$ `& ]) Q# u" v0 P! N# @. p; @
82. 
    : ^* |" n8 Q$ p5 g- k4 t% `
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    * b6 z  U" N* w  f% R. U
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    4 K& H% k9 v4 f1 R2 L
88. 
      @; O. ~) g9 G9 J2 B) c" h9 i+ H
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    5 t' r" s7 K& Z
94. status   =   status.Tokenize(_T( "   "),   pos);
    + x: Y7 D, M* ?% V1 Y4 I- v2 X5 N
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    / P3 J7 \& ^& {2 w  \
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ; T: m6 E5 j8 S: g, i0 g! }
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     - ?- B' x& p* V! y; |3 M; E0 l
104. 
     - `3 R' t0 Q" Q  r
105. int   posStart   =   all.Find(startTag);
     4 R+ O, P; ?7 Z# ^4 R& W
106. if   (posStart <0)   return   CString();
     " c( x+ Q2 s1 v% a6 ?6 P3 D
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     : g0 d! ?* X3 X' w
109. if   (posStart> =posEnd)   return   CString();
     ' z( _0 V& s# N. U9 {, o9 b( l
110. 
     + y7 a5 x: ?, s$ |- d
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     5 [; Q- a4 G9 I. Z/ s; m4 [) t
114. MyUPnP::MyUPnP()
     / z/ g' Z; M& o
115. :   m_version(1)
     ' d9 N. X% e) x- S+ j
116. {
     7 \; b% O1 Z$ ]; J0 Y
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     ' C* H- z' M/ C$ k, O; V# G
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     8 y8 C7 J. N4 J% y% A! b" U
133. 
     * H/ f* P$ N$ }4 `
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     8 B! u  d+ j8 C0 E$ {
136. if(version <=0)version   =   1;
     3 A$ H5 k4 ]9 z/ }9 c+ D6 w
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     - h) _: B  U+ j. F" d! g: c  O
141. {UPNPPORTMAP1,   _T( "service ")},
     ; {+ f' F' \2 T! W1 z$ D5 C
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     % d0 A' h3 e- u# k
144. };
     2 k% g" O9 _- b& o# o* C9 ?6 t
145. 
     & _4 Y$ X- f4 L8 ^1 P: A5 z+ a: j
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     0 ~1 ~  T, i- U( a1 d
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     4 i8 ~% A: \8 s0 K/ V
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     0 X; K: ?( y( R7 f9 j' v1 n  _1 Y; T" \
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     3 K& x  ~# w3 \
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     % p( H% {- C2 o9 `, g3 @4 ^
162. Sleep(10);
     
163. 
     ' g! ^1 A8 e! x1 r
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     6 U& d* c7 e7 w- ^  `7 Y
167. closesocket(s);
     ' j( z) t* X' \$ ]* L; \% {
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; x( @& [# r) V9 a) ^0 _1 N
170. CString   result;
     0 y# Y2 R( l- X! \$ B: z
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ( X. y& \8 V  D
172. 
     7 u0 G5 E9 D- ^1 }
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);
     # W# E: o/ Q' C- q, X+ l/ p
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     " r, O' U1 v( ~4 R+ P
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ) Z. M9 G, X' g" ]1 A
180. name.MakeUpper();
     ) m1 a/ C  l9 \7 [- L1 {  o  L
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     4 k2 D, a6 X$ S* n/ W+ A! V
183. m_description   =   line;
     
184. m_description.Trim();
     $ Y6 Q7 B. ]" ]. `8 w  K- h& a
185. return   GetDescription();
     / O2 P* _' @4 r; ?' Y6 U8 X
186. }
     # E4 L, |% a/ g2 |. D0 `& t' g. ^
187. }
     7 [# |0 ^$ D' k% Q) T- f
188. }
     
189. }
     6 z2 A' H* ]5 W. q
190. }
     9 O% q4 C- t* c) P7 p
191. closesocket(s);
     . e+ Y  t2 F+ P. \7 @: b3 a2 E3 O
192. 
     
193. return   false;
     : _" H  D$ `: N  c- \! T& A
194. }
     

复制代码

天云上人

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


8 u% c4 ]1 y0 @2 F* G///////////////////////////////////////////
4 R: {; F& d8 m  q6 u; J3 R//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

9 Q. [; j% a0 s( _4 A# I' k0 h
: W8 j6 N# F" B: T#pragma once
6 Y" S6 r4 [% B5 j* b; O- M0 _$ p#include <exception>
4 y* z* |( T) _5 V; l1 }! A. j
# L" |2 S: i: m2 a
+ G6 f+ }- n( O4 o  enum TRISTATE{
        TRIS_FALSE,
/ _7 v3 q2 U" u) E5 v        TRIS_UNKNOWN,
        TRIS_TRUE
};
' U1 Z, T0 y! [9 B' P. P) `. ^
- ~, j. w9 ]8 C: Y
: H9 \0 f1 y+ V+ l" A1 Renum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
+ d! X& Z( P1 a& @) D6 ]        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
+ ?/ w, T0 ^5 p/ L+ n
) T* K4 C" R' j5 e. U' w0 m, q! \

5 d2 F( i/ w% k
class CUPnPImpl
{
. v# s. ]! X+ Epublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
7 d  z* j" |4 C/ v; _' l                UPNP_OK,
" @5 ?& L8 v# x2 v/ K. g                UPNP_FAILED,
  r1 p, F  |5 T" o$ r6 G: D1 |                UPNP_TIMEOUT
        };


# L, [0 }* E6 T( u( l/ z, h        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
2 B2 s8 Y; X3 @0 C        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
  z8 T9 k. T0 k) _, k# k( n7 [: Y        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
) u( Z# d. {3 T& k" k$ g( S# J2 A" d( }7 o        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
$ A5 b2 k) y, q  x# j* a' O        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
6 y: b; N5 y  `1 L- ]2 h

// Implementation
: k% [9 P* K3 ]" |5 b3 s1 ?9 hprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
- V3 H6 _4 r6 s& U. X$ L        uint16                                m_nUDPPort;
3 Q! v/ p' p3 U) C/ ^" g+ w, G        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
7 G! ]6 J- a1 ]$ d: f5 e" Z        bool                                m_bCheckAndRefresh;

6 N2 I" A- U3 d; Q( q! `& X; ?) W
/ ~5 \# ^% ?& G/ q. t' Mprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

5 I% C8 ?5 C2 h% k& r) [9 e+ F- M1 p
* G$ ]. }& Z$ ?: L/ ^" A9 z! r};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
6 w5 E6 M  D/ C/ `; X5 f% R+ w        virtual bool        CheckAndRefresh()                                                                                { return false; }
) B( e$ W' j6 W4 s+ N3 z0 R  b        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
9 d: q, S* d5 q5 m0 k        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
. b* w3 U) Y: t; z( [" A( z};

2 q7 z9 L8 _1 A& Y7 r) \+ o
, @& v1 |/ t8 q* d6 T, b/////////////////////////////////////
. _, ?; i  I; W" G7 o; g& L; s3 N//下面是使用windows操作系统自带的UPNP功能的子类


4 H- ^1 V4 t5 t4 y! _#pragma once
) f! h+ m) N$ b& I#pragma warning( disable: 4355 )
2 ~+ E1 X2 [; B

1 ?8 [8 R( ]- I; ?#include "UPnPImpl.h"
#include <upnp.h>
& N. c, U* A7 G#include <iphlpapi.h>
  ], w9 T7 _! W#include <comdef.h>
5 p" Y/ j; Z6 V* q- s9 H/ U3 h( `#include <winsvc.h>

4 O2 O% T, Z: S0 s% o
#include <vector>
: k/ p& o) s# Q0 S) w  T#include <exception>
& h, b) B. ?  k/ P# v#include <functional>
. y- @/ ]+ p" q
( ?1 G" M+ D, _( e4 X5 b5 ?


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
; F. A# \* b3 q+ {9 Rtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
# @2 C3 g$ X, V  Z4 a) d- E3 X9 btypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
8 E- Y) w! {' g! W1 Stypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
/ O# {0 r' |8 _5 W9 Y- ?$ F/ itypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
. D3 M& z' [* O1 z$ }, B  IPAddr dwDestAddr,
1 G  E& n' U  a6 k  PDWORD pdwBestIfIndex
& z! N+ s3 g! I9 T6 f" J5 L3 L);
+ `1 I" `! p& A8 F9 B" M
2 v. W+ I7 ?8 P( M% u
- G* J6 O# r3 k0 p' f% J2 s  Stypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
2 U- r( Z5 v  L0 R, r9 N( q& l* `  PULONG pdwSize,
- ]$ c# z+ y" e4 H; s  BOOL bOrder
7 n/ [% a5 m/ h9 W% \# `% y$ C4 S);

, s$ W: h  r! s% g9 W- p  F* p: T
& V- Q. P3 i: y1 Jtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
" Z  u' Y0 w& _( F( E);
" j  M% g, o2 m) O6 F
- U7 Y: N3 X) f3 V0 g) D' A0 G
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

: k9 z* B. h, Y+ z1 {
, ]3 v& r% H3 n( i  \1 c1 m# Vclass CUPnPImplWinServ: public CUPnPImpl
' Z+ P5 ?) }8 t* c' z4 l# v# S{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
) d" w# |5 t  s# z$ J- E% S        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
" Z% V/ d" E# R+ h2 O# J2 z

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& ]  H: p, Z; Y        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
( A3 x$ k- a8 u5 X" f        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
2 ~; ], t/ }$ d# ?

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


/ N" U$ j% F% V; P* oprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
1 \& n( D1 r& H( d  J        bool        OnSearchComplete();
" w! }& T6 `' l        void        Init();
0 P* U+ \' F9 V# O' z% s
3 \0 w5 g5 J* f3 R/ f
        inline bool IsAsyncFindRunning()
4 {3 }7 l. H; O: @5 j        {
3 v+ S" D& Y- m& q; q                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
- j/ X1 s; Y- A- i  [9 U* `                        m_bAsyncFindRunning = false;
                }
* P6 L5 D! V' n3 j& s) V                MSG msg;
( K& q9 @0 f; |4 s* f3 m                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
- s0 Y6 t; m5 ]                }
                return m_bAsyncFindRunning;
        }

9 z. m2 r: E' c+ \$ s
        TRISTATE                        m_bUPnPDeviceConnected;
+ z9 j. Y% `* M/ i, j" W

6 V( u3 L, t) ~( {// Implementation
        // API functions
3 _0 r  q& ^) T% G- ~        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
8 E( L3 Y& M# s+ O        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
2 f6 d! X, S- X. w        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
! x8 R* @' s- i7 a7 h        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
7 Q/ G2 H9 u& y6 n# V* \        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
! {" n( t, @7 _

. l7 J$ e4 Q' `0 I! W2 U        TGetBestInterface                m_pfGetBestInterface;
9 F5 w( y2 n2 }        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
0 m9 I0 ~- Z' M0 y

        static FinderPointer CreateFinderInstance();
  e# U; M. ~' s0 C        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
5 O* T, y" i4 k) A: K                result_type operator()(argument_type device) const
                {
2 T  v- D9 X! }4 q' M                        CComBSTR deviceName;
: P$ i& B; x/ y$ D) _  z9 n# m                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
- L2 k& J: Z/ }  I( r* e                                return UPnPMessage( hr ), false;

5 \% I5 l8 Z# Q
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
: B" Z7 T7 Z' V; H5 ~& w- t                CComBSTR m_udn;
        };
" A/ @4 k0 W8 |# U7 x, O       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
7 _# {& \& f% j: a- @        void        StartPortMapping();
+ _" `/ B2 h7 W! n0 \        HRESULT        MapPort(const ServicePointer& service);
$ S7 H: C8 `7 ^: |) g+ V. H/ d: K        void        DeleteExistingPortMappings(ServicePointer pService);
5 E# J# a8 v# }- b: y; I1 k        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
; x5 L$ W( L3 q9 B                LPCTSTR pszInArgString, CString& strResult);
1 [& m" l) ^7 ~! w% y& @* b        void        StopUPnPService();


        // Utility functions
6 `1 [- V, F$ l- ?( H        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        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);
' U9 D5 `' _% ]3 w& }* l. x
4 j) |7 `4 F/ _; I) s  }. d$ J
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
: z. t+ G+ {0 I" z        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
8 ~( y# J; e' k5 t  F$ i! ?        FinderPointer                        m_pDeviceFinder;
- k  U* h7 i# S1 R        DeviceFinderCallback        m_pDeviceFinderCallback;
+ X6 h1 Q& T: @8 x/ j# O        ServiceCallback                        m_pServiceCallback;
  q' @" c9 l( `3 N6 l* U0 Q

        LONG        m_nAsyncFindHandle;
1 k, e; L& h* @4 K# H        bool        m_bCOM;
        bool        m_bPortIsFree;
. x8 X9 e% x' d( U$ U0 u        CString m_sLocalIP;
" u: [( _3 I, X& N" S        CString m_sExternalIP;
- u6 U- M, Y( s4 X0 M4 f' H        bool        m_bADSL;                // Is the device ADSL?
, K# D" r" G2 c9 o        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
  e* i1 w' r" L6 V2 v9 ]  ]3 ?( \        bool        m_bAsyncFindRunning;
& H' R- X* ^. V1 n1 ~' D8 Y        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
6 E2 G) ?5 i$ G/ i2 Q        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
, ]0 T, b9 ?3 a( q) y6 X# q        bool        m_bDisableWANIPSetup;
& y8 |+ M" M) R% x$ s1 G        bool        m_bDisableWANPPPSetup;
* V0 _! @! i- @0 w( Y

) m( f4 `- J. R. f};
+ D; {5 i- A+ \6 j! m
3 ?9 z+ N1 a( p' \1 e( N2 W
4 z% H+ w- g3 s& R. x8 |// DeviceFinder Callback
class CDeviceFinderCallback
5 p% M( P# ]' ]4 r        : public IUPnPDeviceFinderCallback
$ {8 k% H9 k& O: @+ O+ M  C  t% Z{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
& n4 Q( K9 s) q# s. k                : m_instance( instance )
        { m_lRefCount = 0; }

  a7 k# O8 ^$ A; N# ~
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. G% q1 q5 J* A2 X7 S$ h/ n% C& F   STDMETHODIMP_(ULONG) AddRef();
" j! F8 D( x$ w8 i/ _# E( N7 p   STDMETHODIMP_(ULONG) Release();
8 {; x! y  C7 q
$ H0 W4 H; ?: S' e
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
! y% L$ M% N+ |5 y, ~        HRESULT __stdcall SearchComplete(LONG nFindData);


: ^7 D' l$ y2 L  t( j& nprivate:
. G. S) R6 {/ g  S        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
' h" s  ?$ t7 E};
- v9 a" I- ?8 d/ c0 i
2 j0 U4 I& [/ A" y% z+ u
& I8 q4 `4 P. ?// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
( ^8 B$ I2 F# u, L3 a{
public:
- k4 G2 T& t: |        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
( b4 Z4 y" ^9 f% E9 `1 _   
. |% g+ e5 u  m2 Z6 f# n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. F& l/ n9 o6 n   STDMETHODIMP_(ULONG) AddRef();
/ S4 b( A8 ^' ?- [  O) K( b" r; l   STDMETHODIMP_(ULONG) Release();
. l7 V; D1 D+ g/ c1 u6 e
: Y% N8 J+ F" ~0 f+ ~$ Z, Q
// implementation
! v( j. T; P& E$ h% pprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
6 n1 f0 `  Y' B+ G- {1 i3 J        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
- g& Q6 Y3 l9 o! _# N

$ v: w3 P3 z1 W2 Y4 kprivate:
+ P# l8 L' b5 [7 V- o        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
  `9 Z( F) F5 P1 p, ~- D! N};


3 |* H  v4 X, X- @/////////////////////////////////////////////////


$ N+ t6 P, v' ?% t: c$ L5 B; t% q使用时只需要使用抽象类的接口。
( l) C: A% }6 F* y! T& cCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
, h  F" a& t1 d) z$ c$ Z) [3 |CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 S7 s* {' d9 S* O4 d, j6 {  mCUPnPImpl::StopAsyncFind停止设备查找.
1 @  {, `# ]* K2 fCUPnPImpl::DeletePorts删除端口映射.