UPnP

天云上人

/*uPnP.h*/

1. 
   4 V& s% H+ B0 d7 t. ]- w, f9 H/ v
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   / o- r. E# D& N3 r' s- `9 m9 n) B
6. typedef   unsigned   long   ulong;
   , e9 |, G6 K, M% Q
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    7 U6 l# V6 H8 w! e1 i! u+ h
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
    
17. }   UPNPNAT_RETURN;
    ; P* u( W# a5 v9 O
18. 
    
19. typedef   enum{
    : X! w+ t& Q7 W+ q4 n- }
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    , u- C8 v8 |: ?+ v* w' ^
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)
    8 }7 Q' `, l9 l# L. M1 x
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    $ u/ A: o9 C0 w2 b
31. MyUPnP();
    
32. ~MyUPnP();
    & U! v0 c6 N7 k5 l# G6 W1 H
33. 
    6 O. m+ W! k1 f) Y7 _9 u4 B
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    - `* K1 B: Q; f# g: o
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ; {4 S* P6 v5 ?1 W5 \( j
36. void   clearNATPortMapping();
    . ?) J7 @" b( X6 E  U
37. 
    
38. CString GetLastError();
    8 b" |- p: M7 d7 B- H9 o- i
39. CString GetLocalIPStr();
    2 ?( |% _* a0 y/ u
40. WORD GetLocalIP();
    ) i( R# j4 a. `, d
41. bool IsLANIP(WORD   nIP);
    6 ?" r! A+ x! K( q- G! e0 H# d3 t2 p
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    # X( [/ ^1 k  U9 a
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & Z3 i, x) o  V8 h
50. 
    6 _+ c1 h% @9 ~2 k$ m
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    " Z3 k3 f, R. t- u0 N9 S& `$ N
52. 
    ) I$ T3 b' B8 U( M
53. bool Search(int   version=1);
    , d0 [; Q/ D: O
54. bool GetDescription();
    8 Q+ }/ Z: ?7 u( y& l' }4 J) X7 r" O
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    : O/ i& [$ M' x0 T  h! k
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 l- B$ f: c! P' ]
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    . ?, Q. E" e% _4 g! C
62. CString m_description;
    7 q# v3 @  o: t- G, A; j
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    ) i5 _, v0 v# j& A" x- p" C6 q
67. int m_version;
    
68. 
    
69. private:
    ; A. e% ]* I4 l& h- N) e% U- _5 _
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    3 ^+ v8 A6 J3 s6 E. V" V4 q
71. 
    : g' v- H3 p* {/ v
72. CString m_slocalIP;
    8 z( _$ ]5 T1 |  b
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    : G( A# A5 h5 Q# B7 y1 _
75. 
    
76. bool isSearched;
    4 i+ ]9 z. ~4 `; A
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   " y  M( y* }% S) x
2. #include   "stdafx.h "
   ; X) y0 G8 `1 R. K& ?) Q
3. 
   
4. #include   "upnp.h "
   , p& y8 v$ ?' D- ?+ I% t
5. 
     t$ t: h% f9 _& @1 @+ D: Z% R
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   , j% g( V/ w8 E; ~
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
      v6 e! O! k0 h, x. O
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    , F4 G# _+ R! M& Y6 U: `2 {
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    + [( K: P! D8 u
22. return   s;
    ( Z% W, G$ Q3 x5 e( ?3 A: W; Q! X9 N
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    % R) v3 S4 c( @( }1 \, ^
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    % _7 a4 ?7 i; p  `# @4 y
29. 
    6 w# `/ K+ _* y- P" o, p7 }
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 n* |8 j. y3 L4 X# B, m
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    . B# \' v: y/ A9 w
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    . w/ N/ Y' h, d5 J
36. {
    ) h9 T7 W+ Z! |- T3 m, `4 m- d
37. char   buffer[10240];
    
38. 
    3 p" y0 }) X! {
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    . t/ W2 r1 q* `, |, L. N
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));
    ! V# ^; d; P9 b2 w7 j4 }2 x" T9 M
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    6 i  `3 a/ r7 f# w9 X6 S6 M) @
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    3 I% y8 G" T. \; p: l% F
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    8 v! q, E" S1 `, ~- j
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    . E) o+ B0 ~# R! {  ]+ a! N- l
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    , n; P" [% I( X( E1 p2 v& c
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    6 q3 P5 \# u# Q( B( X$ {+ c
60. 
    8 s3 f; S# C/ J. [( V7 |
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    & k4 ~: r6 W( W* b) b3 s
63. return   true;
    ) \# p; i% I3 v, n! A1 l
64. }
    4 I1 H' A8 ?3 T; B
