UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   9 Y+ C5 J+ p5 C3 @" X  f' P
4. #pragma   once
   % s. ~( {" V% Q- l, a
5. 
   4 }: c, p5 U' x4 H" }: K# ~( g- X
6. typedef   unsigned   long   ulong;
   + s- t, f; a3 ~. c# c
7. 
   
8. class   MyUPnP
   ! p# N' O5 ~/ a
9. {
   
10. public:
      \* y; T( Q. P2 Q" J+ `
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    9 I. Q/ E+ U& W) x
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    / T; i$ T; `9 ^$ D
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    . ]0 m/ w; x6 W2 [
20. UNAT_TCP, //   TCP   Protocol
    4 ^, h+ ~1 q7 R6 X% r7 E
21. UNAT_UDP //   UDP   Protocol
    3 k6 Y7 C! \" ~& a
22. }   UPNPNAT_PROTOCOL;
    
23. 
    & z% _' i1 b7 l/ T; w) m
24. typedef   struct{
    ; l1 a$ g. r/ ]1 E9 C) a
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)
    
28. CString   description; //   Port   mapping   description
    : P6 B$ O' A4 q0 M" E$ Z5 d# I
29. }   UPNPNAT_MAPPING;
    - {& a5 K$ w8 ?# X+ F: S
30. 
    $ ~& B6 C: T: K
31. MyUPnP();
    ! h7 ]9 \% u/ C* E+ ^. q
32. ~MyUPnP();
    8 x) W% ]7 `$ ~; E
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    : h+ V' M# C' A( F/ G1 k- b
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    6 y6 c' E7 L  F+ X  W4 I
37. 
    ( R6 F6 E, ~$ s* ~$ S- P/ N
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    $ z- Z/ X/ O% W
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    5 d! s, Z7 {3 C/ ?) l
42. 
    5 ^9 N, d- l7 }1 W& u9 f- ]
43. protected:
    $ E9 ?. {3 P9 j( Y$ R
44. void InitLocalIP();
    / E% X$ m6 L9 G, |. P$ D
45. void SetLastError(CString   error);
    " b& _, N3 E7 ^  t! p2 Z
46. 
    6 q1 L  Z- |' K3 u6 }
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ; x' w$ A- d* d/ ?) |) G- C' D$ ?
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    2 V% L$ X( k  e- U+ ]/ o
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    & V$ p9 {' C8 G3 l# c* ^8 O
59. bool InternalSearch(int   version);
    ' s% r+ J7 R$ G% w8 J6 R$ h
60. CString m_devicename;
    3 n/ G! Y7 O4 m7 s7 V- s/ {
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    7 b9 S/ h5 w: }
64. CString m_controlurl;
    5 Q4 h. u# w0 {: a6 _& `; L5 G
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    + P* e/ Q" V1 ~7 x, m# Z$ x
68. 
    # R, [1 l8 x: t6 g
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    0 r( u% {) v' [! X+ ~  Q$ o8 L, v
74. WORD m_uLocalIP;
    
75. 
    7 G6 T$ v% s' T8 k8 T' c$ t
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   : o+ d" G  O2 \/ N- M. X6 C# P2 p
5. 
   7 r  R2 P4 z& ~. S
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   1 Z: {. b' h) D! H7 b) k
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    2 Q; G( K3 ?* M
11. 
    ( g+ {* E, G: l5 X+ n
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    $ O, I" x/ u  i, P3 F
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    0 A6 W* K) A# @
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    % c; M7 X% p8 K3 N( l; i- e+ L  i  q6 y& Z
19. 
    % ?+ S4 k) D5 W. X. w- i0 W1 s* q
20. s.Format(_T( "%d "),   i);
    & r" J5 d# }# L4 f. o. e" H
21. 
    7 ?8 ]1 X7 b2 a8 V( K7 N% f# u
22. return   s;
    ' |4 r: r) I9 b6 K
23. }
    0 w4 ?1 |, g  m) @
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    + Q0 G8 }7 z. r8 A/ g6 g
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    1 Z5 t: k, S) m' }
28. }
    1 C0 i* C1 o3 f) f
29. 
    ( J8 K  ?; c! t% l9 ?
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    3 O- ]! Y* Y- O# o- K& @6 ^
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    % N3 y# v# ^7 G2 N) c7 q
33. }
    1 a7 m$ T( _! r0 ^" [
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    $ g, ]9 D* ?, @! N, n4 ~8 K
39. const   CStringA   sa(request);
    3 r7 T; M" M4 J! p* x9 |" ]
40. int   length   =   sa.GetLength();
    ; h- M4 c; u: z8 B' l9 F  N( I5 {) `9 w
41. strcpy(buffer,   (const   char*)sa);
    $ p  [, c8 f' B- i# F$ Y
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    # u2 P0 j6 N: R
44. struct   sockaddr_in   sockaddr;
    # P+ n( g. T" ?, |4 J
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      _" y7 A9 q( Z% N5 t: T
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    * ^" d1 Z, {9 g/ s1 h# c
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    - h) i' d3 D9 D) ?6 L# T
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    & y+ z- N6 ?' ^5 ^
50. u_long   lv   =   1;
    + V, G' Q; K3 v, h, W, I
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    / k: N% g/ o, j6 e7 ^+ q1 t4 {3 z- P
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    6 _3 R+ J) ^  @8 J' T1 \# f) {
57. closesocket(s);
    + c; I8 a% k3 u# p( g
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    7 H+ u- i/ ]) u/ ^: V
59. if   (!rlen)   return   false;
      D0 M( l& c7 `+ S5 N
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    5 I- ^  l+ M  ]+ g; W
64. }
    4 H2 r* t( Q  d) ]/ ~
65. 
    : X6 z, [7 K; a4 G* F$ o
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    # t) K5 i0 T/ F3 s
67. {
    
68. char   buffer[10240];
    . L% Z( v* J% g1 t9 d6 M4 u5 p
69. 
    - p7 J+ K. ~1 Q6 `5 d/ c( ^
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    5 F  A5 ~. e+ r2 s
72. strcpy(buffer,   (const   char*)sa);
    ; N1 r$ p- k' ^. g4 N1 h
73. 
      @1 D2 ^- v- v9 F2 [8 M: c! r
74. struct   sockaddr_in   sockaddr;
    9 M, @& z" ^0 s) e+ E+ n3 j
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( [" g! v% A& j0 F
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    : W5 {* Z) w- X
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    ( }/ H! V) |- Z0 s
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    : Q; W& |8 T8 C$ R" Q3 p: x" ~2 r" C
82. 
    4 y/ `1 r+ M  m  }
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / c& M$ ~: W" X; K( j
84. {
    
85. int   pos   =   0;
    
86. 
    ; _& l" \. R+ B  a
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    , A* Q- a" G7 }, \1 C. e: h* r, h
90. result.Delete(0,   pos);
    % ]4 m" P! P  ~7 M
91. 
    ' r* w  I+ \& [5 I1 J
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    + p& @- ^, l& B2 }
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)
    / `2 @* s( t' p) `- I
100. {
     1 o# Z2 U4 l2 K. b' z0 d
101. CString   startTag   =   ' < '   +   name   +   '> ';
     / n6 [! P' b( K& h& ]
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     2 \9 x! _, T8 P' @5 J2 z
113. 
     6 i+ h5 z  J) ~5 `! D) F& _, y! S1 D
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     / A$ R& `* ^/ G2 ~* j# ~' C. h
120. 
     $ I! l* @/ N! @! ^2 i. s0 T/ ?
121. MyUPnP::~MyUPnP()
     
122. {
     ; ~% b9 r, }+ x( X0 O% S7 M( \
123. UPNPNAT_MAPPING   search;
     ; F3 V1 I4 _- @3 Y7 \5 Y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     / i2 Q; b4 h( g# {* _& H: U
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     6 e* f  l9 {% v4 l: X2 B
127. RemoveNATPortMapping(search,   false);
     % s5 Z2 b5 y. M
128. }
     
129. 
     6 S( \- x2 B' |; Z6 T/ T
130. m_Mappings.RemoveAll();
     + x+ Y! Y+ B, k6 e; `% E1 `
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     7 @8 i8 k& w  E1 Q0 f2 Y
136. if(version <=0)version   =   1;
     0 ]; G0 q/ ?: H
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     / {* Q2 i3 m  v
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     4 f8 a  M# q9 h+ V  z: O
144. };
     - e+ \* M% U3 g. I0 d7 J6 D3 l/ a8 Y. G
145. 
     0 a8 T# W6 ]' p0 U! X
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     1 u6 ?2 l9 A; T' ?
148. ioctlsocket(s,   FIONBIO,   &lv);
     4 s9 x) j( B# e& ^& s# ^4 @+ w4 A
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
       G# b; t) H+ j5 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 "),
     : ]4 n* q+ ?1 e" b
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     0 l0 u  q4 W" }6 N/ x
160. }
     
161. 
     # h( S4 B0 z8 p9 l% }
162. Sleep(10);
     & w9 c) Z% H9 @2 u! J$ ~
163. 
     
164. char   buffer[10240];
     " z; {# \* T7 I. ?2 r8 S
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ' m+ h# F8 ]6 D0 j
166. if   (rlen   <=   0)   continue;
     / I, l2 O+ Z- t9 [; i( X
167. closesocket(s);
     
168. 
     6 q2 q; Y5 k/ s
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
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;
     ) o1 T  Z7 X7 a- Q/ ]! O
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     + j% G/ b' G! V1 `  f
181. if   (name   ==   _T( "LOCATION: "))   {
     2 Y% W. {+ j' g) X$ q  [& j2 A
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     & w+ p1 M- A( C  d
185. return   GetDescription();
     
186. }
     5 E; h" d" ?  b( H. T8 h% e+ i
187. }
     
188. }
     
189. }
     / f6 m. N3 {* U) I0 i& d
190. }
       u& m1 |/ }+ h/ Y- ~
191. closesocket(s);
     : [3 t1 z# h- @/ D4 S' a) O
192. 
     
193. return   false;
     6 I1 Q& r. y  d. U4 ^3 y
194. }
     

复制代码

天云上人

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

) ]0 _! y7 Y" f" y- O0 ^
///////////////////////////////////////////
5 p' L( W0 n0 R% K+ d: r; F1 b//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
1 D8 O5 R" y$ [

8 m( h2 t; x% s8 W# B#pragma once
#include <exception>


% l: y7 l# M$ E* {  enum TRISTATE{
        TRIS_FALSE,
) b% H5 s+ y. E; ~9 J        TRIS_UNKNOWN,
        TRIS_TRUE
};

; n' q0 w: ^9 z5 t! d. u
% A# B- d& L) Z2 ]- M5 k& Benum UPNP_IMPLEMENTATION{
( a# r2 @2 _* y' s% f        UPNP_IMPL_WINDOWSERVICE = 0,
. ]7 i0 k* L$ C$ W/ d5 D8 ?0 q- N- f$ l        UPNP_IMPL_MINIUPNPLIB,
& v; G, [6 `9 {: U5 y        UPNP_IMPL_NONE /*last*/
+ v0 Y  Z7 B. Y. n+ ]. J};
" n4 G9 j$ @, B# x! X$ k) y



class CUPnPImpl
% ~. x& f* N; M5 ~{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
, l6 s/ a( Q  t0 Q* V0 z        enum {
; g$ V0 V$ u4 b( \/ `                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
1 L" q0 L$ d" U  B( D/ w9 F        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
$ y/ s1 o- j$ Z# a7 ?        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
% `' S$ b5 Z/ u0 h) _       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


" u& o7 n6 Z8 g9 ]; k" h$ ]        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
5 v% Q4 d. f! a# A6 |        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
# b4 w( t7 K# B- F" E9 a        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
" w$ @/ }' C' ~/ I        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
, B* T( S, y5 A9 |3 }. t# w5 h

0 C' o# C0 ?" J! v; W// Implementation
protected:
2 K/ V4 s7 e. i6 K4 t        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
6 v9 @2 s* s5 G        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
3 k+ n! Y2 c5 d: v$ n$ o7 M  U% F        uint16                                m_nTCPWebPort;
6 s  w  R. J6 R, \+ m4 F        bool                                m_bCheckAndRefresh;
+ F3 y3 @% \$ Y8 r! I- `
' H; y( p& f8 S7 l1 a* z0 w
; O. B4 K5 H1 d/ e3 U3 d8 pprivate:
3 ?0 |) l- j$ e1 v* q: O$ g& t) a        HWND        m_hResultMessageWindow;
, ~+ o2 d- q( m7 {4 ^        UINT        m_nResultMessageID;
( Q3 |* v* s5 \+ z

; u8 O, F  a1 V! q. z& _3 H: O8 u};
6 F" w& V# A+ H
* l9 a% y5 q; ^) f2 z# c
; m8 ]1 _9 R2 n0 J* k# O( _" }// Dummy Implementation to be used when no other implementation is available
2 s9 B0 p: y6 z6 i  U9 [7 iclass CUPnPImplNone: public CUPnPImpl
1 }' p6 b+ A8 U) v  G/ v{
public:
8 R0 N/ L4 A4 i- L% j        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 Z! E1 i& R, Z        virtual void        StopAsyncFind()                                                                                        { }
, b8 N# @) I; l" H        virtual void        DeletePorts()                                                                                        { }
3 x! {& x1 j2 p# K1 Y; \! ]        virtual bool        IsReady()                                                                                                { return false; }
/ f# [! E; Q: d        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
4 X; ~4 U) l2 A* A2 f  ^; ^7 Y  W};
+ ~4 O  }0 N2 v5 i
- W2 U0 R; J9 d3 i
% z8 E7 m( }5 f2 O+ G/////////////////////////////////////
4 T3 K; h- }5 [5 d3 s8 `% \, I//下面是使用windows操作系统自带的UPNP功能的子类

* G$ K7 `; X- {. O, H8 H# d
* m, Y7 R1 x9 q: y: ~2 a3 F. m#pragma once
#pragma warning( disable: 4355 )
) M7 W, `" l) A6 n
6 l5 Z2 g" |6 _% p2 \) O7 _' b' A
#include "UPnPImpl.h"
$ {/ p6 j7 X3 l) v  y( d7 A#include <upnp.h>
/ _4 _; u" ?3 t( ]7 N! l3 p1 w& V# u#include <iphlpapi.h>
) t. T4 X4 \6 |" `. ~#include <comdef.h>
#include <winsvc.h>
8 g$ o* r( f8 f

/ y/ `% Y' _6 X. ^#include <vector>
#include <exception>
#include <functional>


+ m0 r, }2 {3 r2 a& I
9 a7 ^* w0 b1 j
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
4 {! [! _7 f/ \# @typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
) b7 Z& z- M  z" E$ `* `. ttypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
, o2 j1 D* f. j- f9 v* N/ Utypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
# U+ U( [, b3 atypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
! u$ d, a8 }$ `" Z, C* k

* n1 ]+ w- T) Dtypedef DWORD (WINAPI* TGetBestInterface) (
( C! @% }. m" S( O  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
) l7 a! C( q/ x6 G% a7 [. b);

8 M. c; t0 v( y, G( H: N
typedef DWORD (WINAPI* TGetIpAddrTable) (
! n3 b& {% k% p" ~  PMIB_IPADDRTABLE pIpAddrTable,
# \# ?# `& J$ ^+ P% z& `  PULONG pdwSize,
0 M. Y: S2 A) \/ Q' H, C  BOOL bOrder
% D& M6 k( v; M& k# a3 N) \);
) F2 W  A9 [8 }9 T

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
! b5 P3 \: m. \- q& k' H);

( y: c1 h; z1 ^9 b0 v. n0 @
7 s- m2 L7 L% iCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
6 x* W! e+ X& o; W: B2 G: Z2 j* s  q: c{
        friend class CDeviceFinderCallback;
0 y/ M" t) w9 g( p) P" D        friend class CServiceCallback;
) y5 w# L$ `- D6 }9 O// Construction
public:
# V& e1 g$ B8 g0 l# \7 E$ Q        virtual ~CUPnPImplWinServ();
6 y# {# x1 |* ?/ B2 d        CUPnPImplWinServ();
, K) a2 x# N$ V7 F" R$ b

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
3 Q5 i9 g8 u* Y4 l  C        virtual void        StopAsyncFind();
: L4 k) I7 Z% i# n7 v        virtual void        DeletePorts();
2 M) L- ^5 c" ]        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
5 V0 K! e' a& i/ y

" c! @4 o& z- ]4 i        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
  z: J& z$ h) U; C7 n        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
8 E1 C7 z6 l. Y2 V3 V( n' Z        virtual bool        CheckAndRefresh()                                                                                { return false; };
% |, i% P. n6 R+ O( L  y3 W  m9 v
4 y/ p& J4 r1 \0 P9 y
protected:
( L' a$ a( ]0 C        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
$ d- x7 f2 J1 r        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
) g8 b+ D2 v+ j& _        void        RemoveDevice(CComBSTR bsUDN);
+ p* ~9 f, _4 Q' B# `* z8 b        bool        OnSearchComplete();
        void        Init();

2 V# Y) U- Y- a2 B
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
8 t" V, k. ]  i' P$ h  h( t                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
+ }* s0 }9 \. U                }
                MSG msg;
