UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
     ]: r& V! o1 U
5. 
   2 a4 c. q/ G% P' Z3 q/ {
6. typedef   unsigned   long   ulong;
   - {! K( t/ g5 ]/ M7 n6 k( A/ {
7. 
   
8. class   MyUPnP
   * s& E" n9 C: P( y1 M
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
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
    $ l) u. k' D' s) p) O3 B5 j0 ^
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    ' h+ @8 }2 I, x( t9 X5 h
17. }   UPNPNAT_RETURN;
      B5 E0 y$ i0 d7 K
18. 
    
19. typedef   enum{
    # w1 ^# p! S/ T$ t7 S
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    1 @. D( P2 I) F. M; v, \6 F7 ~& _& L
22. }   UPNPNAT_PROTOCOL;
    2 F' Y" S9 _* u
23. 
    # C) I' r" r" n, h* A
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    $ G) t& u. v8 p3 g5 ]/ K) N
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    5 j& l6 {4 ]% S" w% }
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    # x/ i7 q8 U4 k+ w7 T2 F
30. 
    
31. MyUPnP();
    - F( s4 @5 b% I4 `
32. ~MyUPnP();
    
33. 
    & B0 e0 c7 C! p5 B6 _
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    4 Y0 X6 b4 I! |& q4 P
37. 
    
38. CString GetLastError();
    * `% ?) w! x. L8 z5 Q
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    3 `. j; j' d5 f# f& `- s, i
42. 
    
43. protected:
    ; o1 k9 p# v  h
