UPnP

天云上人

/*uPnP.h*/

1. 
   ! F% Q# U0 R5 x
2. #ifndef   MYUPNP_H_
   4 P& Z$ [" }3 v. ~- v1 Y; u
3. 
   ! {6 H. r; \& i! B3 F& `
4. #pragma   once
   9 x/ P* r' [$ |
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   * o( E5 C; J6 J' N/ f3 I1 M* m8 S
9. {
   # |. R4 k" D  F; \
10. public:
    3 z9 t6 P: j5 m  B5 m
11. typedef   enum{
    
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
    : E& }8 U" O6 n5 G0 c
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
    3 s. C% Y( e# ]
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ! i& ?  @* v6 T' T/ y2 F3 v- ~( W
21. UNAT_UDP //   UDP   Protocol
    , j, f7 k/ K4 h$ r0 S
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    8 m) q  F5 w$ _6 p. Y
25. WORD   internalPort; //   Port   mapping   internal   port
    6 O9 I! `7 q  x
26. WORD   externalPort; //   Port   mapping   external   port
    / ^8 B# F' o- w6 A. w( @5 @
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    / ~! m1 l- }9 ^& ]
28. CString   description; //   Port   mapping   description
    6 Z9 C6 q/ k4 v
29. }   UPNPNAT_MAPPING;
    
30. 
    ) ]5 y$ Q! |0 W  W, V  S+ n4 k
31. MyUPnP();
    
32. ~MyUPnP();
    / x5 q+ c. O' i% w+ _- {
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    0 D8 o5 ]9 U: B5 Z, }
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    1 d: y5 l2 W' f8 F
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    7 k+ G4 a6 s4 g3 U
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    # ?3 ^7 A5 r3 w: I& _+ c3 o5 {5 F
49. bool   deletePortmap(int   eport,   const   CString&   type);
    + D: h. [& r* s
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    9 R4 E7 Q; b3 S& h2 z9 j' t/ p( \
53. bool Search(int   version=1);
    
54. bool GetDescription();
    ' B, _! x& g+ _7 J" u  i7 T
55. CString GetProperty(const   CString&   name,   CString&   response);
    ' ]; P7 h1 g7 f
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);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    : O" z5 e6 o8 j- J# e
67. int m_version;
    " t! L  G3 ]3 g6 h2 c