! Z7 o7 c) h, A8 c  T                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
1 g) V1 o4 o# v, ~% ~4 U" u) ]                {
                        TranslateMessage( &msg );
5 w, f, ]7 x% u, X8 H7 m                        DispatchMessage( &msg );
" ~5 I3 R# C1 \! H$ _                }
, y! n0 `( e& ?$ {( {& r                return m_bAsyncFindRunning;
        }


- z! O' ?, J7 u0 j- n) z8 g% Y# ?$ |/ D        TRISTATE                        m_bUPnPDeviceConnected;
( u/ k, x, h  Q  d) j7 ^, l

7 S1 _. h0 C1 U$ z& M, a- p" [// Implementation
% e+ ?  d7 L, \6 K        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
1 E. M( n! F- w6 Y% X  F( v        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
- e( x5 Y) Y' f0 _        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
% N7 g9 P8 ?- J4 Y        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

8 J5 h4 M$ |( u4 x5 q
% i7 j. V0 x0 Z4 u6 h        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
# E( T0 U3 }  j6 {5 b9 y        TGetIfEntry                                m_pfGetIfEntry;
3 z+ T) J$ s* X* @" s, f
7 P; s% X& o/ f
' o0 S1 {$ P; {5 z! F        static FinderPointer CreateFinderInstance();
6 x6 O2 d. T! H) V" d: k5 n* b+ B        struct FindDevice : std::unary_function< DevicePointer, bool >
4 U8 Z+ b, k9 E& @& x( T3 k        {
) |+ R+ X& D0 R- [2 P0 E                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
1 u+ j5 p% e2 d! j5 O1 A$ C8 ?                {
3 s$ Z& n6 C; H, ^4 J4 f+ J                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


- E$ E& T, Y$ Z+ ?* |$ d  p7 H# l& S6 U                        if ( FAILED( hr ) )
+ C' Q- C+ |$ g+ I, V                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
0 H& S* k# m! g$ B# V: |1 z( _                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
4 e. B4 o& ~2 s$ Y        HRESULT        GetDeviceServices(DevicePointer pDevice);
3 z6 i# S/ ~' _3 j, y* \9 Q        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
$ h: m: u6 R) i6 {; Y                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

; N( x5 O4 B7 p
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
) e$ D# ?( l. y- j5 v$ \9 \* @8 @        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
0 d% s( q9 g' J2 Q. h1 q        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
$ B  S  N: o) K        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ D+ @# j- z9 R7 u2 h8 n/ k; k        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 c8 i2 [9 m1 h. r        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
# Y! B/ j2 V0 n* z9 m) [' oprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
: x0 M) }8 ?4 O  C' T5 \0 W2 m        DeviceFinderCallback        m_pDeviceFinderCallback;
, u1 P) O. i% V* t( v8 K        ServiceCallback                        m_pServiceCallback;
1 c! Z! R6 k+ G. D! q+ o
5 ^6 h8 s( g6 ~- e3 n
4 i, z" W4 C0 d2 T$ \9 L( g        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
( t& L( W3 C/ \4 Q, W" i) x5 h9 m        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
, D! P' C' Z4 W0 E" h  a: {        bool        m_bADSL;                // Is the device ADSL?
; q" O( f2 F1 x, c! |& d& E        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
5 |2 \+ {8 A7 N: D. G        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
! a; j0 S5 u: c        HMODULE        m_hIPHLPAPI_DLL;
  U' j& j7 S5 g+ F4 N' z  ^        bool        m_bSecondTry;
+ ^, O8 S  p  k7 i8 x- @0 j        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
; ^% N7 u+ C1 K9 r- a) y  X        bool        m_bDisableWANPPPSetup;


};

, `8 b! p) }: n2 g2 }% |$ S5 G
// DeviceFinder Callback
1 n9 O  t& i! xclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
. ~7 i" k0 Z6 Q; f) J        { m_lRefCount = 0; }