65. 
    2 M1 j; z4 K5 o
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & F8 t$ E1 K1 L3 N* J$ @
67. {
    
68. char   buffer[10240];
    % p( b3 d5 s' Q7 E4 s. r; R
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ! E5 i& K: t7 y
72. strcpy(buffer,   (const   char*)sa);
    ( N" Y6 F9 o1 J
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    / w1 E/ I5 Z1 A9 \2 @& ?
76. sockaddr.sin_family   =   AF_INET;
    9 y) o# A2 P0 O- z7 |5 P7 Y9 X1 i
77. sockaddr.sin_port   =   htons(port);
    3 T  d3 M/ d6 |9 f
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    6 N7 H6 G; t. w" b& k4 N
79. 
    " f# S0 K0 x& @" o* E0 j% Y" \
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    / k6 W: ?0 C2 `9 M2 n
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    ; W0 C! X0 m, i
86. 
    1 Z+ `7 G* Z. g- \  N: @& a# R3 P
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    6 U1 Z9 I) x# W5 R
92. pos   =   0;
    5 x  B" m. V; c, R
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    % H9 n- F7 [. S* d5 R- ^
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    * P; Y4 ]$ T6 }: ^% D( n3 o' }
96. return   true;
    
97. }
    2 Y5 @! I$ H1 H& N! ^/ ~4 \: u. ^, ], N
98. 
    $ J9 Z8 {* l) W3 W2 ~; K
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ( o& n& K9 Z' \' `. }- n) C
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     " @( N8 n8 U4 z& o6 Z" j, A
106. if   (posStart <0)   return   CString();
     8 g* q2 b/ e, ]. ~7 L
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     - r( c) {3 ?& o, Q
109. if   (posStart> =posEnd)   return   CString();
     # {+ h6 }& l7 M7 S) ^, s
110. 
     6 G4 m# b/ P/ ~2 E2 G: d# W# {* p
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     + T- T) h. [) h2 q: ^
113. 
     / T. q! o+ i: e: M+ D8 h; d
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     ' }& J1 q% S+ z' v3 `( n2 Q, }
118. isSearched   =   false;
     ' i% B8 B2 k$ q; b0 k. i) ?) N
119. }
     6 e2 t% l6 z% k$ q+ ~
120. 
     7 u, _0 [6 g  N3 z8 E
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     , Y1 c/ p  t0 I* Z
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 |6 T( ~3 O; p/ P2 e  x& T0 T
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     & w- p7 z" v5 L1 {
127. RemoveNATPortMapping(search,   false);
     4 o: J7 J9 U$ q. t) ~" Y- ^
128. }
     1 H2 @7 E( E& s# C, n% E
129. 
     ! p- @, {: Q- B: z) ]
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     % `$ J- E) y" n7 v9 \2 w4 ^! y* M
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     # s- D: p" K. B- x0 q! T
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     ) m9 k& m* ^, W! Y2 x
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     7 n' A$ G/ G9 |: p
141. {UPNPPORTMAP1,   _T( "service ")},
     & \6 _! v1 K, X/ b( X6 H
142. {UPNPPORTMAP0,   _T( "service ")},
     7 h1 H/ W8 R3 B6 D
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     $ t2 \( B5 L- g* B  U
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     % ~0 N" Y: E& S( i( u' O; R2 H
150. int   rlen   =   0;
     7 X& O4 a9 j7 x  @4 k1 r: \
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     / [/ F  l5 @8 [& `$ i0 s
152. if   (!(i%100))   {
     9 E# |" z6 E2 v3 X, `
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);
       {3 B, F8 S, T- J0 `4 {2 i
155. CString   request;
     0 G$ b; Z5 ~! m5 A2 J9 v
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 "),
     7 U: m9 K0 Z: G7 U
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ) r: M5 k7 x% R% P+ u# P
160. }
     
161. 
     5 k1 s; [4 m" L+ T# S! X
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + h8 e& n5 I  ]# m
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ; C0 b2 \2 B( y$ a! g9 e9 G1 v
172. 
     " p  K5 v' ^3 i9 [0 D
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     6 t& c' k  N+ K, y
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ) Z# \( G* G& c/ w
175. if   (result.Find(m_name)   > =   0)   {
     8 d8 e: W, S4 h, U- l- p
176. for   (int   pos   =   0;;)   {
     : H0 M: ]* B. L: }7 x/ N6 D
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     5 J, f( Z9 r7 q7 G( p& v7 R/ R! U
178. if   (line.IsEmpty())   return   false;
     / e( a1 ]+ a/ s
179. CString   name   =   line.Mid(0,   9);
     2 `# r/ x# d* [' C
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. }
     5 l' k, d3 V$ H
187. }
     
188. }
     : x' M& \; M+ U
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     3 y5 ~" Z7 E( j0 ^5 ?

复制代码

天云上人

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

7 J& D: r1 H6 R' h/ D
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
* O3 x" k0 `! G$ ?/ F: l( d#include <exception>
1 V- q* {9 s* ]% B* ^# v* I

  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
2 H1 I) U* C+ ]/ N0 z5 }1 R& i! s8 T        UPNP_IMPL_MINIUPNPLIB,
  m$ m+ l8 y& ~8 r9 T9 ~, m        UPNP_IMPL_NONE /*last*/
};

