UPnP

天云上人

/*uPnP.h*/

1. 
   % L) o; Z: J2 r* p& _* }% f8 w
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   ; U+ \" E- h( q) ]6 w' F
8. class   MyUPnP
   * u+ l4 u+ T) k+ T4 D8 u
9. {
   
10. public:
    3 T" O* h, a& {+ f: Z8 U
11. typedef   enum{
    * Q) R% Y) s) \  J6 N1 n
12. UNAT_OK, //   Successfull
    8 k, Q% z+ o* ~) d, i5 W
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
    % |; T* S6 D: |( T/ B( D- E6 s* D
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    % M" ~: V! _: J  @! G8 {/ r
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    7 ]* `$ q" R1 A5 q
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    , F1 o( i+ w) I3 |
22. }   UPNPNAT_PROTOCOL;
    / G7 H) M; W' l* S2 v8 [1 b( j7 H: Y# i
23. 
    ( {& |! w  u  ~' }
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    & W9 L$ R( Y: J* m$ o) q
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;
    7 F' j- U& s+ X' n3 N2 J
30. 
    ! R! T, p! }( @( ?0 e
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    1 m9 |9 s) i, f, `6 \
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    & @0 l: ]2 z! S$ y+ C# V
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    - V. G% K$ i6 i5 o( C
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    ' I! p2 R, @4 B, x* i, T+ @
43. protected:
    - j# C8 ]: ^6 t/ h9 _% x
44. void InitLocalIP();
    7 i8 \  e6 k. g4 L8 y4 U- S3 z
45. void SetLastError(CString   error);
    1 D) Q5 N$ `/ s4 Q2 N
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    1 M& D2 A* N( p3 X: S" H9 V& H
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    1 a" u5 t6 T0 l
52. 
    
53. bool Search(int   version=1);
    ; l7 P/ U& n8 P- h: B2 R& Q$ y0 X
54. bool GetDescription();
    ) S8 s% ]. S: Q! s, H: e
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    7 K1 C- F& J5 h
57. 
    % s0 N! c' V. E0 u
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    : i2 Q' l" @% p6 i
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    ' ?- Q3 t/ ?' m3 m6 E7 H
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    1 I1 {5 G: w* K( B  {( z! N5 A& W
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    + H4 {% ^6 M1 b+ _8 l) Z
75. 
    5 f* ]+ b  L' C. `! a6 u- H" ?
76. bool isSearched;
    
77. };
    6 ^& m0 X4 h; Z- @! y
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   9 K  ?. g! O/ R2 U! v. f4 s  i& l
3. 
   
4. #include   "upnp.h "
   , g, |4 r+ H; w+ N( Q; F. m
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ' J3 i! a* |# t8 I2 X9 Z
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    9 @3 B0 a! _; s! s4 o
11. 
    ' H) u3 @9 W2 s$ R
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    : J2 L% h' U+ B( n+ H" n" k. d+ d$ P3 `, D
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    . G9 e) K# r  \8 f, N/ s8 d
16. const   CString   getString(int   i)
    ; h9 j! D# M. F6 T