, X( \6 L6 f# U4 D( U
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- ~! z4 B- ^0 L   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 G3 Z8 W+ @) @/ t; c8 `5 ]1 k

// implementation
- P6 R2 S2 V$ M& \private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

' E* N7 d/ @$ R
private:
* x* @+ E7 G) I# W0 j5 l! I        CUPnPImplWinServ& m_instance;
7 }1 h, A- [8 _- N: R        LONG m_lRefCount;
};


, [+ }$ Q3 }3 \6 @2 u/ a. W// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
5 f9 o5 b0 [" ?* ?% }  w        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 n% `% R, |/ d2 V7 ~3 G   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
' X" t) K$ r: S. Z  g! ~
6 D$ \  q: t- ?1 ^( n
: U# R. i( D% H2 [( L// implementation
private:
" f* K1 {; b2 _; P4 l        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
6 `0 d: I, L, ^! y7 G        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

" m2 c4 G  ^- s& t/ ^8 `
9 x, l; u% d8 K7 V0 g4 Dprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
6 E1 \: c5 ~8 u8 x7 C# @6 `$ b

" x+ S( D0 v: a1 }/////////////////////////////////////////////////
9 L, s# k7 @" I2 T* x, p9 u: w( Q

3 ]" V3 r6 X! @% p  w( U使用时只需要使用抽象类的接口。
" p- h8 r3 z) F2 J) f: ~" s0 f/ ICUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
/ \5 a% v0 D7 ACUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
! H; v2 D5 R( I( fCUPnPImpl::DeletePorts删除端口映射.