UPnP

天云上人

/*uPnP.h*/

1. 
   ) [/ _9 [) T/ r" u
2. #ifndef   MYUPNP_H_
   ( O! m: s9 [3 t/ r0 p  N8 Y
3. 
   
4. #pragma   once
   
5. 
   ) x8 y! h9 i4 }( E' P8 x
6. typedef   unsigned   long   ulong;
   - X6 z: f5 U4 _* I/ ]3 G2 t
7. 
   ; d' `+ i- k8 b& P9 j% A
8. class   MyUPnP
   0 E* m% Q9 B" B% b( E' c8 p
9. {
   
10. public:
    / \, a8 Q+ ?+ f6 Q8 I7 m9 m
11. typedef   enum{
    6 ^: `. t) X: D' p$ t* [( r
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
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    . M, Y$ p0 Q" U9 z& k9 _/ H
17. }   UPNPNAT_RETURN;
    
18. 
    % a/ \9 t9 `$ Q# y
19. typedef   enum{
    + [9 w. j6 h6 D) q; `) x; J
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    2 H# }0 o0 q) j3 H6 c1 f1 u1 f
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    1 P) A# }& i( }1 w' O
26. WORD   externalPort; //   Port   mapping   external   port
    * ^; ]' l* M; [# S# s# y
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    , P9 i4 s# u5 y2 ?9 Z  x; z( `- D
29. }   UPNPNAT_MAPPING;
    
30. 
    - Q( U0 v1 h6 Z) u/ m5 }! b9 C+ F
31. MyUPnP();
    - l. l- d( r# l' p4 t2 a, X
32. ~MyUPnP();
    
33. 
      |+ G9 I) u! K! ^& K
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    / q, o0 Q& ~% X, Q9 _
38. CString GetLastError();
      H: W' K" k5 @- V( ~5 Q
39. CString GetLocalIPStr();
    2 w8 I) d6 O. C+ ~( y& F1 w
40. WORD GetLocalIP();
    8 r9 h! I' O6 H8 ]& M2 U4 x, j
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    ' e2 X* H/ ?# A: N+ k1 H
44. void InitLocalIP();
    . ~; `2 Z& \! x8 y
45. void SetLastError(CString   error);
    6 U3 A) \: L* I, \, B2 S: Z
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    * I3 e" ~$ W7 V! v* J$ i
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    & Y3 g/ \' ~4 V8 ^8 _8 l- t1 y9 P5 `
50. 
    $ P; {, d: e0 i# K2 d2 r
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    3 u1 f; t4 T; m
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    $ q4 Q$ y& }( k* b* {! U; Q; N5 I! k- k! j
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ( H2 E9 X8 |; T3 J: W! `
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    8 V% t4 v( K% [0 `$ t7 A6 k  ^7 c; k
63. CString m_baseurl;
    3 N. y  D' H4 P0 t
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    ( b1 K/ k5 [  y1 G
67. int m_version;
    5 K/ V) i% F5 w8 d' _
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
      A% U! J- \& y4 B+ t7 z
71. 
    2 v* c2 n( N! q' R, t
72. CString m_slocalIP;
    0 f) E6 Y8 K9 h
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    - |$ ~* Y% ^8 ^& m9 l  p+ x
77. };
    0 \- D/ N# B+ a  g" ?1 y
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   - T" Y4 E" r# m1 w8 n
5. 
   2 s# g$ b! B) X4 u9 w8 R0 q
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
     [+ K* {( D$ k& ~. L
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   3 k: t2 Y" U0 W/ S- _
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    # A* L7 C6 r7 B9 V- }" D
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    - z5 G( o3 J5 f& Y
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    - G, g5 Y$ d* I
17. {
    
18. CString   s;
    1 o8 Y( \$ m! u0 X% `7 `9 I
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    , t6 j0 S: m$ x" T0 m2 C2 G4 s' i
22. return   s;
    ) x- Y$ x% {) R
23. }
    $ m$ C$ m6 b& i  a2 W2 n0 f
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    3 K* Y3 U( Y; ], k
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    # e( g; W$ y+ ?8 j# j. D6 j8 i
29. 
      H9 A: ]( O. Q$ v, g$ L& w+ r& M' A- f
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 B2 X3 @2 V9 d% E( i# M
31. {
    $ \, s& U' N9 j3 D$ Z5 }
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    2 c7 d1 e2 |7 l( z" x, S
34. 
    - g/ j7 e1 a- _; V0 }0 K- m
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    2 M/ X9 e$ i; Y
36. {
    # z: J% d; e$ Z* `/ I( D
37. char   buffer[10240];
    ; _% L5 q$ }4 |, U8 [# z1 Q
38. 
    6 J& q) u5 d% d. y
39. const   CStringA   sa(request);
    3 E& K' H9 @% O+ }
40. int   length   =   sa.GetLength();
    , V" p; k6 C# F6 s
41. strcpy(buffer,   (const   char*)sa);
    % w; [/ {# B% I* w$ q, E
42. 
    8 ]4 {$ F! z% t) Y3 h6 O6 g$ \
43. uint32   ip   =   inet_addr(CStringA(addr));
    7 b, ^: R3 D. p9 {- Q% p
44. struct   sockaddr_in   sockaddr;
    ( Q9 q9 b% o; K& V. A/ b
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    8 N) I! V8 W% v) A) g! g, m
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    $ Q3 I% R! R- y  H
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    . K9 y* Z3 O1 \/ `: m$ P: R
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ; ^5 M" y$ B8 x
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    / n3 S7 f& |) {  @. }1 z& g
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    8 y% D( j; C1 f2 A, W2 f; u# S. _
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    , N2 t* l' L% c1 }
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    & ^( z7 z8 i& `
59. if   (!rlen)   return   false;
    % {7 I/ v+ g% v9 L% L3 B
60. 
    - A8 E% u* Z$ ~% u# r5 v( B
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    $ `( A7 Q0 ?) ]$ j, x3 G) }& G6 k
64. }
    ( T1 C8 z, X4 C; j1 W- o$ f7 e( v
65. 
    : ^: K! z6 I6 D6 t/ T6 q
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ( G! \( a! F( d7 Q- u: i4 `
68. char   buffer[10240];
    
69. 
    5 n( I4 I& m) m4 L/ j, Z
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    / g# Q8 A  J& r( X
72. strcpy(buffer,   (const   char*)sa);
    1 Q. w# z( E) H* s4 W
73. 
    , P" L4 s  ~- b9 |! ?0 K
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    . F: y4 M4 N9 f, o) ^% J- R
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    5 E) m" s# X4 `4 u6 U4 Q
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ; J" O! p1 T" `# U  H. v& x* B
84. {
    
85. int   pos   =   0;
    9 M" ]' [. A4 ]" u/ ^
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    1 O0 l# `. {8 v/ T7 A. Y! a  \
88. 
    : G# E" P/ q7 j( H* B
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    8 N0 c8 n( t5 S+ J5 x' R. y2 q' ^
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    + ?5 g; K8 d5 Y4 w  b2 u7 m3 h
94. status   =   status.Tokenize(_T( "   "),   pos);
      a( J8 p  l  m) ^0 d
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    7 S1 Q# s/ q7 ]3 {" O
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     3 n$ c- @& x* I
101. CString   startTag   =   ' < '   +   name   +   '> ';
     " a' U- G- f+ S2 c3 Z) }# A
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
       I+ c% E6 A; o1 r% q
103. CString   property;
     + ?, D; Q5 F4 J- l7 o0 X% R
104. 
     ; k& J# Y  Y% n/ ~7 K; p$ g
105. int   posStart   =   all.Find(startTag);
     ( p" x4 |3 \  n  Q; a1 _/ D
106. if   (posStart <0)   return   CString();
     ! e0 Y8 a/ J/ w% Q- [0 d/ n
107. 
     * _' s# G7 J  r- y7 Q) J4 b
108. int   posEnd   =   all.Find(endTag,   posStart);
     + {; |( ?  |% N
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     2 q8 ^4 }; t8 o' B0 v
112. }
     7 c, ~* e1 B( r
113. 
     
114. MyUPnP::MyUPnP()
     ( V' e! c. Q( q) e3 p
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     ! f( z: J  B3 J
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     2 x* e0 I: l( K9 Z+ X
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     / ~; ?( n- I8 o0 a
126. search   =   m_Mappings.GetNext(pos);
     ' O- N$ z$ e  ~& Q( ?5 P) H+ a
127. RemoveNATPortMapping(search,   false);
     6 I! c& C. v5 e6 Z. e6 o
128. }
     # ^) L8 ~8 K! o; z
129. 
     * D, f: c9 @4 p# t. V
130. m_Mappings.RemoveAll();
     " V, |6 ]& R0 i( q- @1 J
131. }
     
132. 
     
133. 
     / D& Z4 X: _* e6 M3 C5 [1 h9 Q
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     9 \1 A/ t: W0 k" n
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     2 R# [/ V9 l/ ^# p1 ~
138. 
     
139. #define   NUMBEROFDEVICES 2
     2 x! C- U* i8 ~- P8 @
140. CString   devices[][2]   =   {
     6 d6 v" |2 z- `. o8 G
141. {UPNPPORTMAP1,   _T( "service ")},
     " x  }7 W; O9 @/ T
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     2 g5 J0 s0 z2 o- [' T
145. 
     ( w1 S$ O( I! U
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     3 f: l! X1 s" i( v0 g4 _
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ' h# }- `3 f5 g. z& ?0 `: {
152. if   (!(i%100))   {
     ( H8 u/ l) C: ^. w! p
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 "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     + B8 k. d9 H' _5 n; C
159. }
     . P4 a& R( S1 E5 S* v
160. }
     ( M) E3 Q0 x2 w% r$ {3 Q
161. 
     " G+ J! y. y8 ]$ P9 c& s
162. Sleep(10);
     % A$ m3 r9 Y$ _) D0 w
163. 
     3 K% R6 R9 D& h& ]) f
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     8 |9 |/ ~( J( E/ A  B+ w  T. Q% ^
168. 
     , b, p  b0 L2 n0 n/ Z
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     3 @6 Q% W$ S  G4 }! M+ _3 ~
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     % @: `3 U, E+ u
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;;)   {
     - @; x4 e$ g* l2 F1 a
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ' U) j! a$ ~% W4 D) S4 K4 w0 Z9 C
178. if   (line.IsEmpty())   return   false;
     ) W* i. u4 c+ c: {  t' s* q: ^, {
179. CString   name   =   line.Mid(0,   9);
     " ~8 b7 e% T5 ?& T, [* p
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     & s. L* e1 J! C+ I7 C/ w  ^4 R2 X6 N- [" K8 n
182. line.Delete(0,   9);
     
183. m_description   =   line;
     4 I  {! o( P" Z5 t1 W4 u' P
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     4 Y1 W0 R& A" v- [% W% s( g
187. }
     1 v# z# x2 s  q/ |- `: f
188. }
     " f4 R/ |) o6 y% e
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

/ z& w9 U3 b+ w) H0 Q& O
#pragma once
. {' l% t, H4 c; ~" J#include <exception>


  enum TRISTATE{
* z, m' {! ?2 d        TRIS_FALSE,
        TRIS_UNKNOWN,
0 P# V. h5 G6 o7 W6 \. P( `        TRIS_TRUE
};
* [9 X4 A, r8 v" ~1 j+ a

) l2 e& J# @4 p1 y! Nenum UPNP_IMPLEMENTATION{
" s+ `3 z4 i1 O- Y$ t6 y$ @2 r! U3 x        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
& i9 g/ {& e9 n3 O  |};




& j& c* n$ d& W: x; pclass CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
" m& B  Q# h  p$ d9 g        enum {
                UPNP_OK,
6 s' K3 N! z% R" b) D                UPNP_FAILED,
5 ~$ r1 }% P3 ?0 i! k7 J3 ]                UPNP_TIMEOUT
        };
* w  o( J% Y: A! w

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
9 h& p- B0 `) K4 n8 v4 y5 ~) |        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
8 Y. Y, q+ f/ }# |       
  {- t* J, Z7 ?+ g        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

! L/ y+ ^3 t, i  t) _8 G2 S
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
% J: S2 [4 T7 {6 Y) F        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ i# n2 y$ I5 S( ?        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) z5 c  t3 O0 e3 m
1 {: m8 D5 a: J
// Implementation
( A; l3 \, {' v: s( Gprotected:
3 D0 b+ Z3 m' ]( l* V. `        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
+ ~0 K5 `! i/ q; M        uint16                                m_nUDPPort;
# B7 V- h; f! A- z0 M7 D( @        uint16                                m_nTCPPort;
9 L1 w; E3 g; e/ M- g& [        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
& w" ~8 U8 D6 x" g6 p6 \" b
% H) H$ ~% j5 }! i2 D" l
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
! e' q( V7 v% [3 J7 U7 v# S
$ j/ H, H5 W7 ?0 E9 K1 `
& F: w! i2 X% `6 z+ o6 l! n5 l};
# L, w1 ?: [: R0 _
7 E+ Q( x2 _! ~- y
// Dummy Implementation to be used when no other implementation is available
: E) C/ C- t( d% @( u) kclass CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
! A% \8 t6 o( |% F% U: F. a6 O        virtual bool        CheckAndRefresh()                                                                                { return false; }
; R: _* e" R% ~. G6 s  [" `# F+ _        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

9 W* k. ]' C+ ~# [2 z) @* s
% q1 Z8 O' o2 Q0 U3 ]- v/////////////////////////////////////
* f& w6 W% ^6 F' Y0 w* @* N- d//下面是使用windows操作系统自带的UPNP功能的子类
8 P: h3 S& d! f/ ~3 K$ H

#pragma once
. a0 r1 E7 T8 e8 ~7 u#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
) D6 O; x+ U9 w" J3 ]9 q! b% l" @#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
9 o' O! o( X$ v# f7 `0 I#include <exception>
#include <functional>
3 v: |# T2 n; l, z0 p: j; p


# n% g8 L# K0 b- h( B/ A- T
0 B2 o+ {' E+ d9 |: Rtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
0 Q) T1 X  w# D5 C* m7 F! Xtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
( b9 n8 }( m/ r, U: ~$ jtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

0 m9 y: K1 Q4 h
( J5 Q) d4 ?* |! p! k& ]. [* ]2 Ltypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
9 c" n0 o5 c) p% G: m  PMIB_IPADDRTABLE pIpAddrTable,
) T) k7 n3 @/ D$ [7 b9 h3 P  PULONG pdwSize,
  BOOL bOrder
);


6 T6 C5 z+ s, j% N; T: {' Itypedef DWORD (WINAPI* TGetIfEntry) (
/ _4 O- V0 @0 W  c3 n1 p' b6 V7 Y  PMIB_IFROW pIfRow
);
! X- ]' s- D% S
- X) ]+ R" ?) d3 Y# v- e1 V
CString translateUPnPResult(HRESULT hr);
8 O7 O' Z$ {# i( h  |& \( I: _HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
! s1 l' K9 y, \2 ^        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
* c" v7 M' M  v/ [5 i// Construction
public:
        virtual ~CUPnPImplWinServ();
* x  H+ Z3 f8 K  z" y        CUPnPImplWinServ();

9 r& e- g/ V, [6 h. t
; T( M$ ^6 C+ ~; Z5 ^        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
( ^: H- G' ~3 w1 V, O        virtual void        StopAsyncFind();
( y6 B2 V; B& A# T9 H/ ^( x        virtual void        DeletePorts();
1 _. [) N, z+ X, u0 X        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
( O0 w" k3 ?  ]1 }5 `% b
5 ~' l+ J2 |: l) S: T) b. l
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
. u. E3 ]* k5 r: o2 G/ K        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

0 Y3 v' I7 V$ u5 V2 y
+ O2 N/ Z* _  uprotected:
, S+ ?4 O1 ?: {7 B* I! o        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
) o0 e; B3 }7 z' w) ^% z' S: o0 G        bool        OnSearchComplete();
. m4 d3 F8 L* H$ u4 d/ a        void        Init();
3 |' \$ j1 `. y$ y* M, k

        inline bool IsAsyncFindRunning()
1 e$ w+ X% G: n9 l$ c# `        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
' |" b2 V' s; X; n/ p1 ?  f; ^                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
6 o  Q- L+ s4 h8 O3 i                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
' P+ e* q+ q% V; z/ Q9 i8 o                        DispatchMessage( &msg );
# a- ]' g: ]* j6 M- S1 Z8 E                }
" O) N' F8 N* J                return m_bAsyncFindRunning;
        }


        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
3 r' B0 L- R% q/ ^$ _2 Q8 K        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
4 \# J% \- ^" D* M! I$ D        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
5 l4 ]6 b  k  G% H8 T        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
0 J- g" a5 L- M9 G        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
# _7 h5 A6 u8 S  N$ [* {
' k4 h( G3 ^6 i5 o8 ~
2 `' S8 W6 T) o4 I. I' n5 _        TGetBestInterface                m_pfGetBestInterface;
( v( }3 x( ~7 j2 }0 x) S3 H        TGetIpAddrTable                        m_pfGetIpAddrTable;
# x3 p' a$ \7 U        TGetIfEntry                                m_pfGetIfEntry;
. |( g8 r0 l0 z5 ^
* U' k2 c0 Y# L6 w
& O. r* h6 G% q% x0 `        static FinderPointer CreateFinderInstance();
8 H% l! `0 Q! I9 P1 I" q1 [        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
1 V+ q, I; |2 i7 C  J) z                result_type operator()(argument_type device) const
0 P9 {2 n: u/ G                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
8 \: k5 n9 s2 n0 v                                return UPnPMessage( hr ), false;

; B& z/ h- T1 C0 }# D% Q& z
3 ^* J- v; e( ]3 u5 O1 q9 q                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
( @* z! i: y1 h; a6 X, D5 c* P% ]                CComBSTR m_udn;
        };
       
# ~  R& @3 o  @8 R) p        void        ProcessAsyncFind(CComBSTR bsSearchType);
3 K% a6 b- W8 J        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
9 _$ C, E  E% e. w; y( g6 q        void        DeleteExistingPortMappings(ServicePointer pService);
, `3 C0 s7 x7 u9 k! M+ A" h6 i        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
* j# X1 E9 Z: N% J  G% O7 h        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
4 a5 e  f. M7 g/ r6 ~

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
0 p6 q. Q! m/ w3 d        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
6 L3 y# b7 x& y( |4 {        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
  k$ Z: q! `* R5 a( p8 @0 f7 T        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
, Q$ ?% P" [; V9 _( Y6 g9 z2 l* f; b

, G. Q9 u' G% Q  y% e/ z// Private members
private:
, l6 C/ Y2 q) P1 z: a        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
6 n9 q: m2 J. D2 |  D" D$ A
/ k, j, a" M( v: K' g, i; r* ^
( G  [; V( ~- B# S- H/ k9 b        LONG        m_nAsyncFindHandle;
- P3 c& ]( A; H+ r        bool        m_bCOM;
: F& x# u( t: {6 e) p' E' y        bool        m_bPortIsFree;
7 r: H! k6 z6 ~. r2 J        CString m_sLocalIP;
        CString m_sExternalIP;
  T1 S* l  ^6 i        bool        m_bADSL;                // Is the device ADSL?
5 k1 E! i. {& d# t$ ~        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
& B) c" p" N8 L. k$ g( {        bool        m_bAsyncFindRunning;
) z. D% @2 R6 B1 U$ C9 `        HMODULE m_hADVAPI32_DLL;
2 T3 c, ^0 b: U( X% z        HMODULE        m_hIPHLPAPI_DLL;
( C/ w! g* j1 [' a        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
9 T$ `; Y6 J  c8 m, \6 L
# G5 {) n) v3 [. n9 C. d! _/ N1 u
9 R) X. N3 N" P+ e. `  G" W( x// DeviceFinder Callback
1 W, Q6 z2 P  N- l! {* fclass CDeviceFinderCallback
+ {7 L+ X& y; B# a2 R* S        : public IUPnPDeviceFinderCallback
2 _+ j- m9 N3 p' q; F5 v7 e{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
0 _- i; W/ b2 @$ T/ m1 y        { m_lRefCount = 0; }
# ^4 ]  S1 s! y# b

* w" X, n0 ~+ T  u$ L# N   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
9 O5 J% M5 N; x   STDMETHODIMP_(ULONG) Release();
: C. K$ F) n) ~
1 e& f7 d, v) H# ~# Y- s2 _
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
6 M7 x( r; M2 N, ^! @0 n  Q  K

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
; t5 O# n( o. N  F6 |! S2 J7 U' ?9 M};
8 q' u! k' O) f7 A# ^, x9 s$ |
+ k0 k8 H5 L/ a" V
// Service Callback
class CServiceCallback
3 M3 R0 N4 l2 S0 M0 \3 H        : public IUPnPServiceCallback
5 j, }$ c- k# S% R, U& l% m! N{
. B6 \5 H+ i  I1 o0 @" O0 `public:
4 e2 j' f9 T+ t3 f) c7 F        CServiceCallback(CUPnPImplWinServ& instance)
" \/ j& ?7 R3 c                : m_instance( instance )
        { m_lRefCount = 0; }
7 z* W8 g1 ?4 L2 W. c/ \* \% s   
0 E* L" _4 ~9 D6 {8 o& b. u' Q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 Z4 }- G+ c1 E* \; N9 m   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 v4 R6 e' r# N$ T
$ s% C7 ^( g) K* J5 G
1 U1 [- w  |  }6 ^% G// implementation
! k, @1 j& p& F! gprivate:
6 t, c! ~& B, m, f8 \        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

  M* i, ?) J+ ~6 O5 Z/ S
# K% x, l( V+ _5 |. v0 w( t1 Cprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

7 Q2 h. f5 y: p6 `  s* Q
/////////////////////////////////////////////////
" s& ?+ _; {; w4 ?+ C4 Z

! `2 E( ]- i& D使用时只需要使用抽象类的接口。
' f' ^1 U& ^- v4 ~CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
4 a/ S( Y% e( a, J0 s# k$ g: h/ jCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
( t- j# A8 A, s% I: p6 eCUPnPImpl::DeletePorts删除端口映射.