UPnP

天云上人

/*uPnP.h*/

1. 
   ! s' t: K) k- Y: a) l3 M
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   % n) ~% A+ }% c, E4 e$ y" O- k
6. typedef   unsigned   long   ulong;
   ; m7 Y1 w5 |5 b. B" P' S2 u- R+ B  u
7. 
   2 G* \) ^6 V, U
8. class   MyUPnP
     F1 p2 S2 R5 l. H  r+ d" C
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
    : R. ]( U- O# U7 W6 p9 A% v! p
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    4 i: x: E( q  v' J" @+ ~
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : z6 W' @0 R  b6 u' T9 k5 U
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    1 I4 v+ v* j" a3 n$ {
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ! o# j/ s) }+ R0 k
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    , [2 p( |7 L  K7 B. V+ p
31. MyUPnP();
    / Y+ p! L9 E6 a
32. ~MyUPnP();
    8 s1 N: a8 w0 T5 k
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    % w0 n- ~8 L+ H) U5 N
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ; N5 N  G6 ^# q# L
37. 
    
38. CString GetLastError();
    9 ?5 |( y3 z& [' ^8 Y6 O
39. CString GetLocalIPStr();
    , w. [' ?( N- g/ ?* D; ~5 f: o" ]
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    " `9 q  `( z0 I5 M4 V* d
43. protected:
    * f3 [+ j6 h' o( Z; q9 d8 p
44. void InitLocalIP();
    $ T6 O) h2 }. d6 f: z
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ; ^+ H, o+ ~) T$ @
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    . D8 R# g* o8 Z5 X) A& P
52. 
    ( \( i$ V, i( \) x+ S3 h: m
53. bool Search(int   version=1);
    
54. bool GetDescription();
    ' Z# }4 U. v5 ~0 [: }; [9 C
55. CString GetProperty(const   CString&   name,   CString&   response);
    " ?) ]7 b' \0 ^, b4 l
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    / H8 h6 y* ~% x
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    / M! _4 C% I' K9 D7 T" J
61. CString m_name;
    " X1 {- C) `& R6 {# i6 E2 ?
62. CString m_description;
    
63. CString m_baseurl;
    9 _* n# j' B, O4 C$ N
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    * h9 Y# [: {: G* k# o
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ' h9 S! G# R' b+ i. f. m
72. CString m_slocalIP;
    ' K  T( a! l* ?: Z& s
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    9 D/ s8 ?( i3 l0 D8 R& w9 \
77. };
    ) w: x% S1 u8 N, P5 V
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   5 v; M8 J/ [6 V# a# W3 R
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   7 l6 k; Z) M1 s4 O8 x* B
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   1 Z. `8 L! G& i( Q! ^* E. D
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    / k. D: V# ?4 I& v5 @
13. static   const   int UPNPPORT   =   1900;
    7 i0 |8 Q  h3 A+ Z7 [. L
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    8 v- R+ s. V% p) Y7 {( D: o/ S- w
15. 
    , X7 H; x. y; ]0 y
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    & N% s% v. f+ E- g% U  C' @
21. 
    
22. return   s;
    
23. }
    
24. 
    6 }1 i2 f. @% O; \) z
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
      `; c) Y( Y( F! P% O
26. {
    6 N1 D/ L# I4 A
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    , S+ x, o( h$ b) I7 W
33. }
    2 c5 j# {# H7 |4 P& |8 c
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    - Y" B1 E  @8 A9 q( C: |
38. 
      w: {4 `/ j/ }4 [
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ( a0 J- ~9 |3 R1 B% I8 @
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    0 y; J% Y5 K$ i. |$ \
44. struct   sockaddr_in   sockaddr;
    , p8 h8 Z0 _) j. v
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ e$ x4 P: ]. {# D7 F1 d
46. sockaddr.sin_family   =   AF_INET;
    " W' `6 v4 _! g+ @" i
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + u- J3 }2 N+ I  |7 W/ u
50. u_long   lv   =   1;
    3 V8 {5 v' ^: k7 C! ]9 ~3 V4 A3 X4 V# D
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; r5 }$ G9 n3 _$ O3 g  \
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);
    / S0 M! i# X) P. z  }0 ]
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    0 \+ R2 c. M( j5 a" k
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    - f3 h" {& l+ ?+ x
63. return   true;
    ' O+ L! V8 N, x* Z" j4 q0 }
64. }
    ) T: Z9 c- |; O! Y
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & t0 y" @" N- `7 L
67. {
    
68. char   buffer[10240];
    ) n9 a7 B+ W: E# K
69. 
    4 h4 a$ ~6 y* p' F. y
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    4 @3 ~7 O( C! d* W. o1 t1 J
72. strcpy(buffer,   (const   char*)sa);
    1 C/ e2 g6 F- v7 P) Q3 U
73. 
    
74. struct   sockaddr_in   sockaddr;
    ' E/ ~3 @4 l1 Z3 u) M1 S1 m- ^5 c
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    " R* G) k, d$ A+ @" X/ J3 ~5 ]2 D
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 a# E8 V  i1 K" e* ?3 g$ _
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 D- s9 R& e, E
81. }
    
82. 
    7 N& }$ O! l+ N$ S
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    # P5 f. R- `3 ~7 b
88. 
    3 l: `8 T4 |6 Z& T# I' o
89. result   =   response;
    ) c4 ?$ Z/ R/ I- O* o3 s" G! k* X& w$ @
90. result.Delete(0,   pos);
    ; D0 S/ }4 X* B, [% I
91. 
    
92. pos   =   0;
    . m5 V1 V; o3 ^% L& L+ K
93. status.Tokenize(_T( "   "),   pos);
    1 j+ m* R8 U; C8 m/ h' }3 u9 O) `& _
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    % ]( w) `. ~7 F
96. return   true;
    
97. }
    ; D' j. }8 d+ U. I' I6 n  ]
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    $ u1 [: f3 F$ P- _, D/ `) u
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     6 a1 P, y7 N% r: C
104. 
     - A# g- G+ N  V# ~# h! A
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     ; b) S" n) f/ M9 z) B
107. 
     ' P. G. t" v) {' U' I) G
108. int   posEnd   =   all.Find(endTag,   posStart);
     ( T# X. u, j5 K/ r8 c
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     8 |% {) P: ]# P# R
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     + j( [/ ^; {" v9 I% g
113. 
     # r( F* l7 ~% H, }+ D3 ~* G2 @( S0 ~, O
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     0 I% b7 c' }- \4 g7 N6 l8 w
117. m_uLocalIP   =   0;
     + R- l6 z% Y' s
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     2 B8 x; E- F" E8 l. Q
122. {
     
123. UPNPNAT_MAPPING   search;
     : K6 Q2 |1 n: H4 B( J. Z
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     8 a! n% h& f  R, B* [+ D+ r5 ~( ~: ?
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     7 J3 k* `, O) r2 S
135. {
     
136. if(version <=0)version   =   1;
     - t* f5 y  v# L# g% Z/ _+ f
137. m_version   =   version;
     6 N" U$ a8 s" y# x/ B3 \8 i
138. 
     1 Y9 P: f1 F& C" P
139. #define   NUMBEROFDEVICES 2
     2 Z$ u. W+ J; A9 g9 K+ E  ?* o3 t
140. CString   devices[][2]   =   {
     ' D9 ]2 c7 n) o1 X! D5 ^
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     * R. q. g. O6 q  `5 a, N
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     * [% |, l% ~) C0 s; Q
144. };
     
145. 
     + \' m+ V2 T! U' _3 H* D
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     $ P6 F2 j& `+ S6 o
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);
     ; y3 D9 k' y+ @7 O
155. CString   request;
     4 o. A" D9 v6 E# D. H+ f
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);
     ' Z  c9 w1 w% u
159. }
     # C) d7 ]. ~' Q, I# G
160. }
     5 U/ p; |8 _7 x9 X
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     8 Q4 P3 g1 R2 E9 w* e6 k
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     / M7 m  c4 a( r/ A. ?4 G! Q
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     . f; v5 f6 Q+ O1 Z2 d" @
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     . z8 e7 o/ R! z* \6 x9 v
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     3 l  {* y! p7 ~9 J( h! A% N$ K
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);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     ; c6 ?0 \6 h7 @, z4 L/ I
192. 
     - ^4 L6 A4 g1 ?0 h
193. return   false;
     & D* l3 f0 z1 ]# {/ F% O+ U% L) ^
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
9 D0 [. w6 c' [7 t3 r( ?6 r
2 ^; M, y/ Q. c4 m: P
///////////////////////////////////////////
4 ]0 y7 k9 G, o; n7 k//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


+ K+ w) w! r: n! t4 \: j#pragma once
#include <exception>

! K! A! k. w; D5 w1 y
- `7 c! T# r3 i- F  enum TRISTATE{
! q4 x$ T. O$ F$ n4 e1 H# \9 b        TRIS_FALSE,
0 I) w* f5 P6 W- c! e2 U        TRIS_UNKNOWN,
! W4 `5 @7 y& P! ~$ H5 l7 @        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
% K0 z8 j' p7 y* |3 ?5 P};
" C- {& K0 G# c; n) y. D0 r+ I2 r

0 y& P/ p4 b0 W4 k6 x; E- U
, R+ f3 \0 J! i
. z( z- b( i' M9 {. iclass CUPnPImpl
{
public:
+ s$ b/ X' u" a! }' m        CUPnPImpl();
        virtual ~CUPnPImpl();
8 Z* L" l" d! F2 F9 @# v1 P3 c1 i        struct UPnPError : std::exception {};
        enum {
# i, M4 V& Q5 Z& S                UPNP_OK,
3 T" w6 T& o/ K6 `: k: |                UPNP_FAILED,
( o9 H& f( q4 m                UPNP_TIMEOUT
        };
4 V: H. ?" c& h/ f# `# q
+ w4 m# i2 x# q: V: \! x
( H* C9 V; X% M1 N) t/ z0 H        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ d5 @0 C2 T2 K        virtual bool        CheckAndRefresh() = 0;
( ]# j: C4 T5 H+ k* ^5 \' S        virtual void        StopAsyncFind() = 0;
7 |4 H: ]! b, _  ~0 j4 C        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
3 ^( u* O+ J9 q6 H/ t9 l        virtual int                GetImplementationID() = 0;
, t5 |+ A7 q! X( B7 g/ k       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
5 f! E) O' N7 d+ O) O) Z# d
1 X) _) Y  N6 V% S! F  x
+ L2 C* D* |" p7 r* \        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
1 @4 x* B: g# g+ s% Y        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
( E0 \# M) D2 V. B# G) U3 w        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


6 ]- |1 G* X1 y" ~// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
/ Z1 A+ N: Q& ?6 z) p        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
" r# a0 e9 m' k, T; O        bool                                m_bCheckAndRefresh;
7 _  a1 B6 B  s" l

private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

) S# v' G1 |4 D$ N9 G+ v3 P+ @; p
! R) m0 F- L4 Y2 z6 L7 I$ G};

8 @7 U$ w: q- l, {, x  Z* E1 Q
4 \0 C6 R% F2 O: a4 q) V9 t# {// Dummy Implementation to be used when no other implementation is available
6 P4 e% C' V' Oclass CUPnPImplNone: public CUPnPImpl
  \3 I4 m/ ^# J8 c, w: G{
" N, r( w* m8 f6 q+ e3 S* c* Bpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
+ o& \- ?, H- n* m        virtual void        StopAsyncFind()                                                                                        { }
7 [/ \! [9 I' Y+ F% N' x. |- S        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
% p4 ~6 S7 w$ z: f! M: @3 [        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 U, ?% e5 @  y. T$ k};


; s: k9 R# v) Z/////////////////////////////////////
$ e% @8 u7 t! [* o//下面是使用windows操作系统自带的UPNP功能的子类


/ {( y+ E' K8 t7 T#pragma once
5 D, M8 }1 F! ~# E' J/ i" s#pragma warning( disable: 4355 )


  I, @7 ~* [2 D) L8 E2 C; m# S9 h#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
" X3 l) Q+ Z( c" n0 n; p! [; I) c
* h% f- d0 x4 G8 @8 x
" b' `' u* [& {8 j5 ~5 w5 z/ J#include <vector>
#include <exception>
; M# l) @: B' W9 k* b3 S#include <functional>


0 o' u! C9 Y" I9 v

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
, a) y, b1 z" F5 `" jtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
( y1 D1 N& R, S8 G! r8 ^typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
) m- C6 {) K  O1 V+ a. rtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
( }2 c  l4 E: R$ y# Itypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
2 I* M( d4 y+ P% @/ R% ]typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
' r& y9 v1 D5 t0 n

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
2 R5 V  V8 m) {! I/ L);

% v6 {% t% P5 P. O& h8 X
typedef DWORD (WINAPI* TGetIpAddrTable) (
' Z5 T; J5 X' Q8 \  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

8 a2 A9 W) A( i" e" k
# k1 A9 z6 ~, B$ _$ Q! R- Utypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
* e+ P2 o' U9 _/ t7 [- M0 V! r: e7 h
* t1 f2 y, M& b. S* T
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

& S5 L4 j' c" ^
class CUPnPImplWinServ: public CUPnPImpl
/ Z* g, m& g" U& m. V/ g$ v{
% B, ~7 d. r  [: Y1 f        friend class CDeviceFinderCallback;
+ C9 D- }2 I* t% Q7 `! o- @7 V/ O        friend class CServiceCallback;
// Construction
public:
! M$ Q3 ?0 b, j% z0 d        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
7 z4 L6 p/ |5 Q& t, ]3 E
  w) `4 N- o2 D% D  E
* U, g& c: ]2 W2 p        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
. g4 q2 L6 a2 J        virtual void        StopAsyncFind();
/ T. f  b* O$ M5 I) P        virtual void        DeletePorts();
4 U' s9 j; B7 ~; K8 V3 k        virtual bool        IsReady();
: i8 F1 p- L1 @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
! N+ M6 r; ]* u$ F

        // 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
2 }9 d9 ]% v3 {# n        virtual bool        CheckAndRefresh()                                                                                { return false; };
$ t( ~7 B( A! e4 N" H) F" ]  N% F

  J9 A2 c' \# V: ^& bprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
. {: i/ W+ [: }4 h. m        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
$ |9 p1 u, {. h% E" h$ p        bool        OnSearchComplete();
; c8 t7 W$ K. B' |3 V; L        void        Init();
+ R0 B$ }; T( S! g5 W, @

        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
3 \" Z# ]& ?8 d8 T                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% Z: \) x) j* m6 D; t                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
  A) I) Q- \% s! G9 `4 B: ]                }
: v1 E' R5 L# L7 [                return m_bAsyncFindRunning;
        }
, C) U9 }; F0 d' B+ T- K0 P

, @; ^" W' z% M6 D$ W        TRISTATE                        m_bUPnPDeviceConnected;
9 W( r8 o5 O; o7 V

// Implementation
        // API functions
# C- {+ Y6 [* {5 M3 [        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 I) }+ k' W( E# a* W% F: Z        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! ~- F: J3 z: W! k( Q        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
+ z  T  t4 P1 d. s$ Z5 r. e        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
: b! i, O6 E8 F% x$ h8 k% c        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

5 C* @# D8 s1 S, p$ m7 x. [0 E/ i  ~
1 @2 i/ [5 k" Y- j/ d7 |. }9 I3 L        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


# X) _/ J3 ]( u  ^  m  N        static FinderPointer CreateFinderInstance();
) x7 M, M+ U; K8 ?" U3 X        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
4 q' J! K& p, w! [+ e                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
8 j) H% E' j5 L! \! g5 ^                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

" `- ?' L* r0 Y$ X; ^
                        if ( FAILED( hr ) )
. U% i. h, K; ~! i; j0 h* M                                return UPnPMessage( hr ), false;

9 B0 w: a# W* w3 S! g* W; q3 H
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
/ Y7 p" A& j: L1 E5 @0 U+ ]                }
                CComBSTR m_udn;
        };
       
2 H/ }' F/ c2 O0 [$ I        void        ProcessAsyncFind(CComBSTR bsSearchType);
& ^) I/ F: \" h        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
, o2 y1 ~0 B* s7 d/ X  ]  I3 |        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
  ?+ y1 i4 ?3 C- ~* S3 j- N, ^                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
) W8 N8 Y/ t2 `+ i& J
! z  V" u' }2 W* M) k3 k: v
: @% \7 D; }) B" r* X  R) C        // Utility functions
        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);
* b5 i) Q1 _+ D        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
7 O. w6 y5 o& X% `- E" }3 _3 E2 p        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
# V) F' \2 `4 N! b1 e5 a
4 q& q8 K+ m% g) F. _3 J) o
0 L& [" U! ]' `. O7 a5 n// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
) `; W" b. {) C& K        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
: s+ s* |5 D3 K& C* x6 \( j. d1 Q        DeviceFinderCallback        m_pDeviceFinderCallback;
; O$ V" D0 y$ v: ?: L4 S( h        ServiceCallback                        m_pServiceCallback;


2 O1 c9 ?' x* v        LONG        m_nAsyncFindHandle;
4 E4 ?9 i& z1 D        bool        m_bCOM;
- c& ]# q( X; ?$ Q4 \+ Q        bool        m_bPortIsFree;
        CString m_sLocalIP;
  D  u; [: y# |( M' A        CString m_sExternalIP;
- E4 _. a6 Y7 k9 @        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  e+ U9 \. v5 x0 ^% Y1 }6 _* Z        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
8 H2 {) Z5 q  C. J        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
& c9 ?' k* }2 f+ U2 ~( ?0 V3 {        bool        m_bDisableWANPPPSetup;


};

+ q" W1 u* i/ T
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
; p$ _- T! T$ g( @/ ]public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
5 C8 n1 \3 x4 u$ g- j- A6 z        { m_lRefCount = 0; }

' B6 r5 E+ p+ ^  Y
3 ^$ d5 ~; X& ^! K5 m   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
0 E. V* n/ B" `: K9 r9 x% Z   STDMETHODIMP_(ULONG) Release();