5 C9 h9 H9 m3 a! V% l0 N5 ?, r


1 c$ g5 T; Q4 J1 I- F0 V) [class CUPnPImpl
/ P5 B8 ?6 t) Z$ H8 v. F9 ^8 j  x{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
5 X) a- b) _. S0 U# A2 `        struct UPnPError : std::exception {};
        enum {
  R/ U6 w# w" u- q* W9 O- |                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

8 R/ Q3 H$ R: e; o! S4 h
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
; k# A4 e$ O8 ~6 H* w' H        virtual bool        CheckAndRefresh() = 0;
& @) H9 b  _& y. ]% [+ u        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
$ y3 \2 T  r: `4 L) J0 a  Z3 r        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
: u9 e6 r1 i0 \: ^3 F: H7 S5 y/ {
' q3 X' n1 k% W$ O# l! C. c' m
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
6 j9 j: [4 m, N, I; ~5 e2 [        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
; V" v3 D5 t2 f2 L% G! C8 ~        void                                SendResultMessage();
, E8 m5 z5 z& }        uint16                                m_nUDPPort;
" P" X+ F; E/ p/ p- a        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
7 `. T, f4 J7 n2 I  }$ m
8 m4 s' N6 u' I* @6 J: P# f
};


% V, P0 }. q5 ~! o; t" p' E- p// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
, q$ s' J$ r, l{
, G3 S0 M" T. |5 s" O6 D1 [; \public:
! R+ _( O; E2 s' }        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
! k" ]3 e8 Y! X3 ~        virtual void        DeletePorts()                                                                                        { }
  p1 `" e) a  {) {        virtual bool        IsReady()                                                                                                { return false; }
! O. a& N1 q) h        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


2 U5 t( w6 V& u/ \# A/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
: ]* Z) R* `7 D! `$ F* \8 z
" O# B0 E6 h: U' T3 A9 }
#pragma once
8 w! |4 ^1 f. f& S. \#pragma warning( disable: 4355 )

* M" v5 j3 O1 x- [9 g
#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
& `# A; |1 |) d$ V#include <comdef.h>
#include <winsvc.h>

7 M# Q% c) S6 f  O/ K* q3 x; l
, G% D5 V0 Y6 r+ f' s#include <vector>
1 }3 D+ y) r% H* J9 n8 Z0 p#include <exception>
#include <functional>
* S. k9 C: T; Z5 y  N0 b6 k0 W0 }