68. 
    : G! x( w6 V8 p( f1 \' {
69. private:
    ; s' y. t. T$ N) O/ X: U5 R# y6 U8 o
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    " Z5 A; d( G$ H' Q
71. 
    
72. CString m_slocalIP;
    0 w9 ~$ K1 }( g  z* j
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    " T8 s1 o; d% U2 Z9 c
75. 
    
76. bool isSearched;
    
77. };
    5 o2 G" L( }' A' P4 {
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   / y* m+ s0 m4 S" s5 E
3. 
   / X8 f7 U4 {7 \4 `& E
4. #include   "upnp.h "
   : W3 j' I# ^( F7 T2 q
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   9 Z3 K- U" t9 Y3 b1 Q
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    " _1 E0 t) f% p- T$ l  T1 e
11. 
    & P7 `& ]# p9 R" o  i2 l
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    0 N6 d4 c% |7 |+ z( i# S( F" w1 }
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    9 T5 \% A0 r1 {' L% j$ e
15. 
    ! i5 X: g$ h% F+ h7 |5 G6 c! h$ v5 x
16. const   CString   getString(int   i)
    3 X# U3 s& o1 ~  d
17. {
    
18. CString   s;
    ; _1 i/ s" ]# d- N" `# ^# {
19. 
    - P0 F0 S9 D; S! @
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    3 P8 q% M  E* }8 v  J; g
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    5 z+ {  Q2 |: F. F: W5 W
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    3 B; B) j; A7 W  o
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 G' j' n6 l% e6 S7 A
33. }
    2 Z. q5 U  E: o' H
34. 
    & r/ w/ l1 X, a' c8 s
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
      @6 X' A) l+ h/ O4 ^' i- R
36. {
    6 l0 i6 h, h" {4 O# ~8 h$ [
37. char   buffer[10240];
    3 ]# O9 k0 Z, }+ [  K" ?0 j+ ?
38. 
    5 p: ?2 F# o+ f: c3 p, w$ p
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    & @9 m+ z: h2 ]( {/ N
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    + b" X, l9 {5 F. }; v, l
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    $ {9 F* u, g( L# k
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    " E6 Z  Y! u$ i  Q
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    # p+ W1 k  {1 X
50. u_long   lv   =   1;
    1 n0 f' A# X/ D! f& [
51. ioctlsocket(s,   FIONBIO,   &lv);
      z7 E& n8 E" _8 b( G
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 N3 P6 P! o' D  m! s/ ]- H2 g# I
53. Sleep(20);
    - h7 T2 }9 k4 z! a+ J+ ~. a4 B
54. int   n   =   send(s,   buffer,   length,   0);
    ; ?0 T$ Y) z+ Y. a. Z2 x8 t  f
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ K9 C$ T* V# v- |
57. closesocket(s);
    / N- B; L  C/ S8 B0 R
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    / S" v6 x& D5 W
59. if   (!rlen)   return   false;
    $ s4 M# c, P# W
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    4 Y0 |: e5 a- P7 c( V
62. 
    
63. return   true;
    & P" z( D* l4 y$ t
64. }
    
65. 
    3 X  \1 a' s, [2 d5 ~8 m$ H
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & h  B, t2 L( s' K  }: M6 \
67. {
    
68. char   buffer[10240];
    0 K; p6 Q4 B4 L3 f) s
69. 
    0 m. F% u6 M7 I; x! r9 a" ^% L' o
70. const   CStringA   sa(request);
    2 {3 M  Q- Q; Q) p6 @* `* W
71. int   length   =   sa.GetLength();
    " R  u1 f$ z8 O- N' M; C8 b
72. strcpy(buffer,   (const   char*)sa);
    ( d2 N' P8 s* u7 j0 S  ?1 Y
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    / A! O7 T) W3 C$ g  G
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    & Y- V0 H  B$ p' q0 S
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # e$ h$ g7 o' ~, q. @8 c2 |
81. }
    
82. 
    " J4 _! x/ `7 t3 U
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    % I4 L5 k& a7 Z  b
85. int   pos   =   0;
      N5 a8 E, _) [$ {
86. 
    $ d. D0 u# n, U6 q
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    # T3 @: J# W2 x3 E. u5 b2 E) w+ j. X
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    - K" k& e, _+ Y6 W* @( S6 D. n; z
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    : h* j. E' X# H8 h& P  a# a
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      ~* e$ ?$ w) I: z. A8 j* Z- y/ }
96. return   true;
    % U  v0 L+ a- G5 h
97. }
    
98. 
    1 ?- L: t  r1 E' A6 ~* ]: l# [
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     % b5 Y" w1 j. O) s
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     / Q' B; X) b1 B- X
103. CString   property;
     $ v# r& @' l# \5 V. Y6 ~; K. @
104. 
     ' y& q) C/ n4 a# W
105. int   posStart   =   all.Find(startTag);
     ' o! p: o( ~1 L1 F0 y$ a
106. if   (posStart <0)   return   CString();
     & v% v. B; A) |* Z: h, m3 s
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     5 \: r9 k) m( l: R1 T1 C
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     . s! u/ p( v$ S1 @% Z/ S
112. }
     $ P/ @! w" l* k$ I0 Y# @
113. 
     % I1 S  a* w- _% f; Z% J6 a
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     * d- M$ X/ O1 h) A( w  G
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     & i5 @$ a7 i5 X
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     * L) W2 H; Y' V/ _& l  H( X
122. {
     - o8 @/ ]! {& Y$ U; e
123. UPNPNAT_MAPPING   search;
     7 z& l% {$ i. N) g% z0 D
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     - h: g& X2 z) H1 ?
128. }
     
129. 
     . X8 a; B" y! f6 F- \5 J* d6 N
130. m_Mappings.RemoveAll();
     $ [7 U# O( d( ^8 {
131. }
     
132. 
       U/ q2 n9 v4 t( T8 }. h& X
