UPnP

天云上人

/*uPnP.h*/

1. 
   8 a' x0 R, e) N) v, B8 ~3 o
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
     t7 K! O& H, w
5. 
   6 S9 ^4 V* k' V: ]) H- k
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   / {, a/ _0 d7 d7 W+ S4 t
9. {
   6 V7 Z. y, G4 d% w
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    2 b8 H+ X" P6 J6 t0 r
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    / [! F, r. W2 K7 S7 b$ n$ R4 y% I$ }
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
    0 @4 I7 r1 e. l- b/ M+ n3 D
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
      a; C' X8 c: E& Z
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ' {. z# _; Q& L4 T; z) x8 J
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    9 n7 v7 p1 h3 [6 g, u
24. typedef   struct{
    + t" g' e6 M% m( P6 Z5 _
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    # E/ S! k  z5 c) T" b. ^
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    : J& Y8 U% ^! I# t
28. CString   description; //   Port   mapping   description
    ! Y# w% a9 d( h4 y4 C# s
29. }   UPNPNAT_MAPPING;
    9 S' r; ?) a! l' l/ J6 l
30. 
    
31. MyUPnP();
    , w0 o/ N3 i# I( O& r/ F. ?- i
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    4 _$ r4 g/ x# x4 N# k7 `0 a
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    % Z; H! |- O$ ?5 G( g6 T1 _. S  x
36. void   clearNATPortMapping();
    
37. 
    & h  P% m* O! _3 s
38. CString GetLastError();
    9 u8 j$ A7 I6 m) [3 j! n9 y) W  {
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    " ^5 g9 w, z4 Z2 Z& T0 f
43. protected:
    " x6 h4 s+ C9 I6 y$ ^
44. void InitLocalIP();
    ( ?' a( L6 N& z6 I( v
45. void SetLastError(CString   error);
      z$ `9 t) r" C2 q( n3 O
46. 
    % R* u- _, F3 O; [4 D0 O$ K& e
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    1 L& @; h0 ]; k+ ~& v/ M; S
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    - B$ u1 F2 [" p. V8 s
50. 
    ( C8 ^6 v6 }8 C5 Y
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    7 n+ w; K, U/ X: _4 E- r
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    2 U6 A, H3 z7 c) u- a2 h2 N
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);
    . B; ~" |( E- I* X
60. CString m_devicename;
    
61. CString m_name;
    0 p2 a6 ~6 d; ~4 }/ f+ o( T+ E6 p7 B2 ?
62. CString m_description;
    / Y1 p$ _8 {; X
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ' P4 _; M4 z4 T% X2 q1 _5 D& y
66. CString m_modelname;
    " ?6 ]8 h# ^, L- g
67. int m_version;
    . I+ i/ E6 s9 s; ?3 u- h
68. 
    
69. private:
    3 I2 `7 ]8 A. h2 x$ P" d5 e) [
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    5 b4 F# X: t3 D; ?$ I' Q- x) L
71. 
    ! t7 C( T, C1 O9 h- X6 Y5 @, A
72. CString m_slocalIP;
    
73. CString m_slastError;
    , L  N8 ~6 P4 X4 O  A5 K- y6 N
74. WORD m_uLocalIP;
    4 T0 s4 M$ D6 R  w4 I0 G1 d
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ; J" h: i4 |# h/ W
3. 
   $ _' l( @; m+ Y' c6 Q
4. #include   "upnp.h "
   0 N4 p5 \# M) s) K' W
5. 
   ) t. S- @" [1 r+ g: [  t7 `2 ~
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   - ^; W# r! V: A0 o
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
     x! {* b# f9 g2 Y6 G% D2 S3 b
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   & o9 S8 P! Q  m; x& g# _/ Z
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ( K# v' L$ h4 k: k# G
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    + _( Y4 q6 }' }, Q5 V) y# n
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    ( h6 x% S' B. O6 z
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 ], `% I6 S4 J* y8 L
26. {
    ( a  T2 H# i' O: v9 f
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    * J: {6 b  A& E  J0 J# P
28. }
    $ ^& }8 ^8 e5 Q! `0 Z
29. 
    ) u. F( Y  Q: p$ ?
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    % a% w( r9 ]3 u1 q$ P" E  }+ ?
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
      f) P( V4 ~' s' t0 \) |5 {
37. char   buffer[10240];
    
38. 
    $ i' w9 U+ H; e
39. const   CStringA   sa(request);
    * a# v4 a( D. Y: H0 m* a
40. int   length   =   sa.GetLength();
    - [' u# d0 F. `; }' E2 M
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    . M2 o2 Z. l" e" p% g
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    * _! l) w4 S6 O6 p# e
47. sockaddr.sin_port   =   htons(port);
    0 \/ y9 Q" L" B7 m. t
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    . c4 d2 A+ k. p- W4 F" V
54. int   n   =   send(s,   buffer,   length,   0);
      g( z8 n& \0 x9 w# u  i" B
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ( A/ \& }- j3 ~" G1 k
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    $ e8 c* l/ v( K0 W8 S: ^6 `4 b
60. 
    ' u9 L$ l$ Z! _$ I1 a: p6 n6 j
61. response   =   CString(CStringA(buffer,   rlen));
    ! p5 Q, A( l, \1 u! y) n: r4 n( ]
62. 
    
63. return   true;
    
64. }
    : x# M2 t7 T. c2 c- [  D
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ( h% O6 Q4 j4 G6 G6 C
68. char   buffer[10240];
    ! a+ ]# V/ m& A0 t$ v; e" ~
69. 
    4 v7 Y$ L- ~% `
70. const   CStringA   sa(request);
    0 r9 u6 @1 ?% Z( X2 P
71. int   length   =   sa.GetLength();
    2 h/ E6 @( `7 P) E9 J8 D& H7 M4 p
72. strcpy(buffer,   (const   char*)sa);
    & n  [1 Y, [" u( F) |0 D; C
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    2 R' F; \0 f2 O5 t& i- l2 F
76. sockaddr.sin_family   =   AF_INET;
    
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. }
    
82. 
    / s6 p5 P& |  S# L: w  p  f
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ' {0 v- T4 r( r/ @( m
84. {
    $ E& X( a$ p- c/ B
85. int   pos   =   0;
    
86. 
    $ ?% z9 o" \$ l7 D
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ) c8 |. V5 M# u1 D
91. 
    / q$ ]2 i/ _0 l6 h# z
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    " l4 H- h- g7 D; F
94. status   =   status.Tokenize(_T( "   "),   pos);
    ; C  {! d" {$ M* R/ V1 r
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    7 h! H0 U; c6 D9 e. @# A. B
96. return   true;
    
97. }
    # f/ q6 J7 l8 a* U1 g. P
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    4 i& f1 d& f' r! L4 m
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     + `8 I2 V* k( m! a
105. int   posStart   =   all.Find(startTag);
     - s* c0 _8 I$ `6 [3 x
106. if   (posStart <0)   return   CString();
     * Z& u" t7 y/ N) P+ M
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     1 q7 p4 |0 y& d
113. 
     8 B+ l  F) ~) w' ?. g9 {
114. MyUPnP::MyUPnP()
     9 c/ T4 e0 Y0 w, x4 B$ V: `
115. :   m_version(1)
     
116. {
     7 }4 h2 N/ M7 R( o
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     $ |) Q* v  d; \6 d& R
123. UPNPNAT_MAPPING   search;
     2 D  w2 R) y- n% k% ^( z: R- s; L
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     * D5 D& ~5 q, ?6 n; T7 H
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
       ?' v4 Y4 r* _4 C
130. m_Mappings.RemoveAll();
     : E! I4 [; f8 T% y1 T3 U6 N
131. }
     
132. 
     : h: D9 k. c7 P1 p' Z
133. 
     ; e. ?1 |3 \" W4 x8 h5 r  D) F
134. bool   MyUPnP::InternalSearch(int   version)
     3 W% w7 v5 S; s% r' }" X
135. {
       l' R3 Q1 X& L( N4 j
136. if(version <=0)version   =   1;
     2 e# e; |8 h, S7 A% z) U* W
137. m_version   =   version;
     + b( f# T5 `- m1 F
138. 
     
139. #define   NUMBEROFDEVICES 2
     # o$ z7 [6 p! v. \
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     2 Z/ @  E# x7 t7 H8 i
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ( o1 m! l: ?9 }8 d2 e+ }) K
145. 
     ; b9 h$ B3 N4 Y$ J' h$ x8 P) A6 ~
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     - P1 B% J) V/ R4 K
147. u_long   lv   =   1;
     + |% f9 i4 q. }3 D3 A* f( F+ T' v8 c
148. ioctlsocket(s,   FIONBIO,   &lv);
     & h/ J# @. o( {1 d
149. 
     2 r# R$ x) G6 V* ]8 x0 N% t& b. @
150. int   rlen   =   0;
     ) A9 f; E/ V7 @" w3 W+ R
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     2 {9 n. y$ u4 n: S0 M; v
152. if   (!(i%100))   {
     4 W6 F$ e8 f- T) A
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);
     / v! \0 l% {8 X8 I
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     + F# w0 ?9 O" q' F
159. }
     1 z" f0 J" f: e
160. }
     
161. 
     4 b# b5 _- N: p! m9 S
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     / ]; ~% \% o% A% W) o
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
       A. h  f! O( C8 H, @0 D# D
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     9 ^: Z( k1 S1 A3 C4 ]; E; Z
170. CString   result;
     3 G' y' F3 q1 G
171. if   (!parseHTTPResponse(response,   result))   return   false;
     5 p% D' {/ Q" W; h' o8 c8 j/ `; a
172. 
     $ X/ W0 n& G  O7 L) K' z! N/ Y& s
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     % }# G9 L% r& o3 @& j
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     1 r7 N( J6 l2 c4 X
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     : Y8 X# R; H" n  n2 u7 r
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ; D. k: ^; s# L8 l+ R) E
186. }
     ! ?6 s$ }" t9 j' N- y
187. }
     
188. }
     
189. }
     
190. }
     . j3 e) K* ~- ]: C1 `
191. closesocket(s);
     & f% L7 U; @( s0 ^# B& K: q3 F
192. 
     2 C8 O4 x6 Q" ?9 J/ Y$ h
193. return   false;
     
194. }
     3 u! f8 c5 r1 Q+ s

复制代码

天云上人

以下有关upnp的接口来自emule，
$ L4 V7 e$ w( o4 x

( T. J# C/ L* }8 L, Z5 U; K///////////////////////////////////////////
6 I- t. S1 B4 h7 p+ O//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 N- F+ j" j1 `% X/ i1 g* x
8 |# q( |2 Z8 [% @. c. W
0 j! i) M6 ]$ l" l  H  l# p5 h#pragma once
#include <exception>
5 t; u, z) ]2 H# k0 L* a

  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
- ]0 U2 I; m! |9 H3 M5 n5 B1 m};

2 t  Q2 R  _8 @# C* M% ~2 q$ H
% {  k# d$ L5 o( \2 c3 venum UPNP_IMPLEMENTATION{
3 l; e0 @! L. t) V. {* o! a! F        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
* U7 s$ K3 X4 Z- k/ i  c        UPNP_IMPL_NONE /*last*/
};


5 E0 [9 A  }' }; X& |2 [
) U' {- F5 n" S
class CUPnPImpl
{
2 A2 M3 M5 e' v+ T- z/ fpublic:
        CUPnPImpl();
" _& j' w. c0 A- }. u. l        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
2 z1 y2 q* E5 Y# y, d/ x0 v& Q7 `                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
8 ^7 R3 m) A! P6 Y        };

& }& Y6 p& ~% `; v
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
* |2 x; I: }# x; m4 J       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

; ^  U9 Z3 B7 F0 P  o! X% T3 A
9 I% Z0 l5 S* _$ p        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
1 v1 g0 Q/ m8 y8 [# a        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
! ]& Q( r+ w/ z/ B        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
, D& w# i9 f& e! _0 @$ s. l# x! }
* O+ ?9 R# h" O  ]4 O& u
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
, d% v, ~, S$ e        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
3 ^! u0 }0 l; L$ Q        HWND        m_hResultMessageWindow;
, C1 y  z# c. F; G2 ]* |) C6 `        UINT        m_nResultMessageID;


};