0 J" K) d: C/ |4 \3 s. @9 ]

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
+ m  p8 W* l, X: c' X- o) Jtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
, l+ F/ @$ W- X) z) S# Jtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
3 S9 F9 B* t7 {8 W. Dtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


- N" I5 h: T% N. C- o& Otypedef DWORD (WINAPI* TGetBestInterface) (
. i4 X; b  y) J/ P7 |7 [  IPAddr dwDestAddr,
4 X5 E: y5 d; c" o2 o; U/ e  PDWORD pdwBestIfIndex
);
9 T( W* l6 }$ s& M4 \+ @
3 E6 I3 W% {# H9 f0 h; ~7 c( e
typedef DWORD (WINAPI* TGetIpAddrTable) (
5 |, H( D: V# b: I5 i6 Y0 ^# e  PMIB_IPADDRTABLE pIpAddrTable,
! P' j2 @8 S/ N/ M% f  PULONG pdwSize,
" v+ o# ?: Y0 d/ Q  BOOL bOrder
);
! V7 Z7 F* \& c2 j
7 f: g% e# g/ ^  Y
typedef DWORD (WINAPI* TGetIfEntry) (
7 s; M' A6 z% ~6 ^7 N  PMIB_IFROW pIfRow
7 R- k6 k& }' {' F$ S2 ]);