17. {
    
18. CString   s;
    7 b, o( h9 K5 P6 g& w3 V3 B
19. 
    * P: w6 i0 E( O) q
20. s.Format(_T( "%d "),   i);
    
21. 
    # @& k8 P: f) h3 G
22. return   s;
    
23. }
    8 W5 E' z+ ^- u8 g$ l$ m3 B
24. 
    9 p  \1 ^6 y3 s! I" Y, Z+ E- T
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    . c# H/ N9 R# B, h
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    6 P; x7 }& u3 O+ T# @2 @3 g' h
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ' R2 m7 M2 M" J9 y5 V" x; E# b' Y
31. {
    + M* f7 G; _# M( r1 y
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    6 _; ^  a: d, h: T$ r. G
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    * F$ [6 i7 i% a$ q
36. {
    : M0 X& U/ ^: u% X* f
37. char   buffer[10240];
    1 q. K- `8 r/ {& x8 B+ X/ u# T" [; V% W
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    2 V. {) P! `; e5 X( a9 \5 G
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
      W* z, z9 B$ ^7 Y9 \: C
44. struct   sockaddr_in   sockaddr;
    - t0 |1 z* H. ]8 O( A5 M* e- c
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;
    # P. j! I2 K4 ?- v0 W5 y- }7 q& _, i" }
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    3 v1 f' a" i! X1 m* }" }- B  q6 G
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    / `5 l$ k8 g3 j8 F
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    ' K0 d* }  G3 Y1 r& W
62. 
    9 @( x- B, U& r* U* ~
63. return   true;
    
64. }
    
65. 
    , _- y% ~6 R  J. H6 ]
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    # P6 X( k6 d8 N
67. {
    
68. char   buffer[10240];
    / B/ p8 G9 F0 g1 W1 o, i( m' h
69. 
    / ?) f& {" v! O( X/ p: M- C7 o: F2 D
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    9 F$ x* N& p' K. o& G- M+ w
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    0 x/ g2 Q& ~. r6 P* n
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    * J. \5 }0 X# E" Q, A
79. 
    * F8 p$ j+ b8 C! g! z
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    9 |" p+ `0 j8 H8 G. \
81. }
    " y) `) m: `' r) c( B
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
      i! ]: u+ F/ _! Q% K, R$ v8 I
85. int   pos   =   0;
    % Q7 B8 z' K4 {# t
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ' T* x  G; ^! C1 _3 y& j! t# j
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    ( p2 `0 d7 u/ \
91. 
    % D" S3 w% c, M# Q* \' h( e
92. pos   =   0;
    % }; |3 b: r" S3 Q! K3 ]8 U
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    ' v+ w) d) O2 R1 k6 H9 ~: O& r
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    - ]2 O, G" |: F# P2 ^8 n
97. }
    9 W7 S& k% n* y: n! b1 L
98. 
    * G4 T5 J& ]  \! w. H: |* w+ K
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     3 z9 N4 |, s. E* v& k
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     5 r1 W; }" I# }2 L: \( U0 D  Q
107. 
     9 G7 ]. u" x4 b3 S
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     1 R2 A- C( f7 W( [( V. q
110. 
     8 w. N# [* U* X7 s) h; A
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     * p1 Y, v1 V! U" J" b
113. 
     $ D" Y. D: y( k. \: u" U
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     6 N6 A5 `( h) L1 }, V
118. isSearched   =   false;
     
119. }
     3 m: Q4 T5 ^* U7 K7 |# ^0 w- V, ^
120. 
     ; H' k9 f9 w% v8 m4 P
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     3 i7 A$ o4 T- {1 j
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     - e& m( P5 ?2 Z* j0 g, Y# M
129. 
     
130. m_Mappings.RemoveAll();
     0 {7 v9 m& L; T# w$ w
131. }
       `3 B2 b' h, n. D, q1 s
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     6 y2 p' C$ G$ f' H8 l4 r" Y
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     % H; x4 y9 v- W
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     0 e2 S3 e  f6 n% r
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " @) d6 E; C. f* G* S, X
152. if   (!(i%100))   {
     3 {8 P9 K- W8 q* F: E( q" G1 j+ Q
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;
     ! s  p, A: m  X
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 "),
     & K9 l% B+ ~  \# R8 T3 Q
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     / B( f1 V4 @- ~; G
159. }
     
160. }
     
161. 
     8 J6 Y5 T9 Q9 q+ i
162. Sleep(10);
     9 r) J6 U/ k8 |) R. j! k9 t
163. 
     ; T5 @/ q  l9 n4 f, X- ~, X
164. char   buffer[10240];
     8 }9 A% R3 K. k9 R  ~8 S: Z( G
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     6 Q; c8 E$ K3 H+ h
168. 
     8 g; K. v" A2 s0 b6 j% f
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     $ h" v+ L" P8 @
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     4 ?6 ]3 N& r- |* P7 ?* F. ]
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     6 g" v) y/ m, N) T
175. if   (result.Find(m_name)   > =   0)   {
     # W4 ~) Y& M! Q: @$ X/ N
176. for   (int   pos   =   0;;)   {
     ; a9 ^% y3 ]: Z3 m5 P
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     . J4 ^- @% t' ]( R! |- x; b
179. CString   name   =   line.Mid(0,   9);
     ( ]! n5 X, @7 c4 z. x
180. name.MakeUpper();
     . N8 {) R) W' G5 u; ?
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     % `( A2 z) J, |2 D4 l% _
184. m_description.Trim();
     ( @( i8 `. P9 R: C3 C% j
185. return   GetDescription();
     , L. l: z( g7 I" Q/ Q; a
186. }
     5 W) b7 e" `& L# [! h; d( K
187. }
     4 {( T6 d- {+ F$ d1 y  C! ^* V/ q6 c" B
188. }
     
189. }
     $ z2 r& x4 z, P
190. }
     5 ~# Y) i- ~5 I! w: P: {: X$ I
191. closesocket(s);
     * P9 P) P6 i" x9 q% y+ a
192. 
     , x; M8 s  B" S( T# B
193. return   false;
     # i* Z2 m, c3 }0 F9 X
194. }
     . `! j. y* W  m) M- ^% @$ \

复制代码

天云上人

以下有关upnp的接口来自emule，
8 z" T, Y% i4 Q2 d3 R' Q
8 _- g1 {; i2 `  z! D: c! [
///////////////////////////////////////////
* M% v. M, m+ @" x! q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

2 S" ^6 \' v) |% E! r
$ t1 S4 J( W, J! ]& w! F& q#pragma once
9 Q8 R7 q! p' p6 S#include <exception>
6 q' c! G! w7 e# d2 y/ A6 Z

9 }9 Y9 w7 R/ R9 D$ `  enum TRISTATE{
        TRIS_FALSE,
0 C" x, a; [* A% h# x" G        TRIS_UNKNOWN,
; v, S6 }3 ^" A$ m5 Y, T' F        TRIS_TRUE
0 S+ a# h- H+ d. ~- J% A};
) r- `  Z2 h  e; a
8 B  H, q; I5 G
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
. b" F( O- u! d4 L+ B        UPNP_IMPL_NONE /*last*/
};



2 q$ p, w5 i0 S& t
+ Z3 ^" C2 S& O" @/ d2 }! Q* wclass CUPnPImpl
+ g: u* F. G3 A5 W" v+ ^) Y1 G{
3 J6 A1 W" ~) tpublic:
* b( A" [$ U, Q. F        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
' o; _- [* ]9 R# }3 u# ^2 A% X        enum {
, c) m& z' e2 d. T4 |                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
& \3 m# v& h! t7 G6 j- |        };
- Q3 R  F2 _3 ~) S3 l
1 V, K5 r& i) r$ y* h
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
5 e8 B/ R9 N8 m' F        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
. D) g, C$ f* A5 i9 G% [        virtual int                GetImplementationID() = 0;
       
- i, Y  j! ?3 z' D3 m        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- X% ]- z8 k5 @3 m! B+ h        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
3 R# U; j; V& g
  C7 L: V9 h( Z. U4 X& x+ J
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
# w! d+ h& K( x( c0 v; A5 m        uint16                                m_nUDPPort;
% y4 i/ S0 P* l" y: I, }7 `        uint16                                m_nTCPPort;
- D" @* _) u3 \5 {        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

' h& ?: H, t" o; l2 u
6 d# J2 n- _- h9 L' m! Tprivate:
        HWND        m_hResultMessageWindow;
! j8 }2 e& j; u) L& n! t+ t2 w        UINT        m_nResultMessageID;
% m' H: ^3 H8 |# [3 w2 y
' N1 V  q0 |. a
" v& x& U# i/ i2 R5 m# R2 w6 h/ ~};
7 [8 y6 I9 F! Y; p
$ ]# g& r, {' r: ?0 m% Z& x
1 I- {" L; m* \% Q7 \// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
, b: i- x# Z( T- {{
5 e$ Q* N" A  C& e6 D9 m2 Vpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
4 l3 f) U7 H; J( h        virtual bool        CheckAndRefresh()                                                                                { return false; }
7 |' q, p% g4 E% j1 a4 M: e: S4 d        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
% S" E3 h$ K+ T- ~2 y        virtual bool        IsReady()                                                                                                { return false; }
' ^9 ^3 p( R2 u9 g        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


$ s9 _5 ^+ [: p/ d4 b2 @2 G/////////////////////////////////////
3 C; [  b# w7 Z9 a6 }//下面是使用windows操作系统自带的UPNP功能的子类

% R# `- b' f9 o, }8 ]& v' ^2 _
#pragma once
#pragma warning( disable: 4355 )
3 X/ `) r6 F+ I3 W' l
: m, ^) Z# \7 W4 Q
5 R" r( j: ?/ V7 I0 c9 A#include "UPnPImpl.h"
# R- M5 J0 j7 d' M- p#include <upnp.h>
#include <iphlpapi.h>
" L0 ^. }* Z% p' L- u% P#include <comdef.h>
#include <winsvc.h>

% k* c. G, ~; _( T2 k
#include <vector>
#include <exception>
#include <functional>
: I7 [# a) j3 @1 {
- G: p" L8 O; m' ]' W$ L0 P

" d! U$ P& Y, X. p
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
, I3 @9 W2 e: Y' G  }  wtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
  G% f9 T$ q9 ]9 l9 R: Rtypedef        _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;
6 W! F9 L- @6 n$ y# x$ D

/ ^$ j) t+ X4 R+ ?: Z& Ptypedef DWORD (WINAPI* TGetBestInterface) (
+ o) z/ h7 r# E$ \# h  IPAddr dwDestAddr,
/ I  Z) y* a9 n' E3 N/ f% p  PDWORD pdwBestIfIndex
( {( H1 _. o, F7 x* X. _* O);


1 e: a. }4 x4 B" Etypedef DWORD (WINAPI* TGetIpAddrTable) (
3 a+ ~6 F% ?, h* R; T! e5 D9 ?" [  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
$ K" H0 p  y, r+ o- }  BOOL bOrder
9 q% @% U+ R8 o);
7 [& Q" X) q  Q% B
, k) H8 e" h+ H8 U, x
typedef DWORD (WINAPI* TGetIfEntry) (
2 K5 }4 `, b6 W  PMIB_IFROW pIfRow
, }7 D3 u$ n+ H& T6 {" p9 @" W* W# i);
9 Z9 `) [6 O; w7 k6 H5 W+ Z/ e

CString translateUPnPResult(HRESULT hr);
+ K# F5 F  G+ ^/ U& xHRESULT UPnPMessage(HRESULT hr);

" ^( q& W: {+ k5 R) n3 D* T, n
+ N8 M- N' q0 E! R3 \, P  ^5 xclass CUPnPImplWinServ: public CUPnPImpl
/ E& _- i  j4 X2 f. U( S* y" a{
        friend class CDeviceFinderCallback;
* J; p5 b: z) Z        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
) c* I0 l" K, P# i3 |        CUPnPImplWinServ();
! ?- [! n0 M  \- ^8 I% c. `( q
  k" X9 w! Q4 \; U9 a
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
8 z5 n7 m' a/ E8 m! H        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
4 N# M2 b0 }  A8 c/ E1 @        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
1 L" X2 c4 W- c0 V) ?/ r% l        virtual bool        CheckAndRefresh()                                                                                { return false; };

) x/ p$ }8 |6 h4 \0 U7 H6 }, D
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
" ?( X% d$ Y' j        void        RemoveDevice(CComBSTR bsUDN);
) q8 W- W& d- B) e/ ~. v  n0 I        bool        OnSearchComplete();
, \1 f# }) P7 C; j  C        void        Init();


        inline bool IsAsyncFindRunning()
        {
7 y" }# ~  L4 G% N5 u2 e% Y4 _2 c                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
: D1 E' M+ f% e) R- {                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
& F% X. h. N+ i7 K# L! k                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
3 @- Y! o; z+ z1 y9 k1 o$ `                        DispatchMessage( &msg );
' `, v* m, h8 Y! d                }
                return m_bAsyncFindRunning;
        }
