UPnP

天云上人

/*uPnP.h*/

1. 
   7 X. V: r: ^. S- X8 t
2. #ifndef   MYUPNP_H_
   7 v( ^: o/ g$ C: N) [: q8 r  [
3. 
   
4. #pragma   once
   0 D+ a0 }% e* l3 f
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   ( y" M% \4 k: @
10. public:
    
11. typedef   enum{
    * A% i9 U* p3 a$ _
12. UNAT_OK, //   Successfull
    # ~1 m" D4 N5 C3 f2 C+ R
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
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    " f) T! Z# W5 h7 G6 A2 q
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    " ~2 Q( _  B& g: F' |: R
23. 
    9 R9 {5 P/ Q- x
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    % ^" b2 S9 s  e% h
26. WORD   externalPort; //   Port   mapping   external   port
    & r% _" @7 C) G; R7 f/ x
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    0 O8 }2 ~& u; N8 [4 B1 T
30. 
    + Z6 o% B+ Q2 w4 E! t1 ^
31. MyUPnP();
    , b0 N3 w, s: x9 ?4 a  N+ d
32. ~MyUPnP();
    ! y; ^) A: i' C' l2 p0 c
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    $ j' {( D( {/ Z/ e! ?/ I
36. void   clearNATPortMapping();
    
37. 
    : p7 v" Y* V9 N7 K# p
38. CString GetLastError();
    8 @( a' D, Z7 b( ~1 z7 L, v  V
39. CString GetLocalIPStr();
    + O' y% a* y. f& {7 Q0 B& Q* L. |
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    6 l$ t0 u9 s$ E- I* E
42. 
    0 D+ g9 b2 S6 }6 G5 Q5 [8 p
43. protected:
    
44. void InitLocalIP();
    % }- c& \- x( o0 U5 j. f3 L+ T
45. void SetLastError(CString   error);
    1 z9 ^  r& E8 ]7 e# ~8 D# \; R
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    / V/ b4 r7 m% H( _
49. bool   deletePortmap(int   eport,   const   CString&   type);
    6 V7 ?; o- B' W. D5 c$ y8 R! P
50. 
    # n& A# {, ~5 s' j
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    9 H$ O0 p9 e. y& S! I% B
52. 
    , c3 f; n9 ~6 q; v* q! p
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
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());}
    
59. bool InternalSearch(int   version);
    9 P& e0 V7 e0 `# W; A+ E; r! p
60. CString m_devicename;
    6 `5 Y% X8 y/ n! j3 S/ G5 B
61. CString m_name;
    % @" O0 e( [/ |  x# A2 }
62. CString m_description;
    , ?. w9 f6 p% G
63. CString m_baseurl;
    
64. CString m_controlurl;
    ' V/ r% P: q4 [9 s! F+ ^8 I
65. CString m_friendlyname;
    . c- i# r9 l' E: E) i! M: \3 v" \
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    5 T* @; }  v5 T2 F# ^/ |1 @
71. 
    ) V) \3 x, t" n# G
72. CString m_slocalIP;
    
73. CString m_slastError;
    - A3 @5 S+ Y  S" p- r/ }" M
74. WORD m_uLocalIP;
    % G/ y( W8 v/ |) c
75. 
    
76. bool isSearched;
    ; p* I  Y: R7 l
77. };
      x0 ~! u9 Y7 F
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ' @8 x' `+ B1 j! v
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ( g: v+ V, p0 Y3 d
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   + `* N# ?2 ^& R8 @
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    * f3 R6 k. v' Z0 f$ o: {
11. 
    1 f* P/ m; m2 }/ S& K
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    1 e' n% Z- @8 t6 Z1 {* p& A
15. 
    + C7 z: C$ g2 d' i  s
16. const   CString   getString(int   i)
    $ R, `$ U" V8 Z0 C: D$ E+ `
17. {
    + d# v* x- }/ f6 O2 |
18. CString   s;
    , _; v( P4 T9 `+ o6 v1 t
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    # f* V' L# y0 `, H& r. o! k
22. return   s;
    
23. }
    3 n3 ^9 ~$ e5 i* P
24. 
    0 G1 i- ^: c/ |' e  Y
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    " Q: i/ b# o4 \
26. {
    # g. Q+ B/ y# I6 C
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    + W# U9 Q+ o' \+ j. e0 v
29. 
    6 A1 R: m& \; d9 Y/ y% @) d