133. 
     0 k9 T* b. e( D0 Y4 G
134. bool   MyUPnP::InternalSearch(int   version)
     6 c9 E8 b6 P. d2 Q7 b
135. {
     
136. if(version <=0)version   =   1;
     ) k" a1 K: U8 e) h* a* }! A7 H" c
137. m_version   =   version;
     
138. 
     % o6 O+ f3 D$ N2 v2 m3 G) s( a0 q
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ( `: f, N) A3 j7 a) d/ R8 u3 d
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ( c. G2 U9 j0 D5 j! |  M
145. 
     : g1 I6 ~! n3 Q6 N. T5 x6 }3 G7 Y9 [
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; w  P" W- L( a% R8 [/ `. P
147. u_long   lv   =   1;
     & a( f; ?4 f$ M. @
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     ; M! P; D; ?% @4 S3 q# W7 ]
151. for   (int   i=0;   rlen <=0   &&   i <500;   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 "),
     
157. 6,   m_name);
     ; J! k- c6 E; w+ ~# o& q
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     & F$ C" Q0 k3 S, ^' v
160. }
     
161. 
     
162. Sleep(10);
     / k8 Q4 q# l9 D( T' {* m. |
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     & n( ^) |& C. d# V
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ' G6 \+ p0 n% ]1 E5 Z" p6 T
172. 
     6 M/ g9 e+ S3 w$ j
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)   {
     : e& k  i0 i+ T
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
       X. F5 T8 w5 W* ]2 S
179. CString   name   =   line.Mid(0,   9);
     * l( G1 N2 T) [2 N8 h( d
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     . |, k  x% m- P9 V
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     4 N; l$ b$ D2 {% O# u0 D$ C3 c1 }
192. 
     # q: x+ J, _  _) u0 T) H% e- m
193. return   false;
     
194. }
     

复制代码

天云上人

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

. Z. U9 Z0 ~  `9 |2 H
///////////////////////////////////////////
. M/ A+ S" g4 w3 A//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

) y: T& |2 \" i# G. b
#pragma once
# T# k0 b2 q7 F8 @0 p#include <exception>

9 Z2 i3 }/ [* c+ y8 s3 F+ k& k
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
% H; ~- G; s& F) F4 }# H7 ~        TRIS_TRUE
* r5 N" R. c) {4 ]};
( j) ^% M" u* M" r

enum UPNP_IMPLEMENTATION{
. H/ G: p4 `5 P2 v' z1 b        UPNP_IMPL_WINDOWSERVICE = 0,
! g3 _" D4 g, }, c/ d2 ?        UPNP_IMPL_MINIUPNPLIB,
  m5 S4 x0 T. Q2 W3 m& S$ i" A' F        UPNP_IMPL_NONE /*last*/
! W8 f1 ~3 x" s( \- C};
0 \# Y6 c5 P! U% v) i3 @( Z
4 ^8 F' d) C4 e6 l
6 }) z! X# `4 ?2 p7 G
4 I( C! o+ q( d1 d1 t
class CUPnPImpl
% j. Q3 y, [; V{
( M: D5 k) q/ W* Apublic:
        CUPnPImpl();
/ p4 m" ]/ b; A+ h8 C+ P$ f        virtual ~CUPnPImpl();
2 }& e0 a/ E8 j  d+ a7 q        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
; _. v" B+ Q: x1 n! g1 c        };

6 }, _# S/ [5 K& K5 f/ p
1 \( G5 \3 a( {1 v( n; l" j' q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
: n  g" G) h2 `. Z$ Y$ M  M        virtual void        DeletePorts() = 0;
5 G+ p& i, d1 _+ A, a) u$ y        virtual bool        IsReady() = 0;
/ y- ]0 J# _% b* N0 B) C        virtual int                GetImplementationID() = 0;
       
1 @, }1 s! t$ a' g3 Z  ?- y        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
1 U% a7 t; O1 \" F$ P! o! A' ?) z
8 n: w/ X2 ]+ o$ u; }4 x
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
! m  g6 I4 L, X$ ]- q        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
3 w# T0 C1 Q) \& D        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
# \1 z6 U/ j7 Q  B+ V5 G        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
3 r/ I8 I* [9 P2 g* u3 K& Q        bool                                m_bCheckAndRefresh;
9 l" s" d6 n' p& R. g/ D4 j, V

& D+ p) l! ], k9 |. d' ?private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


9 x' ^6 b0 q% j/ O% ]# [8 t};
, u# c, {6 o. I

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
+ |3 m8 B; X$ p{
# \# W2 k9 N& I; O$ C7 z+ opublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
" G$ |  s0 E: p' Y6 C4 j. X        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
3 A0 O  r, s- a# ^  r& F        virtual bool        IsReady()                                                                                                { return false; }
- Q, X/ N7 y2 A7 Q. I1 _) g# I        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
5 r% ?  Q' w$ m$ G9 K& {  k};

