UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   # l& g. y/ e0 l( y- t& |
5. 
   . Y- I/ s  U$ x: \
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    0 f5 M( V+ q5 G
11. typedef   enum{
    1 X, D- u2 G2 l- o. R
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    4 |* Y+ ]/ c: N+ `3 G
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    : i3 X/ h/ a/ ^: d2 ?' o: m
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    " _" \/ b" q4 S
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    / u4 M; ?, y7 I4 j
17. }   UPNPNAT_RETURN;
    
18. 
    7 o2 r$ H) x( z
19. typedef   enum{
    . e' ?% _" z) K$ H5 B: ]
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    3 T2 P; h2 o7 Z* m4 G. Z
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    1 P: b# a* k' ]5 c
25. WORD   internalPort; //   Port   mapping   internal   port
    , I: M( \, s: `* L  y& k% f$ v( ^' s
26. WORD   externalPort; //   Port   mapping   external   port
    / A8 }. c$ D8 g# F; h
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ) h( v3 c* p2 \5 I9 N+ }5 p
28. CString   description; //   Port   mapping   description
    $ H; I" d# b; i6 {& A) W
29. }   UPNPNAT_MAPPING;
    ' [" j5 t! E( l& y+ v* M& D! U2 L
30. 
    # w1 N& C& R& H; E
31. MyUPnP();
    - d* n* I( j6 w, J8 K* e
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();
    9 n2 \% }4 d  w; Q
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    : @9 R9 Q! D" v. A& [: S
40. WORD GetLocalIP();
    9 U# K, Q+ R, t0 z
41. bool IsLANIP(WORD   nIP);
    6 m5 ?- i$ K2 M0 y* G, p5 ?; b6 S
42. 
    
43. protected:
    1 S$ s4 `, w/ S- Y& `3 m
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    1 k- y& f3 s5 B+ u, W  s: b" }
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);
    1 H, B# w! w4 ~( T6 O, p
50. 
    $ P' N1 Y! E* Z
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    - y  N' t5 b! K
52. 
    # w; @% }) F# q5 a" @) C/ s