44. void InitLocalIP();
    7 K0 J% m5 ~* X6 Y; f1 _
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / s$ B( j' p! h' G: @9 h1 u1 g
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    8 V! |/ [9 H9 x. A
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ( U1 o* {! @1 d4 f; D5 [& n
53. bool Search(int   version=1);
    9 G' A, I; n5 B* W7 A1 X
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    2 Q: F# e6 F# n6 ]; u
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    * k! i3 B; `- P- W
59. bool InternalSearch(int   version);
    0 p4 @# T) E) z+ a' a
60. CString m_devicename;
    - U8 s/ T" S& s4 X  i+ L
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    7 R: h( n1 p/ u. l9 V( a. p: C
65. CString m_friendlyname;
    
66. CString m_modelname;
    5 G1 G5 Q  ~% f. ?
67. int m_version;
    6 n, E1 K2 S6 }  Q
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    " L; v# s, |8 @4 D  G: J5 t
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    - B9 T( {1 Q9 x/ @( S7 W: R) P9 n
74. WORD m_uLocalIP;
    
75. 
    2 s$ T6 S% n( U4 U$ n/ ~( K+ R
76. bool isSearched;
    - M8 r6 T  k+ G( f" r
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ( ?$ X5 d+ V& {, ]3 d' `; N$ ^
2. #include   "stdafx.h "
     G8 k6 e; W" ]0 H" E! M2 |
3. 
   
4. #include   "upnp.h "
   & l% k1 B; D/ }: C
5. 
   ! F; i' C1 n7 B& W
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   . T/ Q. I) Q: v0 N9 ]  W7 s  a
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ q9 @  f, G6 R$ D  g
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    - y$ B# f- T% f* h. h
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    / e7 W, W, }/ @. ?. l+ `
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    # [1 {9 D& N$ ]3 {
15. 
    
16. const   CString   getString(int   i)
    
17. {
    4 r2 x9 V# T/ w0 @
18. CString   s;
    ; S3 F9 @' |- g5 Z, Q6 h
19. 
    . s6 O1 @  r; c# |6 k6 O
20. s.Format(_T( "%d "),   i);
    & Y  h, b5 ^& h7 Z8 p" N$ q
21. 
    5 f6 w4 K% T  x& O9 \) ]! q
22. return   s;
    9 ^1 u  ]. X! ^0 ?
23. }
    1 y$ ?* a) l9 Z3 o
24. 
    1 s5 F% k$ u! E# u, }
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    5 M# S2 Y, R  M5 [
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    0 _; C2 c, k! c0 {$ b2 K/ g1 o8 b: k
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ) Q+ o2 L, ~6 e9 q& J9 [
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    , s2 B3 N  X" j# @, Y
33. }
    7 w% I% [* Z6 `$ e/ A0 q
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    : `9 x5 y, L0 a; h; s9 B
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    3 g, U7 M; f: x, H9 Q
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
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);
    + g! }* _) R7 H- c% u
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    5 U2 f: K# A, D- d3 g
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( u5 l& ]: K8 P( \- S& t4 P
53. Sleep(20);
    5 u8 \. c, L/ [
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    5 R, Q5 [4 [' i$ a; z
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    2 J# L6 A) R, p/ g8 Q  h
59. if   (!rlen)   return   false;
    + f, l2 H' @' Z, o& ?) T5 w
60. 
    6 g2 q0 O7 p1 A9 t+ K. N% n1 e
61. response   =   CString(CStringA(buffer,   rlen));
    . E: `0 S! z/ e
62. 
    
63. return   true;
    $ Q) Y. @2 q( U2 h
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + u& M  r% j" e, i$ u5 _
67. {
    
68. char   buffer[10240];
    
69. 
    5 U3 z" @& K1 q, D. y% e' j3 H
70. const   CStringA   sa(request);
      k/ {( l& G8 K4 _/ u& w. M3 l
71. int   length   =   sa.GetLength();
    ( _& E- F$ h! ?. p
72. strcpy(buffer,   (const   char*)sa);
    , m$ m- i' O2 e; @0 S; c
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
      e. y3 f" a  D' o
76. sockaddr.sin_family   =   AF_INET;
    ( ~* w8 y# @% W" c
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    % `" B: \7 e+ D) G5 |$ s
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    + R) K) f# e3 N$ m  T* R, M# y' _
84. {
    
85. int   pos   =   0;
    
86. 
    ) T% ]) ~! U) U+ p% r. ^
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    : Z- r7 l" ^( H; e3 W4 `$ ^) E
90. result.Delete(0,   pos);
    1 K# g& ~% R) I2 l1 s
91. 
    9 K; D* t4 m5 P8 F% F6 W
92. pos   =   0;
    ! l/ Q+ V  t5 l8 [4 I7 U
93. status.Tokenize(_T( "   "),   pos);
    0 |& o$ K  x* I& X
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    5 _3 U7 K5 N/ i; ^  g6 S' h
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     / f4 C* b- S3 m8 f! M
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     ( Q9 N+ ^4 n  X1 k
106. if   (posStart <0)   return   CString();
     " \6 V. [6 E3 [/ ^
107. 
     3 h" Q7 P: ~+ E& o6 r1 V; _
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ) K- T3 O: r' e4 D- g
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     7 S8 K' v. J, M% n( U4 u
113. 
     
114. MyUPnP::MyUPnP()
     / C% ]% C0 }3 E: Q8 ^1 q0 c2 u
115. :   m_version(1)
     0 X, o! I/ \6 D1 T  \
116. {
     / |) v7 c/ |; R) T( l8 F
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     # b5 c3 W- l9 s
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     : Y1 Q2 c% }) ?' ]  b3 ^8 I0 ?
123. UPNPNAT_MAPPING   search;
     8 v! W7 T, ~- ?$ r) K3 }
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ! R9 _0 K) i0 `1 Q3 r0 V
127. RemoveNATPortMapping(search,   false);
     # l: S7 B, P8 E8 s1 z- O. Y
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     5 y: T  |& {$ ]: z/ S0 ^
134. bool   MyUPnP::InternalSearch(int   version)
     " V( k# J8 U' f
135. {
     . @' j& V9 s0 Y+ k5 R
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     # ]# Q+ k. H: I7 k2 v
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     7 T; b* S: Q5 b' u! \3 V+ L1 Y" [
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     2 e" f1 L5 N! T' F3 d
145. 
     ; w) Q1 I$ D$ F/ Q" H- p* g5 v
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     - d+ w2 w! ^0 [0 f1 [3 K4 `/ O" O0 J
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     & A: k& F# z+ _# J/ k
149. 
     
150. int   rlen   =   0;
     . J$ A2 }- e, W) D' D
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
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 "),
       Y  @: ^; j* q! H2 Z
157. 6,   m_name);
     8 D. C2 h/ x+ p; W! b  s; g( u
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ' q2 `3 j) A" e. l. o
159. }
     
160. }
     ' h1 S2 `. i$ \# P' B7 S
161. 
     
162. Sleep(10);
     
163. 
     + D) R; j' @7 Z) C! r3 s1 T) @) ]