6 c6 ]  t$ P2 |: A  `- G& L
9 b' H: }0 W3 F
  j' U  @7 Z" N8 g3 E        TRISTATE                        m_bUPnPDeviceConnected;
* ~: `& F+ Y5 y: l+ {: h. K9 i3 Q

// Implementation
        // API functions
% v) l  @& V4 ~" \, b        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
' J8 i/ Q6 W/ Z6 g- l        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*);
6 ~# E$ @( Y" Q3 [8 p6 d5 b6 M4 d        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
5 ?$ z& v% a" v" T/ e$ D
+ s8 f  D- c$ z& K$ w
2 d/ i' V; P8 p7 o        TGetBestInterface                m_pfGetBestInterface;
! n& t3 q7 \$ U8 r        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
. A5 [( z2 ?* M; D

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
8 b; I7 o0 U# S- r' w- K        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
* X9 y! L, `, d, e4 _2 I# C" c                        CComBSTR deviceName;
& n! a8 W1 ?( C0 P. g2 k                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- }2 G, J2 S( W
3 D0 s- j( Z+ Y' ]& g
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


5 Q; P6 {& M! U& p                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
3 ?. b+ I; r; d" T% M) r$ R. }# A        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
4 N' Y6 {2 A7 _! v1 V        void        StopUPnPService();

7 w$ T' ]* o6 G5 {
9 F; Y/ x  R% |" X. n8 I        // Utility functions
: Z9 d5 u* \! U+ F/ J8 Z' i9 K$ `5 S        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
* B2 c1 U, v  R& j        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
7 }( o( w/ o+ B        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);
3 g# G/ f/ B- ]1 a1 S
1 k, ^$ Z1 @4 a. q- O' ^; k* g
9 ?  ~9 I3 l& \* }% B  ?// Private members
4 p" Q$ ~$ }- i- j- S5 yprivate:
4 v' t5 _3 ~1 z) `( X' D        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;
; l: c+ K' y% z8 [6 R! U        ServiceCallback                        m_pServiceCallback;


$ W) C- {, N) l9 d6 [  [7 s        LONG        m_nAsyncFindHandle;
0 e' c, t. V! Y' v% t9 C# a9 B- j        bool        m_bCOM;
1 D/ H3 D. o( Z: a        bool        m_bPortIsFree;
% V  [# U$ ?3 M, b, ^        CString m_sLocalIP;
7 r! g8 \# Z! k$ s; }6 z        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
: c) r, d, q% ^: }) Q5 r        bool        m_bInited;
0 y1 W! Y  @) C) l, V7 K; P/ n        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
$ l/ B5 m  L( p; ?        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
. F. m! `& U* E4 ~! B        bool        m_bServiceStartedByEmule;
/ n: b4 b$ M' M3 G        bool        m_bDisableWANIPSetup;
. g1 R; h) q" i# N9 }  A5 o  `        bool        m_bDisableWANPPPSetup;


# D* g# U4 L% L7 ~4 U3 i};
- F( ]. T: B9 R9 j# B) d/ u
3 m) V+ {5 O) ^- A! R% R, d& q
// DeviceFinder Callback
# l/ f3 T& Z3 b4 O/ W* n5 Qclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
; I9 x1 |  c9 V2 ipublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
/ Y0 J2 N2 q% P+ e                : m_instance( instance )
# G7 B& `7 }# X; z0 h        { m_lRefCount = 0; }
; q4 ?7 l: M2 j# m) [6 Q/ y! Y

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
; e8 w7 e7 G- P: n