53. bool Search(int   version=1);
    & l5 R4 Y5 l# t: G
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    * x! q7 e& k' j' k* ^& j. [  l* C
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    # J  p2 [4 F5 z1 K9 f( H& m
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    6 F: O. w) D9 ]1 D
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    4 E8 _4 Y" S7 V1 A# e! _* h) x- U4 ?
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    / F6 ?: V/ N4 N3 T7 |
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    - d, R. k# e8 s3 Z, Q# O
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    $ T0 T+ J/ q8 s1 t
74. WORD m_uLocalIP;
    1 c7 R1 @7 o3 T! n& q( f, ^
75. 
    ; H7 I  @! R" f; x) X
76. bool isSearched;
    $ W& K3 i7 w( p& s. V9 T0 ?6 h9 u
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ! Z6 O; g# J1 j( A; l
3. 
   9 p' X1 a. Y) b, B
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   8 w8 x! t4 T5 U4 d  g
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    / l1 Z8 j$ \+ l4 a
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # j% N; C, q/ \+ T( t
13. static   const   int UPNPPORT   =   1900;
    ! f; b, Z, N; l
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    2 l9 L: l* w4 \0 _
17. {
    
18. CString   s;
      y% w5 f% n( n5 I% S
19. 
    - ]; q( R: w- K5 I
20. s.Format(_T( "%d "),   i);
    
21. 
    ' s9 \0 v) Q- P$ \. G
22. return   s;
    : x. o- M( ~( j- L, e
23. }
    ; @  X  M5 b- u
24. 
    8 w% i. d. x; @! M1 \
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    % x9 r8 \( o% J- c
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    2 x" u4 x* O7 ~
28. }
    . L" d8 D/ ^  U" N
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ! d3 M# [" Y# e8 k" R* Y
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ; \- y2 ]  e* w+ K% e
36. {
    
37. char   buffer[10240];
    ) P$ b9 z/ r) K5 ~$ E
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    4 M( n8 I0 S, I5 Q+ e( o
43. uint32   ip   =   inet_addr(CStringA(addr));
    ' Q6 @% X( {  M' I
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    5 _& l7 g/ D: N% a; e+ d7 \6 }
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    : z/ R( C, \& B' `" k+ f2 e- y
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));
    3 v! [* p* X' i( r# T+ T. D
53. Sleep(20);
    : ?' c! d8 N# Y
54. int   n   =   send(s,   buffer,   length,   0);
    % |% d" Z/ b) |- o
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    0 d! S) O% J5 P/ a8 I3 b3 \$ e% v0 G
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    - [4 ]# _3 b# w
62. 
    
63. return   true;
    2 h; }' ~/ a+ S2 u' r+ _
64. }
    " r1 B2 E$ w6 H2 D& W
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    / v* |  E. e' M8 i2 U1 V) u! @0 }
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    8 E0 ~8 Y3 d/ `
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 w6 X! S% A1 X; S" ]: F5 `3 z
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    / U5 T- P+ m" ^" U0 \$ f7 w, [1 T9 J$ r
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    $ ]# O" c: |1 V$ B
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    % \$ T% p# i9 f/ b. V" \1 G
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    9 B2 o4 Z$ [  u) L! `1 d+ C
84. {
    5 d7 j7 k/ H) X( b# b
85. int   pos   =   0;
    
86. 
    . b" r1 G) U3 \0 S# O( N
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    % H. p; Q9 v3 ]2 r- q/ M. |! F
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    3 B! j8 E& g$ J% |$ t
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
      Y% }, V+ M. B
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    - s) B. b( U: t! w1 c
96. return   true;
    
97. }
    9 ~, z- V5 R. w. W9 L. M
98. 
    7 z. T' z5 ~' E  a) F
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 K# U1 p! z( P9 X1 T% R' Y5 ~
100. {
     # K( U/ D* N" T2 I, _, [; y
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     , t5 w& w2 W6 q8 B6 O
103. CString   property;
     
104. 
     ) Z  C7 B7 X+ h+ m
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     # r; ?% S9 M* z8 j/ S/ {
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     & e. s" ~0 M& l- Z/ A
109. if   (posStart> =posEnd)   return   CString();
     % x, o( p$ c: \% `
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ; w7 W8 L( z& m: u( d% T" Y5 ?; w% g% s
112. }
     * a7 g3 t$ B- [% ^4 b
113. 
     " r6 M9 E3 j  W/ A
114. MyUPnP::MyUPnP()
     ( f7 i$ Z- n, L. Z8 e( U
115. :   m_version(1)
     % W- C5 C% V$ w9 d. @
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     : M! c7 q' o5 c5 O) `1 ?* p
119. }
     
120. 
       j& H8 }! ^7 g" w* |" O! ?4 k
121. MyUPnP::~MyUPnP()
     
122. {
     8 l2 N  t2 r" E2 M' p& Z, ~! P
123. UPNPNAT_MAPPING   search;
     ! V* v- x" J* L
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     , e& y, o- t$ T  i) t; ^
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     & Y6 V1 v& S7 a7 ^4 M( T1 Q3 l# W
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     ! S7 O* C! J4 e2 V* \
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     % [& M; k9 x5 h* O( E) \
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     3 }3 \  I7 {% l
138. 
     6 c. ^4 d5 Z+ p* }  r! ^  w
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     5 U' G( U+ ~. t8 e( J$ q5 f
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     8 D& O6 ]" X! R/ ]& T
148. ioctlsocket(s,   FIONBIO,   &lv);
     # l4 g$ y2 v0 `
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     7 S, ~$ l/ W/ A/ k8 p' V: ^" x
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);
     $ t! R3 n* D4 i9 F. P
155. CString   request;
     / m( N( |, f- X% z1 a7 k
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 "),
     # E2 z/ o$ C" ^, @
157. 6,   m_name);
     7 ^& ?+ T" B( P' w2 n. J# G6 A# a8 [
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ' z) y8 \4 Z% |- a) w2 {
160. }
     2 p. X. r0 F( [: a# Z
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ( x' s0 P$ g" E4 U9 m6 E
168. 
     6 h& W# b3 @0 P4 @3 Z, T. L. X
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     / ?$ K& e5 R! o7 }
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     5 r0 P+ d& i) N: R( g( v. M
176. for   (int   pos   =   0;;)   {
     7 v+ b7 n7 ^0 ^& z
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     ; ~9 ?, i  E; r. {" ^- \
179. CString   name   =   line.Mid(0,   9);
     . M+ {( Q' Y% ?! W+ O
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     + f  X' |+ @) I6 U: ^0 Q
182. line.Delete(0,   9);
     , q( F# W* ?8 n, n8 _; c$ k, G
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     ( S: t5 @* ]" X& C
187. }
     
188. }
     ! M( Q. ~6 |6 w; C4 w: Q1 G! n) Q
189. }
     
190. }
     ' q9 V1 B! `7 F8 J+ ?2 M; N. E- g
191. closesocket(s);
     
192. 
     4 m1 f2 r) H% l8 C0 j# V+ I
193. return   false;
     ; c& e% A% z% J2 C4 P
194. }
     $ @5 ]5 j5 o1 t: ]5 ~1 }6 n

复制代码

天云上人

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

; P; i# a8 S6 Z3 c8 {
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

6 l' `6 B7 y4 ~: S! S. e( K; [; n* ^
#pragma once
( W9 N5 \) U9 g#include <exception>


  enum TRISTATE{
# b8 `2 u/ M# z" W- W        TRIS_FALSE,
! l- {2 l5 o8 @% h+ }) a7 B        TRIS_UNKNOWN,
        TRIS_TRUE
* U- I# D+ ?' C7 Z' _};


enum UPNP_IMPLEMENTATION{
5 o/ U3 K- f6 ^4 |; k* Y: C        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
$ g5 n0 A9 G8 f1 B: C};
: D5 z, `) F+ f6 G* ?; G$ T
8 T6 {. @1 @6 q% {8 L. Y  e$ m- @% _
4 i6 k8 c  P* A/ a
* d( T9 E$ X8 ^( w1 M
class CUPnPImpl
+ _3 C2 V. N3 x* O: p' R{
public:
        CUPnPImpl();
  @  I( r& d1 [3 q* r        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
4 ~! ?( ~: q$ |1 h. `3 S5 s                UPNP_OK,
; J( M1 E" e: }# P0 j; f# I                UPNP_FAILED,
' ]$ X% W' ]6 F8 E                UPNP_TIMEOUT
/ b! A+ A! n8 f, Q& K% t, C1 J. w        };
+ |* l5 ]. T/ [& d9 k0 o, h

" W9 g8 a" V1 Y) x6 A        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
2 X7 S/ i" H6 z8 i        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
' S9 u2 C7 D/ \- m0 g1 c& P        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
8 [$ s% u8 Z* Q8 q6 `! t. H0 }3 g       
- I& r! \0 X' Q5 t' j5 _        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
4 @6 l. H' U& o        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
9 O6 \7 X) ^% Q2 k        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
8 t& `0 n+ U6 K7 x2 K# r$ W& a+ V

+ g& p) ~, h8 U1 v) l5 P5 ?+ [# h// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
  Y5 u7 I' ]0 y# C1 O, o        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
" y/ y7 F# y3 j/ Q1 E+ U$ L
' R* E$ @0 }' ?1 c# l
0 _7 P! n$ L2 S; b0 I# i% E& Rprivate:
$ f- o) S7 i' L        HWND        m_hResultMessageWindow;
! r6 C3 ~/ P! L8 J$ j! j9 g        UINT        m_nResultMessageID;
+ ~# D2 K' T# C8 c2 O2 l! M

};