7 i" L, Z3 z  v* g, ~' n, m+ Z
+ @1 i/ X( D# |% D3 v7 |/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
4 u4 n. N' v, k6 M' o  h#pragma warning( disable: 4355 )


, `, s( t' L: I#include "UPnPImpl.h"
( e& P' [/ h- O0 v  c#include <upnp.h>
2 a( Q6 J" P& _  p, u4 _#include <iphlpapi.h>
' R' p* ~8 |* }' k0 d  {; g#include <comdef.h>
#include <winsvc.h>
4 z7 W- ?: N; O2 ^7 t- v3 {* X4 y

# ]" `- U# O5 n7 N0 {' X#include <vector>
#include <exception>
3 q, b5 P4 [8 W  M1 n  V+ v#include <functional>
( X$ o" z% t8 q" A! g# e



: a! ?. K) A8 s6 S8 ~typedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
2 z% I, l3 A! K% Ctypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
1 C7 ^0 b: c: f2 z- }" C. y+ b2 X- _typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

; P) i8 S3 G. s' K3 W
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

' a( g/ Z' V5 O
4 O8 Q$ a- ~+ ], T( |6 [; Ztypedef DWORD (WINAPI* TGetIpAddrTable) (
# W- g3 a& t7 d  @; `0 o  PMIB_IPADDRTABLE pIpAddrTable,
" ~8 ?$ G* u# B% a% n! C  PULONG pdwSize,
  BOOL bOrder
, h+ j8 C' D+ \* j# f$ });

, R1 H, s3 V( D% H' g+ s# o
typedef DWORD (WINAPI* TGetIfEntry) (
& f2 o; l% @" r( W) J3 F, Q  PMIB_IFROW pIfRow
);

  H9 @; O7 v! [6 S. U
, T( }% j+ W$ I: {6 [, Y; dCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

+ D( E6 ]7 A2 Z6 A
class CUPnPImplWinServ: public CUPnPImpl
! L; ^4 L$ i2 P: s/ e) f8 g{
5 L  b6 L- J1 [! H, H6 p. M5 X        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
# L* y1 S4 |9 I2 A( T( m// Construction
+ T+ {: U% i$ Tpublic:
        virtual ~CUPnPImplWinServ();
& q$ d3 T5 @( |1 G7 b+ W        CUPnPImplWinServ();
( v0 G; O$ _" `  W+ O

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
! F, Z( U; v4 n2 T! K; M. }. |+ N        virtual void        StopAsyncFind();
: k; M) s% Y. N: u        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
$ ~( J1 c0 U; g5 ?0 ~( S% Y. |" _
7 T. m. S: a! r3 E7 o& `1 ~+ N
; W: j# }* m0 u2 I" ?* k        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
& X( k7 b# {# O1 O8 m; `        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
8 X" L5 k" l' z7 q* T# g  G        virtual bool        CheckAndRefresh()                                                                                { return false; };
8 T/ \4 _* j6 w' K' `7 }
5 T/ l( e4 [, ?+ S
protected:
3 ~6 M1 o, N; M' G( @) \  M        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
" z1 J1 K0 b4 D: G2 Z7 Y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
" i8 S' U7 W1 o        bool        OnSearchComplete();
" B6 o) {! C* T$ I6 Q9 C6 D        void        Init();


        inline bool IsAsyncFindRunning()
        {
3 ~% L# X6 I. T& d                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
+ H: `1 B/ l: R                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
3 A7 J! f# I0 Y, O& H: a5 r                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
. M0 j) ], m( d) M+ ]                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
! S* `  k0 E. u3 f! M! R6 R                }
                return m_bAsyncFindRunning;
4 |, V6 d' t* R- x9 p% F1 D+ S        }

. y5 R* F# U' Z' N5 s! Q/ y' M9 l
9 }* X8 S% a9 s1 M; M        TRISTATE                        m_bUPnPDeviceConnected;


7 J& F" G3 F- Z$ C6 }// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
% W: m1 b- ~' J9 h2 ~
) M. H# ^2 P3 {! {5 S
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
0 {. r% N- Z; Q; x8 J& m- R        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
& f( i: `- o4 S  B/ ?6 X0 C! q        {
% l( \1 [: `" O5 }1 J" T, c8 X                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

- b; s( X! u& l  @: j+ ]8 g
. v! n1 I' m* R: T                        if ( FAILED( hr ) )
* e, b' U: O; b                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
* [  Q5 r1 G& Z$ X& `9 Q6 V        };
       
- F1 E/ l# z5 [0 v4 P3 \" ?        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
& S, V  b: I1 O# d        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
4 |* P- R9 e" M$ z5 _9 B% x        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
/ U5 ~$ A* u8 k2 Y; e  g3 Z- ^        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* X' F/ M  i' F0 a2 i2 @4 m: e                LPCTSTR pszInArgString, CString& strResult);
, e+ p) [' S: N2 h( _# F        void        StopUPnPService();

  n9 ]* S0 }0 F" H* J  ^; b9 R
3 T: e/ P. y' T6 U6 v$ y* r        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
' t+ H5 S$ X8 f+ D5 Z' p        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
( p- _( ^" h8 y2 ^* w        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
+ U) Z0 g' ^5 P1 @+ u; n# X        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
& w7 m0 d8 L# i6 G
& C, a. b5 h3 c/ |3 O
// Private members
private:
( `1 _. \, X) d* k        DWORD        m_tLastEvent;        // When the last event was received?
9 }2 `0 d( k4 B$ h# ^        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
4 o) f, g& T1 ^8 z5 q, m6 e; E: j        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
8 A( j$ r9 n3 x        bool        m_bPortIsFree;
' Z: z  x; J7 q        CString m_sLocalIP;
        CString m_sExternalIP;
% \7 i4 r+ z7 W" D0 j. e        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
0 z" _- F8 s. g: h        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
5 Q' s: `  P0 u% R$ a' Y6 b        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
# ~" k7 ?" S) I( O9 A: p, e. u4 D* S0 g5 J        bool        m_bDisableWANIPSetup;
" U6 P7 Y- Z6 J) r% z9 z2 v8 n. R        bool        m_bDisableWANPPPSetup;

4 }) l# {7 p$ s
! J& E! p9 ^3 J8 [! o& X};
. u6 F- a0 [! G) y, b

) z: F( {# n! T8 B// DeviceFinder Callback
class CDeviceFinderCallback
+ Q# S1 D/ B0 Q) r1 L        : public IUPnPDeviceFinderCallback
  F' q6 ~/ x! `+ F& q( U' r2 Z, {" V{
public:
( W% e8 F8 ~2 P5 [& G        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
5 M7 s& e  o, @7 g' b" ?
# ?' o0 d( d$ o8 R
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
+ H6 }2 A0 u3 i1 d9 Z- g! P+ g

! n  J- ]" i7 m. I7 a// implementation
( Y5 s" b. Z/ I6 Jprivate:
. L! w2 ^% Y. o8 P9 C7 N        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
9 _" Q/ d! U& K. i/ ]3 C+ N        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
4 Y3 `6 ?% H! A$ M: |& `& [

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 V2 e0 n9 A  o% c2 f( M+ j};


/ s) e6 W& q5 d: W* _  H% m// Service Callback
, D' q4 \' ^( l6 Q9 |; ~2 @class CServiceCallback
/ U  W6 e1 W+ o. E' J# M3 s        : public IUPnPServiceCallback
{
) T; @( u) w# i$ f+ }" N* e9 Rpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
! M9 B" o9 |3 L$ Y% C        { m_lRefCount = 0; }
2 l+ m) d1 Z& r& }$ s) e" h5 g   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! ~& X/ S+ B" X! ?; u  z   STDMETHODIMP_(ULONG) AddRef();
: Z8 Q) v4 V  \, Q   STDMETHODIMP_(ULONG) Release();


// implementation
private:
$ E  G; G; ?, ?6 m" q" V8 |        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


+ n4 z& r2 M& I. D% L  `. Lprivate:
        CUPnPImplWinServ& m_instance;
5 E' @9 C7 A5 O        LONG m_lRefCount;
6 h2 A) ]- `; O' ]% N# N0 P};
+ O- L' I6 S% G7 X( J  b9 m
+ a& p% b* S+ Q2 {
; i$ k6 t3 z7 b$ F: g" f- Y/////////////////////////////////////////////////


7 I6 r  i! t# P% i" }使用时只需要使用抽象类的接口。
0 d- L8 U' C$ a: f3 [CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
* u0 o/ }3 o9 OCUPnPImpl::DeletePorts删除端口映射.