30. const   CString   GetArgString(const   CString&   name,   int   value)
    $ {$ O/ p' m' R3 u, G
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ' {9 v0 S- E  j7 q) _1 a1 L
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    1 \* ?: u, F! q2 Z
36. {
    
37. char   buffer[10240];
    % q( J6 {$ H4 \4 ?+ _
38. 
    
39. const   CStringA   sa(request);
    ' N- B& K$ [9 H, Z: \, S
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    3 \/ t: r1 Y5 Q1 f7 D
42. 
    % E/ R( }% `# p
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    5 _# M+ M, [( ~5 l! t7 \
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - S$ t+ _9 {" i: G
46. sockaddr.sin_family   =   AF_INET;
    / H- _0 C9 _; G% v; B7 z# c2 B
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    : M( i' E4 ?+ J8 m3 D* Q
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ' L' v4 T0 ?0 J9 |* \5 J3 Z, T
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ a5 ?6 J9 X* O% X3 i4 j
57. closesocket(s);
    3 ~8 ?+ D3 }( ]6 f
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    9 G7 ^% V. X* Z
59. if   (!rlen)   return   false;
    ) |6 g1 l) X! K# i9 j
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    ' I2 U  r0 p1 A% i6 q. ?
62. 
    5 h2 ~7 \8 ^* L. M
63. return   true;
    + Z8 S" {' u" l3 U
64. }
    
65. 
    ' ]: e( B( d& e& O4 C
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    , e$ U! d& ]. ~4 n+ \
67. {
    1 `1 i4 k! e+ j! `6 e
68. char   buffer[10240];
    / a$ D! M1 n2 H: L& ~% `
69. 
    5 I# ?8 M1 p2 H
70. const   CStringA   sa(request);
    1 ]4 U  Y2 H* \  y8 @# g1 l! c
71. int   length   =   sa.GetLength();
    & E- O% v" |$ [( k. J
72. strcpy(buffer,   (const   char*)sa);
      q# Q: a# ^# g
73. 
    ' ~- f2 n! l7 u4 Z1 ?
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    4 ^! y: ?, f2 `
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    ; k9 }. R* m& ]1 K
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    : v% \% r6 f- l
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ y  t2 T* `/ ^1 V: J+ Y
81. }
    , N5 H6 P3 G, C" @9 ~, f
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ) g% ], X' T0 Y
84. {
    
85. int   pos   =   0;
    ! U" M/ k4 C+ R
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    4 n/ |# z7 |( V5 s" f* |
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    ' a0 m0 q5 o* @2 G3 y3 m  ]- K
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    4 }/ ~0 W& ^  ^; K+ o& p
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ) Q; J% e; H8 B' w1 T
97. }
    0 g( {* M# l+ ]7 f1 ?* l, Y: W
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     / R0 B& @, B% g' Z8 }
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     # T5 ~3 V1 h% H. Z
109. if   (posStart> =posEnd)   return   CString();
     - b3 B+ E/ b  G7 L  V2 y
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     ! x$ A  w% ?6 R$ Z
115. :   m_version(1)
     5 M) Y# t+ ~- R1 w+ g/ y
116. {
     # I" j9 `2 w+ s7 U
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     / H9 `& u* E: o  {8 B
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
       M" N; ^  p& |& J. j4 K8 K4 w2 @
122. {
     
123. UPNPNAT_MAPPING   search;
     * |4 w/ Z: I" {
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     ) Z6 s: n. s) f  f  K
130. m_Mappings.RemoveAll();
     
131. }
     2 o, O/ y- P+ @+ c
132. 
     1 a  X- X$ L6 Z- @  G) R
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     5 x6 I4 x( O! B8 B, \
135. {
     
136. if(version <=0)version   =   1;
     + J$ d2 l1 Y% X
137. m_version   =   version;
     1 w! Z0 Y* v. S  Y
138. 
     
139. #define   NUMBEROFDEVICES 2
     + i) y( Y7 R7 _& T, X: J
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     . j" t0 e$ b, v# W% V* l
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     8 `0 i* j  W2 q) v' n
145. 
     - w9 X9 ~' B4 O. h  a$ ~, M- v
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     5 M) J$ R8 K; b* ~% S* I: u
147. u_long   lv   =   1;
     % I4 c4 l9 l* o$ w1 `+ L
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     + }7 n- v$ s  B9 V8 m4 l- q. S
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     5 Z- Q4 N% f$ O3 {0 U2 z
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 "),
     # e/ [+ ?5 l! b, f  E6 l
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ( V) u! I! A8 j- z
160. }
     
161. 
     8 e' V, c2 s8 c4 M
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     + c5 n1 ^: o" _) v! u/ `7 @/ Q) `$ T
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     " E8 W* h5 D" V6 \, n5 {
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     8 r; Z7 n9 [$ Q! U  _
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ' l8 I$ `9 F7 @8 u5 }  X
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     + s$ t+ M0 j: i
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     * I* p+ }* q. O; ?' e
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     8 \2 s" \' F2 W8 K7 J3 W9 N4 Z
184. m_description.Trim();
     , @4 j3 x+ [- O
185. return   GetDescription();
     
186. }
     6 n5 v+ [0 R2 ]7 i. H6 {
187. }
     6 q* U0 x$ |. Y
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     : c$ e, ?9 m+ k# d9 h1 [
192. 
     ( H% j9 }& k9 k
