UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   4 S9 O7 h4 ?* ]  M
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   , U& _! V' T0 @0 x
7. 
   
8. class   MyUPnP
   
9. {
   2 `* {: l  ^% Z) J  m+ y
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    6 t2 n1 L; U  x9 F5 K+ k/ L
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
    0 c0 j' \5 g& \# N
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    " N3 N" c3 F& `5 i$ I. s7 Y
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    - Z7 j: A0 D- j6 G3 S( t$ ^# S9 o* ]& F
17. }   UPNPNAT_RETURN;
    
18. 
    % n; b) O* Q* M3 w' j
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    + e, P3 K5 \' |. c1 d, W) Z7 H
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    & F2 G4 F3 {$ q# G4 t" i: u1 i
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    " @; {/ |" R" Z& S5 e9 |6 _) r  e
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    7 c) B2 z  j3 V6 d
31. MyUPnP();
    
32. ~MyUPnP();
    
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. 
    * c5 ^; f$ k) g8 Z9 G& E& N1 g6 E3 q
38. CString GetLastError();
    * [" p! I  m7 v" y5 x
39. CString GetLocalIPStr();
    " a) e# j: ]/ e  x* H
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    : w* ~. U' j! N' X2 G
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ! P0 M& u( N6 o& e
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ! w0 R- Z* _7 ]7 V; ]2 }
50. 
    , e0 l( R& h* q% N7 N+ G; p
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    * k7 {1 b3 l2 \% T6 c( y
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    % W" ^/ R/ K/ [5 _9 M6 D
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    " d& W" q$ D4 f9 q6 h& K
59. bool InternalSearch(int   version);
    , R0 c! |7 i# t- s1 I# I
60. CString m_devicename;
    ! J; {0 D5 O& ^% m+ u
61. CString m_name;
    . E, f; y6 K2 `  B' v/ W
62. CString m_description;
    % S) d7 [& ^1 x. ?) x
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    5 H) [) C/ I, m3 q, d3 Y: L1 c+ w$ w
67. int m_version;
    
68. 
    ! f5 J% `7 E8 Q1 V
69. private:
    % O! f. k, |9 v% J2 D) e* V" W/ n
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    * F. c9 S& Q9 u. o' I
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    # R7 v) ?/ e) O
75. 
    * _' e7 k" x" T: b
76. bool isSearched;
    , g0 w; J- d- w2 g) f8 t8 B, J. Y
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   2 r" ~0 C7 s: Z* j' W; @2 w
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ) s9 |0 r  F5 [2 j* _- H
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & [' {' u' F  Z, O7 t1 p9 i5 m
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   & L6 \. J. @9 F1 Z6 O) Y
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    " j5 Q. p7 M4 a; V; q- G1 P
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    3 x& u4 g$ B# T6 i# B( t4 N( M( Y
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + P6 C, D' i, q" }
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    $ p4 q) d5 P4 L6 P0 K, E
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    % r- [$ [3 C& O! n
23. }
    9 K) Q  L- F5 a; v
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    5 P" O# Y9 U2 R* ]/ y$ \0 }
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
      @; A4 K. t" d0 [6 |
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    . ], B8 g# t+ a, Q" T
33. }
    / [. J0 g* Z: F. u+ o
34. 
    1 W3 |) a) s2 [( j+ ]
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    0 s# C- ^+ @8 x, c6 {
36. {
    7 s4 k- `; B: k, v
37. char   buffer[10240];
    
38. 
    0 i  R& P' X- W* B9 {* |9 f
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    # t4 B; N* C. ~. w
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    / B/ B/ F# m) l
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . n3 ?# M( u# w# A
53. Sleep(20);
    9 u* l, A  Y' [( }
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    * F" k' n/ u/ g; \
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ' F0 j9 Y" Y% s
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    . A* z6 H8 p' o2 {/ z" x
63. return   true;
    - w  S0 d( v$ c7 W0 n
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * m' J0 ]/ V( S9 Y. O8 e$ l. g
67. {
    * p4 S% Y, e5 `5 h
68. char   buffer[10240];
    " F6 z: w6 X' i7 O1 Q; {* O/ `& c
69. 
    
70. const   CStringA   sa(request);
    5 W# o/ Q/ I% b* i$ ^6 z3 X, c
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
      q$ v6 C+ x- G% T
73. 
    & M1 S+ n( j& l- }) T
74. struct   sockaddr_in   sockaddr;
    # O& S+ A# F: S# y# P  u. d
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    " |3 [8 v  E2 _  ~0 a9 a1 w. {0 O
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 ~# |) V# n6 l0 J
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    6 K( z# i: `9 i
84. {
    6 |5 k+ @2 S+ O8 x6 ]
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ) h  R+ n# Y1 g7 X1 M: Q$ X
89. result   =   response;
    
90. result.Delete(0,   pos);
    . G6 N) {- S$ [$ X# `/ L
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    * F- _/ d0 n0 o- F; D' m+ c
94. status   =   status.Tokenize(_T( "   "),   pos);
      H2 `* F/ N$ U3 P
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      R+ S; Y6 P* w& y& T
96. return   true;
    $ H8 [8 y+ V& D* n  r
97. }
    
98. 
    ) j: r! f; L! f: f" w' d* _8 M
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    " \. g/ l  ^! \( z
100. {
     / y& P2 L# N$ a0 X! {5 F
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     " f0 d& ]  G$ v
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     " H7 b' \& |4 k' c5 a# W' r3 T% y2 A
107. 
     : \/ c' m- Q( _0 Y  N/ x  f4 p
108. int   posEnd   =   all.Find(endTag,   posStart);
     " e" t6 t, ~) s6 n& K
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     & q) ~0 {( o/ M- T- {) M0 D
112. }
     , X5 U4 t3 K; ^0 p3 v! P
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     " [# {3 G, U0 v# f
116. {
     ! B/ @/ o; b- q3 v" Q1 b* y( }# T
117. m_uLocalIP   =   0;
     / o) c  t, f7 A% Y$ g, [
118. isSearched   =   false;
     # C/ S4 [3 w0 c8 T
119. }
     - P# |: I5 `9 \
120. 
     
121. MyUPnP::~MyUPnP()
     8 u% a) V. Y0 W  p9 o5 g
122. {
     
123. UPNPNAT_MAPPING   search;
     6 ^/ F  w/ M3 S" M& [
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ) V4 Y7 [8 `8 O, Q' k: ~! Y$ E+ }
126. search   =   m_Mappings.GetNext(pos);
     1 C3 h) t6 \$ Z( G* l) G
127. RemoveNATPortMapping(search,   false);
     
128. }
     - d7 \9 h8 z) k; d
129. 
     
130. m_Mappings.RemoveAll();
     3 B& T4 `4 f! \2 `5 C+ q
131. }
     
132. 
     % s1 s" G* D8 v/ [- D* |# D
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     0 n, b. \$ _! T" k6 o6 d
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     . ?" g" y& A7 G9 w. o! P, Q9 s" B
138. 
     
139. #define   NUMBEROFDEVICES 2
     2 @0 W' ]7 Z( [7 u! n: _- a8 f
140. CString   devices[][2]   =   {
     & c( q, ^$ f, P
141. {UPNPPORTMAP1,   _T( "service ")},
       e* W& a! A  X9 v9 N& w
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) U* G* W9 H7 c( q! _; _; G
144. };
     
145. 
     7 l! e! @& S# Q% X0 l+ A
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     $ ?! X: O! h! h0 ^- `
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     , |! v! ]8 V+ m+ H
152. if   (!(i%100))   {
     # C' |3 O, U) ^4 W
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     " b! J+ }, |* h" g0 i
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     9 j  x' s0 I' X- u3 X5 ~/ ]) ^8 z
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 "),
     $ u7 L( I$ |, d" N% P$ J5 q6 R0 {
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     $ d' O! `" a0 q, ]$ z4 ~
159. }
     6 [; R2 z$ e3 o! U- ^5 {! E
160. }
     
161. 
     
162. Sleep(10);
     ' G& Y; j+ R/ x+ \9 X7 j
163. 
     9 @8 I% ^4 u* @) ~2 s# u4 L7 f