164. char   buffer[10240];
     # K" B( K! w$ y' I, _) W( B) U
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     3 O, g. s; _& i7 r- P! g& ?
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     , Q; O$ [' M  g$ L4 \, l
170. CString   result;
     ! h8 D9 h  K) N% {1 g0 z
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     + Z, f' k+ @2 q$ S# i  b* `# e1 m
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
       ^; K$ ^0 s6 O. Q
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     / t  G* T) |$ r( S
178. if   (line.IsEmpty())   return   false;
     9 n2 ~/ a' w$ M4 g  n/ g8 Y2 m
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     7 a5 [, n9 Y) l4 m. [. i9 c
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     7 b9 N9 d' |- ~# r% @
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     ' V1 h& ]- D% }1 J) b4 ?, q
190. }
     
191. closesocket(s);
     5 W, j: ~  W: q, z" |
192. 
     8 e* u+ u/ a. G
193. return   false;
     4 ~% d: j8 [- x+ x! }  ]
194. }
     

复制代码

天云上人

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

& [- ]- U" ^$ W8 y, i
& [  n2 M8 I2 }2 e7 q7 l' ?///////////////////////////////////////////
' i* v; h  @! B2 R7 v# H+ }% N//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
% b$ z% U7 W! K& C/ ^
. u: J: d; V: j& R0 Q
4 h7 s2 z& e% P! _# v! o% E1 m  ?" q& M#pragma once
( L9 e9 x; \# j#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
/ s: V% Z% Z7 g+ n5 u7 t        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
6 O5 w. U8 I/ d/ v: v};


! O% ]& F. F! ?* L4 t: l$ B, v0 A
; l  \  r0 k9 R! U
+ D( i7 g" e& \  @9 w+ Hclass CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
5 @( Z5 n3 X- e6 c* d: ?$ K( c        struct UPnPError : std::exception {};
        enum {
' w" d1 F* U& ]5 `1 }                UPNP_OK,
8 O$ x  i0 p8 I& q% J" n                UPNP_FAILED,
+ P, o% p6 `8 ~' S7 w: f- L4 p                UPNP_TIMEOUT
; |* ~) f4 K  d2 H        };
, @) g6 |& [1 p

. a3 W; N9 x; Q: L2 |7 e& P        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
8 P( M/ o# H) V# \& z        virtual bool        CheckAndRefresh() = 0;
& x( A5 M( b; l) |- a' L+ V# p) m        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
6 {: m& [5 V/ v, X' ^        virtual int                GetImplementationID() = 0;
" w8 g5 L7 r# P# D* w       
: }! S/ Q& j0 v* {, s4 R        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
' ~/ f2 Z: r7 U1 f

% w6 U% K) v% w9 Z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
7 t7 m. B& W3 f: D4 Q5 [7 R# L        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

, ?1 Y& g) B% w+ f; A
$ I6 z( |9 O( p6 o7 ?2 `2 [// Implementation
protected:
( z( x& L2 j$ D& j( z" j" |9 y4 v9 b& g        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
" O2 k" C7 p8 O9 k8 z; l6 p  q        uint16                                m_nTCPPort;
8 {* X! _2 r0 {  `' y7 q! G2 ?        uint16                                m_nTCPWebPort;
: i7 u, v' |  v: h/ c. y        bool                                m_bCheckAndRefresh;
  }% h% t+ Y8 s4 R

% p9 ?& C* w. @  b% Pprivate:
8 v- T; y5 E" F4 ]1 I        HWND        m_hResultMessageWindow;
' k: E4 l9 I4 b4 [/ l( g        UINT        m_nResultMessageID;

) [$ Y* l# ^( k/ F: G+ G, V
};
) D1 [) Z9 p! x1 [# d
3 S  }/ p6 w$ a6 H" S9 s
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
- _) b* {/ l/ z' l% l/ Ypublic:
9 Q) g4 R/ e- C' ^5 q4 B        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
+ F# V2 _5 s( Z/ s( m        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
/ f8 D* b: `6 O: D};
4 ]. ?) i* w3 T* b& \, a
5 d* u2 }; c# }& n
$ d+ T1 I3 o8 `3 n; p( o+ p- s/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
& ^8 Y9 l0 D' N6 \8 i  J. F

; Z1 X# V  ^! _, g4 N5 _" n! v+ t& k3 ^#pragma once
#pragma warning( disable: 4355 )
( Q. B6 B8 Y# y2 z- y
' C$ z( i! W. z3 |" L
#include "UPnPImpl.h"
# M, X( b. {, I/ j9 z#include <upnp.h>
( l2 d6 T6 z( S5 l3 B7 w#include <iphlpapi.h>
9 j+ `! a2 s5 R#include <comdef.h>
#include <winsvc.h>

1 j( A5 j; e! ~
8 B& F5 v) p+ z* o$ I# q#include <vector>
#include <exception>
#include <functional>

- p" e3 e# P+ e# _0 [$ N
: S1 f/ h5 X+ |

% U5 R* f( ?1 u  \& a! ytypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
$ ]2 f2 ?) ]# e. k! |# D( M/ G5 itypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
/ I& z: z: E3 U0 m  c2 ~( ptypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
8 Y9 G* I0 a1 e8 J0 rtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
7 K2 n% L1 ~4 B3 Y

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
3 M; l3 R; Y/ }. s' M% ?5 G
- |8 `( Y# n- g
# X! `3 M' D% A$ Y5 Vtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
+ H. ~: }) M% j2 y8 v' _+ O0 i  PULONG pdwSize,
* M" f. }( U: ]5 G4 U5 d7 G  BOOL bOrder
% r5 t; t* o- l);
* s3 S/ B) G# S4 y. K. q
- I( Y7 O; h2 q2 ]* b. D
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


