UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   2 s. \. U# t* Y* e! A1 X, ^8 e& q
3. 
   
4. #pragma   once
   $ Y& k, q1 e9 [' g
5. 
   
6. typedef   unsigned   long   ulong;
   # g' [" t/ l( Q& f# e# Z% w' R
7. 
   + v/ N' L: M% g* ?+ k# M5 s
8. class   MyUPnP
   3 x, d+ g4 o- |6 f. d7 m
9. {
   + _) o( @* b$ j9 T# h
10. public:
    + u2 n+ h+ `! m' W  \, A
11. typedef   enum{
    ! y7 N5 A  v; J4 D; p
12. UNAT_OK, //   Successfull
      e' i% E$ R, N) ?8 f
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
    + b3 r3 a4 K( a; x! k: d
17. }   UPNPNAT_RETURN;
    
18. 
    5 I/ F( t! Z* ?  m+ u! b
19. typedef   enum{
    7 U- B" Z& f6 m. r( X
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    4 ]8 q4 T7 C" `2 H; n1 x
22. }   UPNPNAT_PROTOCOL;
    
23. 
    " |- l( v* M9 U& V- x- g3 H
24. typedef   struct{
    9 t2 m8 |2 {* |! B$ `( h
25. WORD   internalPort; //   Port   mapping   internal   port
    + f4 i- V. W# k4 y  k/ E  O
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
    
29. }   UPNPNAT_MAPPING;
    1 q& q/ A; o3 C6 H: `- f2 r7 D
30. 
    ! j4 H9 H8 {" Y7 _
31. MyUPnP();
    
32. ~MyUPnP();
    $ X8 \9 m7 R# g% k& a1 ]: z0 x
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ( Y& G" v9 y2 |; x0 M/ x) Z
39. CString GetLocalIPStr();
    ( `$ f7 n) c6 N: o9 o* t
40. WORD GetLocalIP();
    8 I( s$ R7 s  ]- K
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    & V' P9 l# ~. `
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    - j$ Y  ^: P; u& s7 r# v+ Q% Y+ K1 ~; a
48.       const   CString&   descri,   const   CString&   type);
    ( F9 K1 C# D5 Y* X
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
      Q( M/ n- Q8 x+ r* k4 p0 `
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ! W2 W  K  h' `6 J# E" \
52. 
    
53. bool Search(int   version=1);
    , r  W' \8 |* U* y1 G+ H$ Q) I
54. bool GetDescription();
    7 u& Y9 c. p$ }9 N
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    0 ]5 R' E( _8 @$ W
57. 
    # C/ C0 i+ Q6 d3 v5 U! V; j  u8 i
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    & i; o. S7 I1 K9 W5 p! j: C
61. CString m_name;
    
62. CString m_description;
    5 E- j( }/ {5 D
63. CString m_baseurl;
    ( ^( K' O/ e2 p' t
64. CString m_controlurl;
    ; J  A7 A9 j/ c! a: Q% `  `: J. [
65. CString m_friendlyname;
    
66. CString m_modelname;
    7 v  x! M9 h$ G1 X, ]" w6 T
67. int m_version;
    ' G# ?/ R2 S- g$ s3 A% L* U
68. 
    ' }# A" K. W( Q# ?9 o7 S4 e
69. private:
    4 _8 b5 X3 R  k5 |
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    + f1 w$ I7 ?6 O5 B) t% r% D/ Y' ~
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    ' Z3 q8 F0 S# ?- j4 {
77. };
    4 E8 J2 N* @/ f! E4 u0 B
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   3 ]+ c6 {+ H+ w2 T2 ]
3. 
   : M) v2 R8 N- y+ L, C; j( s' J
4. #include   "upnp.h "
   / t1 F  }3 K, g
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' U" x) f/ `/ N" D
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    3 L( w( w4 j. \% K6 Q; M
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    : m$ e+ y- O* j
13. static   const   int UPNPPORT   =   1900;
    6 i4 S) f' j/ g) ?, ]$ f" F
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    * r6 n7 }/ y: \
16. const   CString   getString(int   i)
    : k) ?* ]  l/ s! E; S' {/ n
17. {
    3 m! k- d1 ^& h3 ]1 A
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    ! c$ u( D* R1 h# O
24. 
    9 [! q/ D0 I: X3 S
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    - m. m. J$ W9 E0 Q
26. {
    : G$ i% Z6 E& J/ |' E) z, t  k
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . P) Y/ ?' t+ r3 i! K, I: v9 e
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    + a9 D3 ?+ V& Y/ p+ H9 K
36. {
    
37. char   buffer[10240];
    9 X, Y) I7 g* h$ y) P% i1 l! V
38. 
    
39. const   CStringA   sa(request);
    0 U3 D5 I. ^: X. m. u2 `  p: G
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    7 w" [. K' h* @" C( g9 D- p. w) T
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    + `" a5 k! c/ s6 l5 `1 R# i# z
47. sockaddr.sin_port   =   htons(port);
    2 E. X% r1 V" X% Z$ ?% w
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    4 q* y. B# ~- _) a0 b5 p
50. u_long   lv   =   1;
    4 u2 M, u$ m3 T0 ?/ _
51. ioctlsocket(s,   FIONBIO,   &lv);
    + O3 d  T0 ~& q/ y: I& A
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ! S$ D- x# ~5 g; ?( c$ @
53. Sleep(20);
    $ y' X/ S2 p0 g9 h
54. int   n   =   send(s,   buffer,   length,   0);
    4 C6 H0 _; f) o; D+ Q
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    # P9 K6 p9 R9 u. v, F7 |% E; P3 Q
59. if   (!rlen)   return   false;
    - _8 M7 J+ R! w0 o+ R+ Y; V, H
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    ' i) {, i' L8 a3 t- N. Z
62. 
    " C/ \/ Q3 I+ q2 a
63. return   true;
    1 k3 K5 ~% Z  M; G; I) x; |
64. }
    / {! }) f7 z/ l# a
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    / d! c' g4 N8 p& k7 D. q* f; L* u
68. char   buffer[10240];
    / C0 x  d6 t  E9 {; U* l. }
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
      U( n5 s! k. J2 w2 {" I. r. ^
73. 
    
74. struct   sockaddr_in   sockaddr;
    / i* n4 }  i6 G  w, v7 L, `
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    + ^" J0 U% o) u0 d. w
77. sockaddr.sin_port   =   htons(port);
    % p& s1 E) Z4 |8 D; x
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ) F. m% v3 X0 [5 ^9 c9 V
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    ) i- v( E$ |; I- z; h
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    7 A4 K9 g( B) `% y' @5 Y& t% @! M4 n
85. int   pos   =   0;
    9 |5 {+ N# j/ g8 j& D' Q: n
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    & f( c5 Z0 u  _# r
89. result   =   response;
    
90. result.Delete(0,   pos);
    ) X  f/ D! z' Y$ v& f: ]
91. 
    # T3 L6 O3 K2 P3 \
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    : K; v# N/ T* g# T( `
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    3 `( A' k' b8 H6 v8 T  B
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     # n( {% J# M: C2 z  Z
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     + s  @' U4 K8 V
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     % ~) A1 ~4 \/ @0 E1 X- C+ G3 E
106. if   (posStart <0)   return   CString();
     
107. 
     # q5 o$ e, h( x$ h' o8 o5 Y) J
108. int   posEnd   =   all.Find(endTag,   posStart);
     ' Z* k9 w! S% N0 a; Z! ^: C
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     # }* d( F' o8 ]0 t3 T5 Q
112. }
     
113. 
     ! T- I" l" q; m
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     : F0 ~4 E% f- n' G
117. m_uLocalIP   =   0;
     ' p2 ]! |7 q( \. X1 w
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     ! o1 [1 E  x9 t' i& q
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     / S' j6 V( p& o/ H3 h/ p! B
125. while(pos){
     , H: H: y6 C6 X' O1 i. i
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     $ x  }9 e; e5 u
128. }
     
129. 
     1 B0 R% i+ \! P6 o% L
130. m_Mappings.RemoveAll();
     8 E+ M  S/ p$ F4 S3 D. S
131. }
     6 @2 a  x( g4 p; B$ p# R' [
132. 
     . j- s: p( N5 t
133. 
     : J2 d# L/ K) ^7 N( s! f
134. bool   MyUPnP::InternalSearch(int   version)
     7 @: k& S0 a1 W! `
135. {
     1 Y1 l' J4 Y5 T  [8 Z* W
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     2 x, q1 `6 M6 V- [5 }8 s
138. 
     
139. #define   NUMBEROFDEVICES 2
     7 o2 q2 |$ o0 p
140. CString   devices[][2]   =   {
     " Y4 n7 X- ]; H5 w: A$ _
141. {UPNPPORTMAP1,   _T( "service ")},
     ! C' ^! t4 I3 I& Z
142. {UPNPPORTMAP0,   _T( "service ")},
     $ v" @& q8 N, w0 ]. j2 q- H
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) I9 F. p/ ~" W" i
144. };
     1 I! y1 c2 f3 D" I1 C
145. 
     * W: v2 x( N7 U. Q
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     & ^1 _( g4 W2 @% i$ }$ j+ I0 x7 {7 ]
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     , v0 I# ^9 c. @  ^
149. 
     2 F: v$ s) m! V7 p6 O3 b+ r
150. int   rlen   =   0;
     " p: O+ V3 ^2 a1 M; ~
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     + a- b) }* E" W9 D  I
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     * ]8 Z6 o+ r. G% \
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     3 h5 p" u7 n- \& Q( _# f7 R- J  \! D
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);
     
159. }
     
160. }
     ' o" b3 G2 u1 \; I5 v6 o
161. 
       w% E4 _# j& S7 z
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));
     3 x* f- A) |% E
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     - ~9 R% M, d" E; [1 d; I7 y% a
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     - b9 ]3 F* ]1 t8 R4 t, S+ ]
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     3 g2 i$ D, ?7 W% Y6 C3 G% ^
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     9 Y" k9 K& p- J9 k
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ( S4 ~1 r. N. I( P
184. m_description.Trim();
     # y5 q0 g5 K  u, P5 g" |
185. return   GetDescription();
     
186. }
     - Q/ X) x+ {, b1 M& N( {
187. }
     
188. }
     
189. }
     % C* ~3 V5 F3 i, n3 q
190. }
     : k/ l! F8 `  Z
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

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


, R$ x1 E. `7 @///////////////////////////////////////////
+ @. r# K) [! j//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
, N# z$ L. t5 m) @4 X: T
! Q. S/ U3 M  g( ^$ \5 v( W
  H: f6 x' e2 ^, r, T8 O6 _#pragma once
* q8 l$ E8 u% Q& Z+ v#include <exception>
0 D7 \& G. U" o9 C

  enum TRISTATE{
* ?2 ]" M. V) C7 p' |1 N+ `5 J        TRIS_FALSE,
        TRIS_UNKNOWN,
! c" i1 j0 f8 ]        TRIS_TRUE
};
) ]. Q% T- ^% E* H. C" ], u0 _: G
+ R: q; s3 Y  r3 l3 h2 G- T
enum UPNP_IMPLEMENTATION{
' W, Z% z; @- ?! ^# \7 ]        UPNP_IMPL_WINDOWSERVICE = 0,
3 @7 H4 }  m* M5 M. `( p* y: `( d        UPNP_IMPL_MINIUPNPLIB,
2 w- _2 X' n# u+ [        UPNP_IMPL_NONE /*last*/
& }/ a( D  V. z};