164. char   buffer[10240];
     4 t! @; ^; r6 u( z/ V  T1 b  c
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ) F, \  |7 ~0 q5 N) q
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     $ @0 x3 W, ~3 ~8 w/ V* O; u
171. if   (!parseHTTPResponse(response,   result))   return   false;
     9 f9 H+ L- ?) }" n+ L2 a, ^
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     $ {1 v- ^3 F9 D" P7 g2 d3 M& H* U5 ~
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ) `& D- b+ b% U6 z7 W6 M# G/ [
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     0 f4 T6 y, |4 |1 G( U. T
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ) R! L( d1 m8 y- w4 y2 l6 i
180. name.MakeUpper();
       t3 n0 C7 W+ h4 J
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     + U4 y. j( `; k/ w
183. m_description   =   line;
     : m: O( e3 b6 ~! J6 ]
184. m_description.Trim();
     * K5 s( i, z" g) W5 M
185. return   GetDescription();
     4 c2 u$ r1 h! ^  r1 ^
186. }
     ' U# L3 Q! `, T6 @) G+ ~5 Y( j: e
187. }
     6 c8 ?/ M6 z% f5 K# m, @% ?8 \) q
188. }
     : h2 T5 L& F: A! [; v
189. }
     , m9 S' g" M$ j' n9 H6 C
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     # c) K. x. M' z6 s
194. }
     