193. return   false;
     & z' c1 Q, R9 y# q, i4 Y. y
194. }
     7 @& M* A2 g! z9 r  n0 z3 [/ e6 V( a" V

复制代码

天云上人

以下有关upnp的接口来自emule，
8 O+ r* U5 ^# @0 }+ F" Q

- \2 Z, k' t3 p9 ^* S7 z& q: D! j///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
( {. q  u, }) \- `% U) [#include <exception>

2 j$ q/ w! a+ b# H& \% p* L8 [' `
( y" J. e; g  q% m$ @6 _  enum TRISTATE{
        TRIS_FALSE,
6 B# Z; s$ D% \$ r2 f, a        TRIS_UNKNOWN,
        TRIS_TRUE
};
4 Z. ?7 C/ c9 j  d3 r5 `1 ?9 `
/ d4 m" k, L1 _0 l# k! y- M
/ R( O% {$ C# W, I' h  \# @' kenum UPNP_IMPLEMENTATION{
5 V4 ^2 `& v& t- @7 \  z  [        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
: ~( v, Z: o) a( j4 T};

3 T6 C5 t7 I) N& m, W2 k
1 V- J8 a. {% w- e, _6 S8 E
1 c) E0 _# m. i9 F0 p; }
class CUPnPImpl
. E0 F# K+ C* J* a2 s{
2 _. _6 A2 T7 l, ^& Epublic:
4 U) `3 o7 }1 u, z  K        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
" N) ]1 z( O+ R7 L' u                UPNP_OK,
- s2 \; `, l9 Y% p) I  j* W; q2 ^                UPNP_FAILED,
                UPNP_TIMEOUT
        };
' P/ d# `  k2 {% _/ L
9 Y& ^# x" x5 @0 @8 F, W
6 J6 }3 Z9 [( g        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
+ _: i0 Y( D* f        virtual bool        IsReady() = 0;
3 f) B  m1 l8 g: ?; V: h$ _6 u        virtual int                GetImplementationID() = 0;
       
$ I0 q8 V" x* ]  x        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
6 T' ?6 e- o* X3 z9 q" i2 [& |; E        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
+ o& L. s: |3 m) A# o! N        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
! R; i4 o" f9 C' _% z
1 D3 m3 v1 d) i* j
2 W3 B1 Z9 l* x; }# Y  h; e// Implementation
0 o; ]- D. x2 ?2 }! t4 {protected:
: f5 f( ~+ u0 N; B- P        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
; A. j2 l3 X, U. H3 t        uint16                                m_nUDPPort;
. v6 I, e( g2 D( m+ V" O        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
6 Z5 Q7 a6 b- P1 ~- Q        bool                                m_bCheckAndRefresh;
5 A9 [' M: S3 W2 y( \+ Q1 U% f$ M
" A8 m4 X/ ]/ |0 B0 w
+ u+ R& b5 r3 V% [* fprivate:
( Z/ {. d; d9 j+ x% H8 D        HWND        m_hResultMessageWindow;
$ N8 x4 D1 F- O( J        UINT        m_nResultMessageID;
5 H5 F% {; w2 ~# g) S& C- _
9 O& ?: q# {! Q; l& o. f& a( C
};

' C# L1 m0 g1 J4 C5 C5 E
// Dummy Implementation to be used when no other implementation is available
9 `/ K1 v- t  R, Eclass CUPnPImplNone: public CUPnPImpl
{
# F6 A0 T' `& W# {, g6 u* Jpublic:
5 F$ q; W) e% g7 X7 V0 x( J        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
2 A6 A: X. ], ^& \2 o        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 @/ }7 D4 r) ^. R        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
5 t, o. z1 R) ~$ ^! K5 y        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

4 A: u' Z; ^% a
/////////////////////////////////////
3 C5 v/ v; O! |) b& j% E4 ?//下面是使用windows操作系统自带的UPNP功能的子类


5 g* ^$ z( ]- n" w  Y1 j) v7 K: J#pragma once
- u3 T" q2 C& ^' I2 D" s) R#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
. f. f$ B: Q, i+ x7 _#include <upnp.h>
#include <iphlpapi.h>
! s7 p+ b  _- e; Y$ ~4 D#include <comdef.h>
#include <winsvc.h>

) X3 m- u# B' t& [+ A8 o6 k% u4 b
#include <vector>
#include <exception>
* P3 S9 G  a  I, l) H#include <functional>

& W7 P$ y, e, ~6 E9 ]0 P# h