* P7 h9 ^: }- C$ j

+ U6 {' r8 Y- K# x% Mclass CUPnPImpl
  q) _# E% |2 ^: h3 F2 w  V{
public:
- _2 |" V2 h% w1 X- W$ l3 D* R7 e0 h        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
) A2 p+ s4 F  M* L( M        enum {
  I1 D8 @4 F& c- b                UPNP_OK,
: f. b; u+ y- }' u: P9 B5 q$ D4 K; M                UPNP_FAILED,
7 N9 |) S3 K% A$ I                UPNP_TIMEOUT
! L6 q! G; U! q! f        };
) T7 s4 q9 G' j) j7 o

* F6 _1 e/ T8 J7 s        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
% k* X# v! |6 h9 [& t( ^        virtual void        StopAsyncFind() = 0;
! ?0 ^2 @: ^" C) f) E# ~        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
- v$ Y7 ~& N+ B# l& \5 K        virtual int                GetImplementationID() = 0;
& G- k4 Q$ [- R# H+ K/ P4 p$ B0 w       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
' w* q( W3 a$ G- Z: d6 o

- W( \  d/ P3 N2 w: t$ n        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ N( p: F* M: a; U& _        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
1 x4 f1 B7 w/ W0 h
# F0 F7 j: D7 d- m5 N
// Implementation
$ {7 z* i6 K: i( t  \0 G: L/ @protected:
, C) z% y7 D4 b, [9 U* b( j4 @: `9 _; ?        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
1 U$ N1 u; {$ m) Q/ B- R        uint16                                m_nTCPPort;
2 |+ O  g: A. O: h5 P        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


/ Q( f& u& w0 f/ m( D  nprivate:
5 B3 [3 a; u; c, A$ R7 G" U        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
. L$ y; Z2 @/ f" D) S- H
; I9 y( I  @( V6 I, Z4 C
5 i2 I) J% v& j) r7 T+ J};


* I( a' f0 a% O, E2 r// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
1 T5 S5 d! [+ V* Bpublic:
# T2 f- c1 B) |' E# v7 n. k9 ~% V, }        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
! {' _( v& F# q+ G2 w        virtual bool        CheckAndRefresh()                                                                                { return false; }
. _3 b, u, i) o8 B        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
) O; G+ z2 B3 y1 f  z        virtual bool        IsReady()                                                                                                { return false; }
2 X0 s/ R( s( F/ B        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

: |7 W9 n& X8 R6 l' I6 |
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
' k3 j0 O1 [0 M/ X/ r- S5 s
  u4 ]8 S8 ~1 o; l7 d
#pragma once
; r/ ?; M5 `& t0 e#pragma warning( disable: 4355 )

% m/ M$ k( r6 H7 Q5 [
#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
4 N% x' b2 s5 S! y, E" N. a- s#include <winsvc.h>
: k2 t& a7 Y9 P

#include <vector>
0 F4 {. n( V2 R: O& ]#include <exception>
#include <functional>
& H$ x: [2 H' d/ d
# Z7 ]& c, F5 n6 |& }' I
6 q; m. H' x0 Y( R3 S
0 \( O1 s$ s9 [- Q+ f
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
4 H* H2 K3 y) Z/ E' `% Mtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _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;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
2 ^" {' A0 q9 H# f3 ~  IPAddr dwDestAddr,
7 ]  L4 R$ u  `4 S) W  PDWORD pdwBestIfIndex
! T! t- p! b( N: W5 N4 D% B' I3 D8 q);

8 g5 A  g7 d3 d5 b
9 T! `3 N9 c( h! Z) d( T* [typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
1 e& ^8 J+ E# B  D, b0 w& s6 L  PULONG pdwSize,
  BOOL bOrder
. N6 Q' T, W' u+ s6 V( M0 w);


* a+ w' C% O1 B4 h% b0 {( Jtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
% Q3 e3 X9 l  o6 H. k, S);

+ [& X2 |" ?( K) S
! R+ `; u) b1 KCString translateUPnPResult(HRESULT hr);
4 Z" v( w8 q* Z& B! wHRESULT UPnPMessage(HRESULT hr);
9 s( G) ~+ ?6 @) o% y/ X) C4 I
  Y# M& M- D/ L% E4 M: a0 [5 T
  G4 s/ u0 [1 ~9 y' Uclass CUPnPImplWinServ: public CUPnPImpl
{
6 L6 v! ^/ g9 v8 g" U* y9 U( P        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
7 n& P# Q9 j( e/ f9 y/ jpublic:
- h; M0 A9 Q: x  C* V) O6 x: L        virtual ~CUPnPImplWinServ();
! C1 @3 w1 v( l. P4 {* F: W' H        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
) ~0 Y1 P5 f# H" a1 J2 o: G) n" S        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
5 |6 V8 x+ l% @* x# u        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
& `6 M1 }7 Y" A- k: ~9 ?4 r

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
- l7 s( U! i  w$ _% e# E        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
$ m; A! B5 J) p) N1 \        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
/ N8 |2 h  m* K. I7 `        void        Init();
  n+ O. k" A- v

- d- V  R0 Z# n5 j/ B        inline bool IsAsyncFindRunning()
+ }* o4 a4 @2 ~+ N6 t        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
. S, J+ v/ k5 s9 Z" I                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
7 h$ o) Z( C* J: D* B4 _2 a                        m_bAsyncFindRunning = false;
                }
+ Q& F7 P7 Q) @# i# E                MSG msg;
0 `' a) j& ^8 s1 ~6 `/ k                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
/ A, ?8 z$ ]2 J- n, P- T6 k# Q                        TranslateMessage( &msg );
- W: ]- l3 I$ E                        DispatchMessage( &msg );
                }
; F" ?7 J$ k, S; l                return m_bAsyncFindRunning;
  A- Q3 A) h0 s' D5 q        }
0 S" O9 a( ^7 G) J

        TRISTATE                        m_bUPnPDeviceConnected;
- [, Q& i5 [, E. b. i* \9 ]
, ]* J0 z# z4 ]6 ]  H# c1 Y" {
' F: J1 V$ m1 n( A4 y2 S// Implementation
" v/ V$ ?1 A' l% h$ h" V        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
: h6 N9 P" ~% x  E        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! G# Q9 I/ Q- G' F        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
' J) `3 X" B' Q, ]$ g" _+ D; W! L        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
# ]: q- ^6 d- h+ i

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
, [( p7 I* C# @  _, _5 j, u

        static FinderPointer CreateFinderInstance();
6 A; N  ]; _3 H' p3 g/ \! T2 t        struct FindDevice : std::unary_function< DevicePointer, bool >
1 l* U3 o$ S+ C        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
8 M, n1 c  a) e, H1 g1 T/ t                result_type operator()(argument_type device) const
' Z6 f/ `1 F: `7 s+ F' j( g4 v5 f+ Y                {
( A9 Y; j* j- V5 o# T* t3 [8 x                        CComBSTR deviceName;
7 ?4 o, H$ g1 r7 P/ }5 n                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
$ D+ f+ S; F: C1 j9 q( e9 ?
3 E4 ^! y. v8 o9 H5 J3 n
                        if ( FAILED( hr ) )
' C8 N9 i6 y- v' I7 c  Q% I                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
: A( G* Z: u5 o1 ?. r  y; W3 ]        };
# @* w$ U( u' V* e, n. b       
2 g2 c  k# X; K/ T! Y/ T' x        void        ProcessAsyncFind(CComBSTR bsSearchType);
) a3 s" b, X; J2 W' F4 \        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
1 A, Y4 x, T8 t  C" I        HRESULT        MapPort(const ServicePointer& service);
1 Z5 {! N; O8 d5 |: t% c* j        void        DeleteExistingPortMappings(ServicePointer pService);
' @. Y) @" H6 F- c- J" {# S- v        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
& @: G( x; k5 [        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
0 @  h5 `2 T" o

% _" W1 Y9 R& \. k        // 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);
% O1 i& {9 |( R! I  q; y1 x8 R7 I        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
1 N' J8 w! N/ K: a0 F, b+ x8 w

// Private members
9 O. g( ]! _- L; T3 Aprivate:
( W+ O3 f5 g: i6 P6 ?8 A& x        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
4 _2 d' v' L" D9 j) F        std::vector< ServicePointer > m_pServices;
! E6 Y" Y% n  q        FinderPointer                        m_pDeviceFinder;
7 w( b8 s% V7 H3 }, m' C1 `        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

/ K9 e4 w* S4 e, v/ _4 [+ z2 a/ c
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
& k# D( k1 g, I$ Q) C( S+ Y: I" ]        bool        m_bPortIsFree;
0 u3 B$ W7 {5 O5 D1 }( C8 \        CString m_sLocalIP;
        CString m_sExternalIP;
9 [9 ?  f& z" s( f+ L5 E1 o        bool        m_bADSL;                // Is the device ADSL?
' J3 z( Q9 H# F        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
1 C* x& f2 J2 T' E( L; r4 {        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
5 i8 F6 Q/ s8 |' R; H- D        HMODULE        m_hIPHLPAPI_DLL;
! v2 Y2 M# V' Y$ J5 G$ I        bool        m_bSecondTry;
& Z- m- |( z, z0 a- B7 v8 ]$ m        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
& k( O" J: ^0 a/ G8 u0 Z8 K        bool        m_bDisableWANPPPSetup;


4 u; q: V# [2 _};

5 e- U& v+ K8 Z/ t6 \
& J% H( B+ O* }/ i5 F// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
; S0 B' w% P" [: U- t6 T. a% D  o        CDeviceFinderCallback(CUPnPImplWinServ& instance)
0 D+ h) D8 F* W4 J                : m_instance( instance )
) @1 y: b# l/ m4 C7 ]: ]: h        { m_lRefCount = 0; }

  G! Y3 H# j9 u9 M6 C4 q
. f7 l  w8 c( P3 Q& L   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
3 T  G3 Q0 d5 |* H: e8 J6 B- a# i   STDMETHODIMP_(ULONG) Release();
) N! U- D4 U- f2 N( O% L
2 }! |& L: l# |) @0 M3 W
4 \: C, O- L$ q5 ^8 ~& ]// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
0 X2 q* {/ r5 K8 D        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
- {' W- Z3 Q% Z3 A! O

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
: L0 O) e# W& H3 Z5 S' P: m};

& m  N% ~- X* H9 f9 @7 }
// Service Callback
5 I- ^) `# d0 Q9 L) L5 lclass CServiceCallback
        : public IUPnPServiceCallback
{
  K0 S" j- H1 b: H% n9 h, h) Mpublic:
9 c  F$ W+ x1 b! z) N        CServiceCallback(CUPnPImplWinServ& instance)
% `: m' ~) q/ M. A% }6 F                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
( V2 \0 q! w1 W2 j: \   STDMETHODIMP_(ULONG) AddRef();
# @4 T5 ^) x5 t4 |   STDMETHODIMP_(ULONG) Release();


% I$ U! M8 j( N# B' j// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
: X6 O6 o8 P& J1 N3 m6 F7 f        LONG m_lRefCount;
0 R1 j! U* w- G" x3 [2 l) C& Z};
2 O" ^9 ?2 |2 \& d
' X# j1 |- H( k; p. w
/////////////////////////////////////////////////

5 z( [+ g* @# j% A  T0 G2 [
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
/ Z  b& A$ I8 r) \7 `( B4 v& _6 `CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
, |- J$ n/ x. gCUPnPImpl::DeletePorts删除端口映射.