3 x0 \/ t' N  t6 k+ S% Z9 Y+ o
// implementation
- N7 Y; E: z: A4 v/ H. {, s& k% uprivate:
! o- T( }6 u9 j! T        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
1 L9 k# l' l: P4 H  \. ?( Q  {        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
5 ?3 P: P# z, W9 f) U
6 _2 K& G' O6 l. f
private:
        CUPnPImplWinServ& m_instance;
) [0 E: l4 ?! }+ }* s        LONG m_lRefCount;
2 b; k" L& O" e7 e: I7 ^& m};

, z( ?, I* A7 e- g# J$ h
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
6 a/ D7 M' e! E$ d$ U  y                : m_instance( instance )
4 D! o9 E" B* d% }2 B        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 W( T! `. u6 f1 ?   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
0 _+ T2 R3 _6 P  m9 E3 I
3 F( ]/ r# [! E* G* G  p
! y" H7 d7 Z/ g// implementation
7 w" [4 }5 `+ d! Z  uprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
- n, V; V& A9 Q) X* i, Q9 T! a        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

; \0 q1 z0 L5 T& i' ?% T" [
private:
        CUPnPImplWinServ& m_instance;
! u  }) F9 X! p! z6 e: t& E! q        LONG m_lRefCount;
};

( k5 ^$ H/ k, ~0 T+ A- r6 f
/////////////////////////////////////////////////


! M: i4 y* g3 Y1 V使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
3 y# }! c3 q8 H' W! Y2 v8 FCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.