9 J8 F: [( w+ P' b! G  h// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
) P: I* [* ?# Z$ v4 Q        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
8 |+ a- H1 V( k, R. T        virtual bool        CheckAndRefresh()                                                                                { return false; }
- U- g0 L9 J4 V5 x( }5 Z        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
% |: i3 i' ]1 u- g7 b        virtual bool        IsReady()                                                                                                { return false; }
0 ^! j! r8 ]4 u6 W0 ]0 X/ m        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
/ K8 n3 I+ D5 I! x};


/////////////////////////////////////
0 _2 I: @6 z5 T//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )
. M% C8 c$ V) U5 z2 @) a3 h' t

#include "UPnPImpl.h"
+ f0 g' T# |  \6 K" L#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
- m$ D5 e  T% @% ?# n( `# t

#include <vector>
#include <exception>
( l; B7 r5 z6 R2 d! h#include <functional>


% O6 O; H3 z, T) _( F, l5 y; b- ]9 A
$ ~; x: u  N( V. U
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
& o3 T- k9 N2 z) B' [  Ltypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
  D- g1 s6 V& M3 ?6 P2 Gtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
- G+ c( z1 K, N7 b/ E! W( rtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 C: a' V9 R* s) D6 S! K" Mtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
( X4 Q  J) s8 j9 W% [typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

2 `" Z" K. p" n, B8 h! S" }
typedef DWORD (WINAPI* TGetBestInterface) (
: y$ n, _9 \- t1 c  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

  n+ K8 E0 {5 {7 `+ K3 f) z
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
% F; {4 z$ s6 _, O  BOOL bOrder
3 k* p# U( p: q3 l1 n. P; E);

$ Y2 k9 V7 ]" J9 `4 q( ^. M
. o1 M1 A: X; C: `; @* }* {0 Jtypedef DWORD (WINAPI* TGetIfEntry) (
: K5 _" h& `7 v' O  k1 L% p0 ?  PMIB_IFROW pIfRow
);
$ j" }, U- `- I7 s8 X0 e5 g5 C

' q6 N7 P; F2 NCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
# S3 M5 b' y0 R+ a. }

class CUPnPImplWinServ: public CUPnPImpl
5 e4 R0 _, h! v, T1 J4 i{
0 C2 R: h! G" [$ G, |4 @        friend class CDeviceFinderCallback;
7 h; f% B. \' b3 u5 x5 ^        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
- k# m  |* C7 g& a& u: ?  {0 D0 |        CUPnPImplWinServ();
1 I  k% h" t/ |* z

3 |, u8 x2 v# W1 i) F$ t: ^        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
3 f  ]9 d. G: P) b0 c( l* P3 H9 d& Y        virtual void        DeletePorts();
- x% K' h+ B1 d! C) o        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

) w- s4 l) @4 c0 U4 o$ V, P/ a4 A
        // 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; };