- K$ x* ]  I* |2 f, c4 N! Y% C( C
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
8 g/ W% {! x/ |6 P8 s4 i$ s' |public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
  f: j4 K5 [; y+ ?, g5 V        virtual bool        CheckAndRefresh()                                                                                { return false; }
. o0 W. D0 e/ G/ ^7 g5 S: {' J        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
# r6 v$ z( L8 F0 ~& ~        virtual bool        IsReady()                                                                                                { return false; }
* D7 _# j% y# g0 }4 |- u% s) O/ a        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ L( {: a0 ^, y- F' f};


" n2 O& X, y& d; o0 u/////////////////////////////////////
, w- i* z* l) x; I! i- G//下面是使用windows操作系统自带的UPNP功能的子类
" D) U0 U4 O1 L  T7 l! C
& P0 u5 b3 k( U' z4 V; F
#pragma once
#pragma warning( disable: 4355 )
# k9 g* a- X4 a+ y: _' L

#include "UPnPImpl.h"
: s( v1 f, o  n  O" u: C6 }#include <upnp.h>
$ \2 P6 I* u& ^7 m% r7 L#include <iphlpapi.h>
9 M0 Z' |1 R+ z- W#include <comdef.h>
; x# E/ V' W0 M#include <winsvc.h>
+ L; R, F% y, h1 }3 _
" H: Q+ z  U0 t
. ?: l' R0 s4 K9 L#include <vector>
' S+ f6 M' P0 f#include <exception>
#include <functional>

: N, K1 [+ G+ g, L$ u2 D2 F( W

- w, N9 [+ J; }$ R6 F
( N+ x3 Y) V* Q$ ^9 k# X6 d% Dtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
- f1 r; I4 M; C/ [typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
% J5 c: M0 ^4 ~9 P! O7 T* B. Stypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
# E! b* {/ o( j! P3 S+ k. wtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
0 U# @3 K7 ?6 q# [% Y' @typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
# d, x) m: w  y  a- z+ Ftypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
4 \% o: u6 P- k6 W- d  PDWORD pdwBestIfIndex
5 P  \& }8 z2 d8 K) R( U. h+ N  w);
7 d/ d& v9 V2 C2 s1 h6 m4 [
; B- \; @7 L1 d: i3 R1 D( t8 b
9 |4 ?3 D' s; o4 T0 rtypedef DWORD (WINAPI* TGetIpAddrTable) (
: v3 b6 B$ ?; L5 p  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

2 c+ a' J) Z2 j4 c( s
+ @8 x- j% J+ |7 U" vtypedef DWORD (WINAPI* TGetIfEntry) (
! X! C( r  ~' B- N7 d) E  PMIB_IFROW pIfRow
$ @1 F$ k% R4 a);
  }+ c- S8 P% @% \
+ Z  C8 R' {8 V& ]6 C" `2 T* n
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
: N2 l: @* k) X! a2 B: ?; ^{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
3 a( p) ^5 F0 t* z% Z; ?& v, Ipublic:
, U  g* H3 b7 F! s& ~1 d$ d& Y        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

0 W6 q$ v0 p* v( a# O7 i* }
/ A! K4 p7 A, W( G! B% R* }# Z5 z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
( c7 @9 T$ D2 w& l# K        virtual void        DeletePorts();
: A9 _/ x- O9 p! J! L$ ~        virtual bool        IsReady();
; e; O2 U; |* @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
( x( m, x8 I; Q# S9 O, f4 S

8 o) `! X: B& ?8 R* `  }2 y        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

1 q0 x) a; \& Y: ~
9 t1 r4 s% K5 h3 }protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
) T: }$ E5 F4 X& u        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
4 g# i" @9 }' f: d. f        bool        OnSearchComplete();
        void        Init();


4 s( d6 E! P! U6 L5 v: M        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
, M( _# U8 t8 f, a; a; J" r                {
$ x* I/ x: P+ X' g                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
- R8 ^; H4 S2 z) @2 {                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
/ T% K2 k9 b0 z& X7 x+ ~                {
                        TranslateMessage( &msg );
" C, M( W0 @* f9 e0 I" [; [                        DispatchMessage( &msg );
( H: }& V3 p: {6 S5 _& N8 o9 Q                }
5 I. b) H" z4 }; M, F3 e4 v% |. m3 p                return m_bAsyncFindRunning;
( x+ x; Q: t/ Y; O7 D+ }        }

( N* K* ]0 B0 o& U7 N8 |
0 i5 [& K+ i% d        TRISTATE                        m_bUPnPDeviceConnected;
' W9 L. q8 G$ D8 P* U5 D+ _

// Implementation
& O: p% S9 p. f" j        // API functions
+ K" ?7 X! @- P+ z+ }. l5 h; S7 ]( c        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
; S# n2 z) R: U, ^        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
5 i8 o' a# n( L! W/ D( l1 ^; m        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
* `: C* n$ i: u* s  ]

% S+ n" \- d* N% T6 b* ]" _        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


6 Y3 O7 v8 p1 v5 t0 g7 k( |' c$ G        static FinderPointer CreateFinderInstance();
5 x# o2 M. c5 K* |6 q        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
6 `, k1 }; Q! f9 y9 ~                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
" U! H7 ^6 V# B* G
  N6 p7 B) G' t! {$ }5 p; b* t
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
4 v7 H- ?- P7 {8 y8 h& H
) `0 Y7 w8 V4 o  Y
9 g  b) W& K& ]( {( Z0 e% z3 k                        return wcscmp( deviceName.m_str, m_udn ) == 0;
( X% i+ @( M" u! J8 A' D# H0 G                }
( H3 k9 B. |5 d. g+ x                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
% V/ ^" y3 c, n5 o/ P! @0 ^        HRESULT        GetDeviceServices(DevicePointer pDevice);
: l' c( f* s0 r        void        StartPortMapping();
1 C) y# j9 B2 x* F" E3 |        HRESULT        MapPort(const ServicePointer& service);
/ f7 N3 `7 G3 o  z% S  P        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
* b1 A! m5 O' u) v8 ~        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
. h0 N7 y6 x% f" l

        // Utility functions
8 d, S% {1 H. u/ @9 ~; o3 I9 k        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
" K. j1 O8 S$ r' A+ Y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
2 h4 X5 }" p/ s9 T. Z        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
( k# ^5 L# }4 M3 f0 m$ t        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);


// Private members
* m# J# `7 B% f, T" Hprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
/ \8 e& e4 ~9 {$ G        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
5 E: T" \7 M7 i! c9 q- l        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
: {# M3 Y4 ]0 z; m$ W. W/ M        ServiceCallback                        m_pServiceCallback;
6 W' c4 S% j6 Z& t8 B% M

        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
/ b) h1 H. i3 u1 N2 M        bool        m_bADSL;                // Is the device ADSL?
6 L* Q6 i  x0 _9 F  d& H9 T& |        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
1 \* L* ]3 P& |5 |. |! P        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
7 C! ?  N! l8 J# [        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
2 |' s+ [0 M( G, U

. |1 H7 X( s& @. _* y};

/ d8 p$ d4 a1 p( q! W  I1 j
// DeviceFinder Callback
  s% K$ e  J, T+ j9 I; B( ^& Fclass CDeviceFinderCallback
* g) S) a- L/ `5 y$ k! S7 n1 f        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
' R- u5 {# z8 P                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
) I- i/ o9 R. P, U8 r( P; s   STDMETHODIMP_(ULONG) Release();
* T$ [& A7 w* g3 X7 m0 L

! I+ {, F' h, G+ J8 E1 h// implementation
private:
$ l6 A7 J5 g9 `% j/ d5 t        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
9 `/ d+ d& W( [5 E; T" b- h7 G        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


& o* M5 H! U& J5 N7 P+ m// Service Callback
class CServiceCallback
% s0 K+ r8 X: h( ~9 r9 b        : public IUPnPServiceCallback
: Q$ g9 B# E: _1 G) _) Q{
, B8 |, q0 r$ J( \* Opublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 T! X  C+ P$ \# O$ s5 ~2 }   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
. L5 u% f' H/ J4 K: W
1 q2 z: s$ V7 l  X- `) ]" c# u
// implementation
; M8 ]" i+ Y. Z- e  n1 L6 d; `6 lprivate:
0 C/ _3 q. Z6 _; H0 w0 _        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
0 E- _& o( ], H        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
  C7 @# T' T: ^" N/ Q' X
! _( y, O$ u/ H2 h$ C+ R! z
) p6 X0 X; A: N9 Iprivate:
  k, Y5 |- V/ D! r        CUPnPImplWinServ& m_instance;
' z, {6 o7 N6 y# C8 _        LONG m_lRefCount;
: t' g4 F# E2 G( M  Y( m};
5 @& s6 A" {4 x) Q0 P& p; ?' a
1 I. D" Y/ ~. ]  @
$ Y! H* m3 s  y/////////////////////////////////////////////////
4 b2 S- C9 {8 P" r5 h

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
) \" Z2 R9 l; }' PCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
3 E6 l  b' B" s3 Z( ?; N6 o( KCUPnPImpl::StopAsyncFind停止设备查找.
5 W+ E5 [, S( d9 l6 o. s+ jCUPnPImpl::DeletePorts删除端口映射.