5 j) A1 o2 }( N4 d. C2 Q; _
6 U/ X  h8 N; W( v4 V4 B: vCString translateUPnPResult(HRESULT hr);
+ [9 @+ \. v' x) R' P0 j7 D8 o8 Z% BHRESULT UPnPMessage(HRESULT hr);
" j* s$ W/ R5 c6 }
# o0 V+ K0 K+ D5 p
, q; ~$ p9 N+ l) n: Qclass CUPnPImplWinServ: public CUPnPImpl
! Q2 G( w0 }4 a( k9 o+ H{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
2 a4 W& ]2 [7 R  R$ H$ c0 K# ?// Construction
public:
; [& s4 w% H: B1 D0 ]! t        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
3 {+ J3 j& q( ~7 L9 u& b" G; e
9 f* Y2 S3 b: M( B# H( ^+ I# A
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // 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
* S. K' V7 q& X: j% C/ p4 x        virtual bool        CheckAndRefresh()                                                                                { return false; };
8 B( C+ L) r: a4 ?
& [# G+ H" O: R  f5 m+ T3 I7 i, [! f
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
# r- j; S1 z( ]        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
, w; s% Z' k9 `/ [5 d1 J        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
# d0 p4 t3 t5 \/ C/ e

        inline bool IsAsyncFindRunning()
( B" d! b8 _9 J* F3 [3 [        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
3 I& J' e- |/ s4 y                }
                MSG msg;
9 i& h# f" d& e- {                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% ^8 F2 M) J( e, A* _2 G- L& m) P                {
! s- A  N) u  T2 i0 Q( \' G9 y                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
) p6 @: m8 y3 V; u2 q                }
5 T8 X- [! j5 f                return m_bAsyncFindRunning;
$ h+ x- h0 D' }- H* M# M        }
2 g& b" E& L  i! K* i& o

        TRISTATE                        m_bUPnPDeviceConnected;


; x4 ^0 c' H2 v% N// Implementation
        // API functions
, k/ ^" q2 m5 S0 h9 R        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
5 W6 S% K9 N# s: V) G; t1 N        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
: l  a. S$ C4 a4 S" W        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 z/ ~+ j9 F4 i- o8 m* Q% [        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
1 y" X) Q) ~  g& z' [0 ^( y% p6 z# P

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

2 @8 K  B( L% C) ^4 h- C& Y2 f
2 p5 E# m. d& \# u5 U; x. V        static FinderPointer CreateFinderInstance();
9 c2 ]3 N, x$ j  R1 H8 m        struct FindDevice : std::unary_function< DevicePointer, bool >
3 F/ o& F1 t) {3 E( [! ^' K7 S" _        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
$ N) y- l* j! Y9 h8 i4 b3 g                result_type operator()(argument_type device) const
( B7 v1 _+ D( {; u. a4 Y                {
* z6 s* c* K' [/ Q' f) p; ]                        CComBSTR deviceName;
- B$ Y0 Y0 t% O9 ^+ X" d' V* e                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
- P$ Z% F! G2 C$ H# D1 B  A                                return UPnPMessage( hr ), false;

4 R/ U9 K' l' _! h  u6 W
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
" S1 b) g! D" M' _$ D5 R9 @        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
9 o4 [" q# G0 T9 P/ \        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
2 [# u8 l1 X5 _        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 @; e0 C2 j: X. c8 Q" ~5 R* k                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
2 x) x: q1 @( |0 X" f4 r! @

) P+ p( J4 c( ]% V1 `- ~# k8 f. z        // Utility functions
- p4 R; `* F& X; @4 x1 N' f        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);
1 `4 }2 E# d- q7 m4 E, v2 g        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
( c, u& w5 f5 [        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
% U3 g! w/ y- l/ n- C

6 C7 ^3 t2 {) z4 [) V  W// Private members
private:
$ g4 Q& z0 Z  F8 K# o+ t( q        DWORD        m_tLastEvent;        // When the last event was received?
4 A0 h6 M: {8 ]/ P        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
/ D; f7 w4 [9 R        DeviceFinderCallback        m_pDeviceFinderCallback;
/ k- B3 z$ R3 D        ServiceCallback                        m_pServiceCallback;

: I# |7 P+ T: K6 @* g8 c: A
, s$ f9 z0 h, v2 [        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
% M* K/ H" D* a* W: L+ P5 ~) [        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
7 g+ D7 h/ w  x5 Q$ E. A        bool        m_bInited;
/ C8 Z& b$ }4 L# j/ u        bool        m_bAsyncFindRunning;
, b  a9 i! ?2 |9 }, E5 R. w        HMODULE m_hADVAPI32_DLL;
. ?. j/ S  j$ A        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
. W7 Q0 ]3 A9 D% t        bool        m_bServiceStartedByEmule;
2 h" T# \( i3 h7 {        bool        m_bDisableWANIPSetup;
! @) d* v/ E1 P6 B6 c4 d2 C& b2 p        bool        m_bDisableWANPPPSetup;

  u1 ], M& v7 l, W. t/ Q% X
7 o# _& S7 V- {" L# ]" K* [};


: ]4 F5 j/ g& K4 {// DeviceFinder Callback
  u1 k5 [3 ]" B$ Q8 u! mclass CDeviceFinderCallback
# d! P( |. N, V- T2 K5 T6 S/ n) p        : public IUPnPDeviceFinderCallback
' g: E7 @0 M6 Z$ [{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
. c, t- k' r( n" P( ^. B                : m_instance( instance )
        { m_lRefCount = 0; }
* ?; W# f1 e+ k7 |7 i4 V% \" h
" J# Z4 T, r6 A
2 b+ r1 q3 Q' h2 }   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
0 U: I3 B8 r0 a) Z& ~$ T+ k
) l2 Q* |' ~( H% X$ o
2 g) ^5 ]# r  N( C' m! b( T3 t// implementation
" [6 r& r  @3 A' ^3 z7 Vprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
( P$ l/ ], l' z) d0 D        HRESULT __stdcall SearchComplete(LONG nFindData);
  s3 k: t2 _! w) G# f9 T0 z! K
" t2 ~# l& V% T0 S0 j
private:
3 H6 ?% n  i* s        CUPnPImplWinServ& m_instance;
& Y5 S/ V- P7 P% ]  u6 W3 Q        LONG m_lRefCount;
2 x! U1 p9 _% `1 f6 x7 S};
' ]) u7 Q2 a+ }  F' d# Y! p

// Service Callback
: w" A% i2 _7 I9 Iclass CServiceCallback
8 s1 f  c6 L% M' e  ^        : public IUPnPServiceCallback
{
, {% P: x! @) X4 C- mpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
2 d! K3 t6 H/ U* ?; ^: g4 y                : m_instance( instance )
        { m_lRefCount = 0; }
   
. `3 n+ z& i' W' n% ~! I# R7 s   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
: j' l7 l5 X" @! r

// implementation
. j2 h+ @8 b$ V( wprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

5 z4 S0 O/ Z- H  U3 R- F
& l8 V; K) c3 C8 M) ~private:
        CUPnPImplWinServ& m_instance;
' ]; S# K& o  H$ z( |  H3 d        LONG m_lRefCount;
};
* \- b# ^) H  O# |) S

; s+ c$ O& E3 e/////////////////////////////////////////////////


! O3 |! y* Q" P) h6 c: r4 y使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
% q1 i- s2 H; i. k9 DCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
- e! B" x5 j# Q, X' H. fCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.