5 T+ k" P- E$ W  }5 W' k  @8 F
; M" P4 @9 X1 d& q- }7 h
, K3 s/ Z# w$ t# N% S  @protected:
1 c& C" @4 q4 [  z7 p# H% s. L+ u        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
# d0 D# z8 J, z/ [. n        bool        OnSearchComplete();
        void        Init();
, `! J; D8 `4 r8 E! N; i

        inline bool IsAsyncFindRunning()
        {
% Q/ Q" l- g# k  T  e                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
' I' c% S/ {7 e1 p5 I% L                {
. R# q8 Y9 N( u# a7 b. L                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
9 Y- ?4 |1 \" E: {                }
                MSG msg;
& D1 v5 L/ B$ P                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
9 D# P( ~; R9 \! f9 M, q0 s                {
2 U, y3 l! M7 X( l- H" Z9 T6 Y5 r8 H                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
* @! Z! E0 ^$ C: C                }
1 m2 \; I' s) i0 V8 \                return m_bAsyncFindRunning;
        }


  s7 @" i+ F6 G% g/ o9 f9 l        TRISTATE                        m_bUPnPDeviceConnected;
1 ?% a" {: d- d9 I, I! e# U7 P

8 U. V7 y5 i) z5 I// Implementation
+ h$ r/ x7 X0 Q0 s: _        // API functions
4 J1 e1 }: z$ U" L        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
# g! U9 X# }) t" \" ]+ R; F        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
. H5 C; e4 Q. o2 l+ L, s7 I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
; t2 p* ^5 r  w6 ~: P( d

) v! X8 @5 u7 N/ u        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
, v7 p; {# y+ s6 U$ a        TGetIfEntry                                m_pfGetIfEntry;
# o6 E( U1 [  L/ \$ c

        static FinderPointer CreateFinderInstance();
) z9 S6 ^% ?+ k  {$ V2 i/ P% S( Q' O) r        struct FindDevice : std::unary_function< DevicePointer, bool >
, E2 X( r! @# C, O% d        {
4 Q0 b6 |, P& l; L/ g: c                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
& s% B+ a- w/ d+ B/ {0 }0 `2 \* a1 _                result_type operator()(argument_type device) const
                {
% g: _+ d2 Q+ _* ~3 o6 @& }" ~                        CComBSTR deviceName;
/ F8 j" L7 P# ]2 @                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


5 W) x! i8 ?# O$ q) q                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
; k/ o# z( t* f$ C6 c5 A8 [

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
; w) w/ y; ~, `; c# \                }
1 n/ z1 D6 r1 l5 }                CComBSTR m_udn;
        };
       
8 E8 U6 X( A. ~5 z6 N. Y        void        ProcessAsyncFind(CComBSTR bsSearchType);
( O" a: _: }2 K, V" m% O$ ~( c2 j        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
# {2 a5 k/ _4 a. U8 ?        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* l2 x: I: N# A; K: i$ _3 A                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // 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);
" q9 S$ s" d: C. w        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
* p- R- ~! v+ g" C2 v$ M        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
9 ?2 y* d; v) j6 y  Y! u, ^# s, r% L        CString        GetLocalRoutableIP(ServicePointer pService);

& l: [0 _. \3 `& P6 A/ m
& n8 Z$ F4 s. n9 m" d  G// Private members
/ k& {+ E, L/ e# s  p: Oprivate:
& D& J5 J1 ~/ X7 b9 N9 j        DWORD        m_tLastEvent;        // When the last event was received?
% @+ w7 z0 w  b3 P% d1 x& G        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
2 r4 z. k4 v! E' Z  F        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
  K# V. u# w4 z) {" g
7 d# K5 L. K4 d! }  Z' g( B1 m* n/ ]
        LONG        m_nAsyncFindHandle;
+ L" D1 w/ {) I7 x8 I        bool        m_bCOM;
0 u9 f3 V+ D# k! w2 W" I! T        bool        m_bPortIsFree;
/ ^# A; w5 h. I( h% A        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
. l: {/ W3 p1 m+ F5 h        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
5 W- f" G# ]2 c        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
" ^- K, J4 K! z& i" H' u
: c, X4 C& V' I+ c* h" Z
};

/ S3 S8 s# J) j9 G% s" C
// DeviceFinder Callback
, S, ?' |, ?( {6 v4 hclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
1 [. U9 U$ [$ e4 b3 q
/ z8 g4 e' a; E& z! V
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
) L( z- G6 g4 W4 p& m; h   STDMETHODIMP_(ULONG) AddRef();
# h# C* X% |1 `* p" C   STDMETHODIMP_(ULONG) Release();


// implementation
private:
5 `, r* p1 O$ ~- |        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
5 q) ^+ X6 y  c3 J        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
: t* D9 k7 p: C9 A+ F& o4 P& a        HRESULT __stdcall SearchComplete(LONG nFindData);
3 y  |# b) t# Z8 n4 s: T( q

6 H6 I; z# l7 g' m( zprivate:
        CUPnPImplWinServ& m_instance;
6 T6 W. G1 d6 [. c+ y        LONG m_lRefCount;
};
. B" j2 ], H+ }/ D8 h2 ^7 U

// Service Callback
; A! O: \% x! Y$ C- A9 M7 Aclass CServiceCallback
; Y* R( a% V( z! x        : public IUPnPServiceCallback
{
8 }( |& W! l( j2 [& dpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
* y7 S* e* S: a: y  p+ b                : m_instance( instance )
        { m_lRefCount = 0; }
0 Y  p8 K  R/ X; y% G   
) w5 m" i& g- W6 `3 {% t) w  J: e   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

, u. N+ w  e5 F; }
5 U4 @+ ]1 R  k$ y// implementation
' N. Y% D% }) \- vprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


5 G2 p7 Z+ Q6 s9 Cprivate:
! n( f. _9 U' S/ @9 A. Y, m: K1 C7 n9 P        CUPnPImplWinServ& m_instance;
% X" a$ w$ N1 U; t' f( A8 G- s+ g        LONG m_lRefCount;
};
3 ~# A6 W' o' L7 n
2 V8 [, A& w, }% C1 [
/////////////////////////////////////////////////

- N& r$ L" P0 F7 g( [
6 \* u% S/ W3 L- m使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
& S6 A& V7 S0 kCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.