/ h7 B% B7 {( G
& }1 Q: R6 o9 I- |. e& d* ytypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
* Z* B  e7 R( W1 Xtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
$ U4 F. Q7 M% B  ?0 K! k9 i  \# Ptypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
' G1 ^& j  B: h+ m) p9 z- L6 Jtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


4 a2 o" y  ~! Ntypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
- Q8 e$ B# r4 F- ~5 e( G7 q  PDWORD pdwBestIfIndex
);
) s+ n8 F$ L5 E2 r7 d# g, j/ Z5 \' n
. w* c* |7 R  Y) H* K0 C2 @6 K$ u
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
6 s/ l3 t; v. v7 u2 X  PULONG pdwSize,
  BOOL bOrder
& a1 w' q$ }( c# s);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

' a4 q0 r% E! k$ L  z) J
8 q! U$ u8 z9 ?4 n& |& i3 ECString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


: Y- k6 U! z) V5 i/ Jclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
& ~) l& ~* @' F( ?" M' A; C$ q        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
3 @6 n: F9 y/ T* L, s  ]7 n( f1 V        CUPnPImplWinServ();

( x7 n9 ^" L. E% Q) k! j* g
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
6 B& L- K, w3 T0 ]) E( k0 n( R        virtual void        StopAsyncFind();
4 D1 k0 k- ~9 Q& Q! I, O6 \) G        virtual void        DeletePorts();
- e7 y. S) W- C+ L: l        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
7 F/ b1 S; M  {
' L- x) u) Y5 {1 P9 E* |0 ]8 h
        // 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; };
: B1 f; \; @$ z) _2 t
8 f3 Q: ?# n& z% k3 Z/ l
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
& y4 @$ `$ y8 n- k$ w2 h) v        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
* v( G3 x9 [5 ]( S+ g        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
4 O8 `% C# _  q% R' X6 i, `. y
  Z+ F% o: Z7 L3 t, c