- E7 p  g* A/ q; Q  m, @// implementation
7 s+ P9 x1 `8 b! ]! A* K* N- m; Rprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
3 n9 G+ r' U, P5 h        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
5 w, n; B& A$ G        HRESULT __stdcall SearchComplete(LONG nFindData);

" _7 N+ h0 e) a- y& C
private:
3 h: M& ^$ n. Q        CUPnPImplWinServ& m_instance;
3 K9 I. X2 A4 G9 @) ~        LONG m_lRefCount;
7 Z; A5 g6 b# l4 h+ K* s3 z};

% T3 d6 q; b- T  c" d* ]( S" Z
) I3 H3 C% `! D& g( F3 N// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
+ p. g0 u6 N) c' o2 H' }) }{
# R9 K0 H' H* [3 a* R5 }6 epublic:
        CServiceCallback(CUPnPImplWinServ& instance)
7 K  c; N  f- m  H6 v% F                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
9 m+ t7 J7 x# y' z. z   STDMETHODIMP_(ULONG) Release();

  n6 L/ {* r( l
9 v% h" J0 E5 C. M// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
0 v. e& S% J  \% p3 K3 p8 A        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
. d# E3 I) D. N7 g        LONG m_lRefCount;
};
; U/ _+ ~1 z% j! |+ ~- F

/////////////////////////////////////////////////
0 }9 A. W/ X: L8 u4 @5 `3 Q9 y& l
! b; p2 M( c' Z  A( X
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
, n% N! b1 m; J9 s4 rCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
5 v+ U: o8 q' z1 `CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.