/ k4 ^7 `' `) Z/ V5 @/ }# eCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

2 k; t9 _' T* o) r* L3 [& s1 ^
' }- Q$ c. }" b! fclass CUPnPImplWinServ: public CUPnPImpl
{
6 W1 O( f, ^( z* `( x0 u" A8 A1 o        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
& \" v9 I+ r' b. f9 u% w// Construction
public:
. M! o# L3 B: _. |! D1 P! `- e7 B1 m' |        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

  k- w7 p4 b3 J' \3 s* N
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
# v$ s! ]/ l5 N# ?+ @* K        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
1 l9 ~0 B# C& q) W# A! [0 E

3 j& o7 K# {$ @- x. k3 \1 f" s        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
6 C- I& V0 u- A% R' `( p        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
7 f" p  g8 ]( X1 o9 `" h1 J. r5 D        virtual bool        CheckAndRefresh()                                                                                { return false; };


' H8 l* Q# h% b  j# I' a" o+ o" b1 nprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
% T  V1 F8 V  _        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


1 J* L- Z  |5 b- k) }& c+ n        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
0 n4 F) l; S( ~7 B                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
& G0 A% J4 z4 a1 e( L                        m_bAsyncFindRunning = false;
6 X5 O' O8 N% {. K                }
8 m1 O0 I% o3 s" A                MSG msg;
2 h  l( `6 S$ o2 y, U, |! \4 M                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
, ^* Y$ i; _  _; X& d                {
" B. X+ ^$ e6 p1 K                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
; H/ M* L7 M* B5 j        }
- w. ~% f- J& I( ]$ B8 l3 q

0 B5 C- f" w" k/ @5 a  R$ a( e        TRISTATE                        m_bUPnPDeviceConnected;
# q; {3 d1 E& b* _! ?

// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
- }  I2 \, A/ F' [8 X: f- Y        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
* {& d- v( v; t1 I( G) [        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
5 ~% P+ n+ P, D) i1 {        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

4 _7 [! j% ~8 G: A" f+ I  V
( {- u/ [. h2 ]$ r        static FinderPointer CreateFinderInstance();
8 T2 Z) c1 {. v' G$ I% M9 d8 Y, S, t        struct FindDevice : std::unary_function< DevicePointer, bool >
5 n6 F! d5 z! a1 B- v        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


% _1 l+ T0 U! a/ a) o2 {                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


+ |8 K+ w4 B: n& M3 m% ^/ P( T3 g                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
  T  z. }$ L1 y& \$ i6 @3 ?        HRESULT        MapPort(const ServicePointer& service);
5 ]# K, n  Q6 S1 {2 F: ]        void        DeleteExistingPortMappings(ServicePointer pService);
% b& V4 ^( k6 _1 k        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
( g- f6 w8 i/ B9 Y                LPCTSTR pszInArgString, CString& strResult);
( x) k5 ]$ U- P/ c) F        void        StopUPnPService();
1 r# b; w. q. q1 ]
$ j# p2 ~3 {6 b, D/ v2 z/ N
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
4 X! O# Y) h2 M        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
8 ?9 j1 D0 I; y9 @' ]+ ^        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
0 o+ L, b6 ]2 Z4 R/ |3 u4 J! H        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
! N8 I1 S, V. F7 E        CString        GetLocalRoutableIP(ServicePointer pService);
, M/ O, P& Y8 J
( H$ a0 C5 ~: [2 K7 d
4 @- g; T# R6 X  E// Private members
) `7 ]  @& w# z+ o0 mprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
7 B5 @5 z# c$ T6 ~8 z' K$ \        std::vector< DevicePointer >  m_pDevices;
: C% v' G  \7 ~8 l0 U        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


, P- Z4 Q( \% N) L        LONG        m_nAsyncFindHandle;
3 G, J& A% Z5 `2 D2 H        bool        m_bCOM;
        bool        m_bPortIsFree;
4 q# _7 V5 O# X        CString m_sLocalIP;
: {( r- A/ _; }* S: G7 G- L" l        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
- V! ^* r0 \% O3 S2 m        bool        m_bAsyncFindRunning;
8 I0 ], q/ U+ X1 Q8 M! {4 w        HMODULE m_hADVAPI32_DLL;
6 a6 R+ d+ O' l0 z8 W        HMODULE        m_hIPHLPAPI_DLL;
3 c& I4 F0 o4 A# y+ R% N3 `3 n        bool        m_bSecondTry;
( n) j) V, _4 Y/ b        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
9 I/ p/ g) p  K6 t/ l0 b1 ?        bool        m_bDisableWANPPPSetup;
+ J7 u; y/ ^" T1 {7 \

};
4 |$ ^1 R& [3 |0 r8 n- h& ~
+ q  W$ b1 x* a9 B. K/ Y
3 S4 O+ ~& H$ p4 e2 w// DeviceFinder Callback
class CDeviceFinderCallback
5 K, l+ V1 @7 D8 L4 V1 ~        : public IUPnPDeviceFinderCallback
# T# g9 V. y- v& y{
public:
/ |9 S0 j  x0 ^; t" t- c9 |1 d, m        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
, T2 b1 D* l, q$ r6 {! H- I        { m_lRefCount = 0; }

) \( o  C1 f. X! d- l/ S; T
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ o8 C- z0 y( H. H   STDMETHODIMP_(ULONG) AddRef();
* ^" X- p& M$ c1 S1 R( \   STDMETHODIMP_(ULONG) Release();

1 J( ^2 p$ G4 B5 N, {
// implementation
private:
( R" [( e" B- \, t0 y% L6 o$ X8 L        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
7 }1 i* |' G2 n& U& y- G9 g" o        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' j; W5 \) B" U        HRESULT __stdcall SearchComplete(LONG nFindData);
% z: k* O+ ]1 }! q8 `" R: }2 n
. Z# ~6 ?* V/ T5 P
private:
: U3 Z( {/ r- \  n5 L: j& I- b        CUPnPImplWinServ& m_instance;
6 O7 l! s: p( ~# H" e' K        LONG m_lRefCount;
5 b$ o; L! `* x1 D, h4 d};
1 o% o( A, Z# G# I4 ^3 D0 d- }' z
! ]0 N' d8 E9 j+ s
// Service Callback
class CServiceCallback
- j) y7 q6 i: B3 D        : public IUPnPServiceCallback
9 U0 D" Q1 W, [6 }$ g; R{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
+ |; I- }4 n  f; H  Y. A                : m_instance( instance )
7 K. b/ f4 Z" F" H        { m_lRefCount = 0; }
/ k9 X" A9 ]4 Z1 s! Y. s- W+ i   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) E, ?8 E4 X; @4 [* d

// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
2 G9 U! k& O  ~( O2 K+ O# p) c

1 d( t, L  o5 F! [" n' Kprivate:
# m/ y) Y/ G0 J8 e5 I        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
7 D; r! O0 }' U# H  M) A. ?' ^, b/ w4 m
) S" Z( I# F- ~& N  F
/////////////////////////////////////////////////
: s( m: t. F  s7 S# _% ]4 l) `0 N! `$ @

# @9 B1 I' X# F& i$ ^2 ?使用时只需要使用抽象类的接口。
' z2 O* p# [  B9 [CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
- C8 U9 C' `1 d' g5 I4 ]4 XCUPnPImpl::StopAsyncFind停止设备查找.
0 E0 E; G8 t; b+ O* N* v  YCUPnPImpl::DeletePorts删除端口映射.