$ V# x1 s5 l. ?8 H2 b6 J- p        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
- n/ ^" a% Y6 v8 l1 |                MSG msg;
' \( c( q3 a' T& V0 p( N                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
* |. [' l9 H+ x1 M& t$ n; s                {
7 Q( M6 `2 e2 u6 @: _0 ~' K                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
7 M# [* m% M2 W3 }4 S1 R        }

% ?( U% A) _6 m
% j, P& D( c6 k5 H, F$ [        TRISTATE                        m_bUPnPDeviceConnected;


, \* ]4 e/ h& d! A/ m// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
4 L5 N3 I& \8 z& O        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
$ u% ?1 z. i% j4 ^& ]" A2 w8 ~% _        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
& O' [2 r. V' Z7 d% q        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

* s4 j5 U' c# N2 g. T" u6 J. K0 f, z. s
/ ^0 T! l; m- f2 F6 O+ F        TGetBestInterface                m_pfGetBestInterface;
0 {( N8 m8 O- I! R( H2 U% \        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

' s( j: x& D+ G* o, o
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
# l9 ]0 L- z6 v5 f! p. x# L/ M                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

# I' _* K8 k6 U2 [! s: h* o
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
/ C$ s. R2 R' u6 Q+ H' x# S        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
& @1 A: A3 d2 B% Q$ ^& w        void        StartPortMapping();
0 @( m3 E: Q  t" W( J        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
! F, p1 n5 E0 |  Q2 I' K        void        StopUPnPService();


4 [, Z' l2 W) F8 P4 _# R, G5 K9 w        // Utility functions
* H4 u, X3 A$ ^" y" d$ W        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
! S5 [' c  ]# S3 |& s        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
5 v7 _: @8 j% n& Z        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
! P4 H. H' T4 a/ ^& c& i        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
; A5 J- \1 p# W/ h( y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
* l  x  B# @5 b" z; B3 ]1 @, }        CString        GetLocalRoutableIP(ServicePointer pService);

8 G* b: R# I8 a
// Private members
% e* z9 v- C0 Z* N# c/ s* \private:
        DWORD        m_tLastEvent;        // When the last event was received?
# X- Y7 W2 f) E- g4 x        std::vector< DevicePointer >  m_pDevices;
3 D8 L% D, o2 \        std::vector< ServicePointer > m_pServices;
0 M6 b: ?7 K- J; z: [9 w' Z/ b        FinderPointer                        m_pDeviceFinder;
" s$ N8 G2 G3 B: w        DeviceFinderCallback        m_pDeviceFinderCallback;
+ q8 P: n! S# k% B        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
- j# x: \$ m( c& S        bool        m_bPortIsFree;
        CString m_sLocalIP;
7 Y3 Z- _- c6 R  b" b# @5 G        CString m_sExternalIP;
. Q8 H, M8 o( e/ D        bool        m_bADSL;                // Is the device ADSL?
3 V4 R  y9 [1 m. @7 L/ Y        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
* f5 z) i8 }/ G2 Y, z# I        bool        m_bAsyncFindRunning;
* A1 `3 V3 N& c' r7 J        HMODULE m_hADVAPI32_DLL;
  [9 v- A' O$ B5 c, m        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
9 a4 J, R2 G5 u8 P  o        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


8 b/ K) U4 J' p, c( N+ D- c2 ?9 H" c+ w};
/ P) L) p& P# [$ k: Z, m
6 J, W& G4 f/ c; F* C
// DeviceFinder Callback
- O8 b2 m- R4 eclass CDeviceFinderCallback
: Y! M" H: a. E' S2 k- a: ~( N        : public IUPnPDeviceFinderCallback
: J" r9 K" _! n- l$ Q7 m{
% X7 n8 b$ s# g% N; W7 Jpublic:
8 s; X* }, J& G* G8 x# P        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
& ^  F! _: t$ Q- H8 _7 Z        { m_lRefCount = 0; }
- a. @+ y- ?% y- k/ h1 n
0 E7 I" p) g$ L1 X5 H
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

/ U4 a6 [" z: u, p
$ [* ]8 L- [' m; F& f; K+ Q3 O( i// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
6 ]. ]  z+ v( v) ~! @        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
- y# {& m  H  b" C' a3 M};
& ?: B) n6 b. |

+ a( h9 x2 z2 ?: w" w- f2 C// Service Callback
class CServiceCallback
- n2 o  [; {' |/ P& P# Y: n$ D9 o        : public IUPnPServiceCallback
% o2 p  z: M; \4 ^{
! d- M7 M& ~6 w8 t1 L+ k( z% |8 Npublic:
1 a1 f) Z( |8 x, _6 |        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
$ _" E) Z6 J, u* O
& P! u$ K' z; [% V
  J( \# x$ H1 t/ ^: r+ Z// implementation
private:
$ o; ^8 j$ v' g& e3 g6 ]3 ^9 U4 d  R+ L        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
! F6 N% u1 i! \

private:
        CUPnPImplWinServ& m_instance;
) T! Z& ^, I% n8 y. _        LONG m_lRefCount;
};

. p/ g6 ?. L7 g+ P1 M  p
. F7 u8 M! U+ g1 A/ \/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
% o1 ~3 S+ Y' C& bCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
0 n% `# q( N5 o3 bCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.