复制代码

天云上人

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


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

: l9 _3 U, {7 O8 h( l$ k$ y
#pragma once
, w' ^- s6 p+ T5 u, {% A+ o#include <exception>
' l$ H0 `9 l+ x2 T* C
+ f% F4 e0 u' v& u
$ d5 Q( X$ i3 U7 \  enum TRISTATE{
; K( m* F, ]! o4 u* r4 h        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
0 W; F/ N0 o9 {& `9 o' s$ x};

, U1 j' o6 J. P' k: p1 H) w
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
+ d5 u! i6 [, @+ P. O        UPNP_IMPL_MINIUPNPLIB,
' g% q$ a) _# {2 ?# F. W  }4 L        UPNP_IMPL_NONE /*last*/
};
* H. s7 p+ E$ ~3 V9 g/ {" W



class CUPnPImpl
8 |7 w3 c- g" C4 Y' Y5 E{
public:
9 V/ N* ?& T0 o# d; b/ U        CUPnPImpl();
        virtual ~CUPnPImpl();
- {+ r$ n& C) f5 c' m: ^+ T" e& _        struct UPnPError : std::exception {};
        enum {
2 r+ u* s" D2 |1 c' d. J' c                UPNP_OK,
+ |* r" w. y( B8 ~7 D" n                UPNP_FAILED,
                UPNP_TIMEOUT
        };
( Y) G5 D! [# P# }9 a& c
; {, l5 ]0 M, \- {, N2 G0 N  w
9 B: G7 u9 O7 X& g! R1 x1 S        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
1 a7 A) E5 H3 k& r$ ~8 R# y        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
8 ^4 `& K. Y6 @, l4 W- D7 T( L        virtual int                GetImplementationID() = 0;
5 D5 ?2 G- S. h' V/ T       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
% H( l: K; J+ Z; m; A3 R! S  I- \        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 O7 C/ j' T% T# L        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
' ?" \3 g3 v9 D; [; O- lprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
5 o8 e( h+ a1 z! O/ a6 w' L        uint16                                m_nTCPPort;
$ m+ E+ ~* n/ @& O/ G  z        uint16                                m_nTCPWebPort;
% c+ e$ R/ w# |, b7 y0 W( \, T        bool                                m_bCheckAndRefresh;
* S' M0 H$ r. Z  C7 j
. s% g- m4 b. g6 L" o
" O9 b* s. X/ ]" w$ A# Nprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

" k+ s3 y9 `. }" k: ?, M, y) S
! L' j( I( v* c. D};
+ `* c" j$ i7 Z; V7 Q3 b
5 K+ B. u6 @1 o  u
* p( s% c! a: i4 ~# ]6 Q// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
5 ]; P; T5 G7 N9 E' M{
public:
1 U1 v9 \, q9 C- w        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; W  c9 g4 T- z0 p( T        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
4 Q) W3 n5 N4 X  K        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
3 u/ I5 m) s2 V& h& T};


( E( v# C& k4 y- v2 \/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

$ L9 Y& o2 e4 r" X1 W6 K- x  R; h
#pragma once
4 l) H3 J9 J: y+ K1 k5 R#pragma warning( disable: 4355 )
% b' b4 J( n6 B; o: o8 [

* h5 c/ L$ ?  J; U#include "UPnPImpl.h"
1 W5 _' ~) E- {% p! v# f#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
' T' Y/ j5 I1 H* k) O2 ^

#include <vector>
#include <exception>
#include <functional>
5 S5 i2 G6 u# B2 e/ r1 ?
, W) t2 L% C* r" ]- }) [6 P$ j
2 w& M, M4 i: B6 ^
( t7 R/ Z$ c1 b$ a( I1 l
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
# w3 ?+ v* j( U' P' Gtypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
: n$ Q. A7 x, b7 g" W" ~typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
3 M- S& J# T7 b3 q. }7 Ttypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

' o. o' F2 h- n: ~: c3 Y9 ?
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
% j5 Z5 }) j9 ?7 E, Q. y  PDWORD pdwBestIfIndex
);
5 C% y; R4 A/ v. z: }7 V4 ]0 x

$ k; `+ q5 B- \7 N8 btypedef DWORD (WINAPI* TGetIpAddrTable) (
+ ?' V+ k" L6 ]3 \  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
" g8 a7 {) U" Y+ s) d' C);

& @  w) @$ f7 O6 s
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
$ r, I  Y7 z) _2 J5 q, h, p$ M  \4 Y6 qHRESULT UPnPMessage(HRESULT hr);


5 u9 h: f& _& }2 b$ R( V" }class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
/ k4 b/ ^/ Q$ c& l4 F8 z8 ?        friend class CServiceCallback;
// Construction
/ u$ E1 ~$ l/ s) Z1 K, s& \public:
5 j# a4 G7 A. `" O' U$ r        virtual ~CUPnPImplWinServ();
1 X5 S% s1 J) y0 B$ J, B- {        CUPnPImplWinServ();
! v. n& P3 {* Q

' L( A: G6 M/ v        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
% i" C. d! c6 ]% v4 S9 ^  y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


" i" Q8 C8 f" [' i) l* O        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
4 q: B$ N6 R# V; ?3 a. f$ R        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
( ~2 q, m5 J! C! _
" ~1 [& b7 w8 ~5 V8 l" `( h
  K$ C/ H4 t5 S7 Xprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
; p# [0 A. v+ t, C: k. p        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


' j& z3 [& M! ?1 r$ D6 {        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
5 M9 _  o1 Y3 a4 _. o                {
4 s! I) Q! H0 S, p                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
7 V0 J2 s) w% A, r0 _                        m_bAsyncFindRunning = false;
! D- \6 B0 x0 s4 K                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
! S$ U( R% ]2 I( {- I, e% ?9 t                }
                return m_bAsyncFindRunning;
        }

/ Q5 n- l' F' y* z
. w1 X3 P# P: X0 ~0 L( M9 y        TRISTATE                        m_bUPnPDeviceConnected;
- I- L$ H; v# `

// Implementation
( u" ]# J1 D; t        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
% {: E( F8 W6 l8 ^        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);
+ h4 h- Z8 X. W; j; {, o( f
  F! E5 s9 R. p% D* T5 u1 e" i
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

) E, r4 Z3 T) n/ G
4 p! m" ?; `6 H! o; w) H( d        static FinderPointer CreateFinderInstance();
9 V+ `# B- `- D* b7 b        struct FindDevice : std::unary_function< DevicePointer, bool >
( _! [# s! @5 z9 L$ D        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ L, \4 W! N; ]                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

, @# b2 C" Q6 c9 ^# ]" a
                        if ( FAILED( hr ) )
4 |+ u' p4 }$ z# V                                return UPnPMessage( hr ), false;

7 D' R& Z. ~) Z, u  ^/ J
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
8 V# Z! w$ @! g" R9 ~/ C' y+ f+ y. [        HRESULT        MapPort(const ServicePointer& service);
* F2 z) p" n0 Y6 W4 |# C        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
1 {. ~% W1 I7 N* k/ h8 ]        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
+ m6 k6 }# u4 G+ j        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
9 K$ Z0 q5 `; `/ H: j; R        void        StopUPnPService();
2 j$ I! \. E1 n0 `1 P( R2 F* J
! v, K/ e& H- i* d9 V
        // Utility functions
: q9 m7 f( y7 k! D& R# r        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
* Z2 G* K" ^1 q* t        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
* |- S, @$ q9 D        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);
* S. i% b' s9 \: U0 D
3 ~* b6 M! ~% z$ C% e- ^
, e! A1 Z7 D+ t% I1 k' `9 Q// Private members
$ g' n- |# i  L0 n$ `, Q) Hprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
$ e& y+ L% z5 v& q7 i4 X8 I+ H1 @        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
  c$ G& ^0 G  t9 D        FinderPointer                        m_pDeviceFinder;
- R' Q. A( R$ f        DeviceFinderCallback        m_pDeviceFinderCallback;
0 ]+ b3 b# b" k8 W* E$ r        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
1 V4 ?/ e+ M" i- r        bool        m_bCOM;
        bool        m_bPortIsFree;
4 c) v, n& X7 {/ q        CString m_sLocalIP;
3 {- n8 `7 i. ~" w& p        CString m_sExternalIP;
- G/ @9 |1 P% v2 C$ |. J        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
/ e1 C7 `" n  F& Q" k$ p        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
% ?/ }, I* x. ?  K. e+ I        bool        m_bDisableWANPPPSetup;
/ Y) E, J# U$ _7 @& @* I- ^3 h
+ M* H/ c/ T8 s; W1 k
};
- J5 x3 b5 r# y% `! q  D
* c) a& o0 H  d/ y8 }( s: K4 o9 z
// DeviceFinder Callback
class CDeviceFinderCallback
+ A) q. |( L8 R, D% q. t2 Z        : public IUPnPDeviceFinderCallback
{
8 i( c1 _9 H# q0 c* `7 spublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) q% o/ B6 s! I) {' H                : m_instance( instance )
- q, e9 q$ m4 s7 f; h        { m_lRefCount = 0; }
$ N8 Y8 V+ x5 A5 f; n" C0 O
: V' s0 A$ E$ d2 \1 _8 Y1 E
* F3 ~7 P: R; C6 r9 P   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 L6 [* u4 R$ @, }4 D   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
/ |7 ]( P7 U$ j) b  G* K
) o7 Z( R! @/ q: K# l
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
! }6 b  ^8 D6 B5 e# n8 O2 V8 N        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' t9 p2 D5 z9 s5 X3 ]; j- `        HRESULT __stdcall SearchComplete(LONG nFindData);
5 i# j+ c* S1 I7 q/ S* b+ P
/ E. ?) z' K7 o1 i2 ^
- p+ u6 ?7 a& _$ W% Eprivate:
8 ^$ D) R" h* }5 I        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
& B# q: t/ m4 p- c4 `; T};


" `' X) Q# f6 l9 [" }* u1 |// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
& C1 {! J) Q/ N* @) M/ upublic:
        CServiceCallback(CUPnPImplWinServ& instance)
! y6 y4 a! R* M( ^1 q! Q  f                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 y4 f" s3 z9 W1 ]+ h   STDMETHODIMP_(ULONG) AddRef();
# @8 m& N' n, `  g( w$ L   STDMETHODIMP_(ULONG) Release();


// implementation
private:
2 e9 r' }) d- L        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

/ O& d5 M+ e0 a) l; I' ]9 I
' j' H4 P$ n: ^) _: x9 v& J: s/ Zprivate:
' F0 B& y# f. \2 S/ i) a        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
  [8 M) J) B4 Q* G: k4 j( `

/////////////////////////////////////////////////

$ x; p1 b& I) `* s- [
使用时只需要使用抽象类的接口。
& G+ Q) l% Q! U$ PCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
. }) k7 M$ d, C( m# MCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
6 }: h1 k  P0 ?. W+ UCUPnPImpl::StopAsyncFind停止设备查找.
3 E, j# P6 B7 d. WCUPnPImpl::DeletePorts删除端口映射.