UPnP

天云上人

/*uPnP.h*/

1. 
   5 @( x( f2 r) s
2. #ifndef   MYUPNP_H_
   0 w' x2 N- Q- l
3. 
   5 j0 q5 @5 t* ^' [% i
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   8 Y5 _) v" o# P0 w! M* ~
7. 
   
8. class   MyUPnP
   0 Y' F; z& R5 F7 ~7 e
9. {
   9 @5 Y) }1 W! z& l
10. public:
    + `6 {' M3 r( w% e6 E& F0 N) l
11. typedef   enum{
    0 G$ `# D6 p; _, E! J0 i
12. UNAT_OK, //   Successfull
    7 a; b4 M/ r  a3 N1 y% v
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    , h3 `, v$ L1 P( g
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    # L, Y5 Z5 Z  j0 S: g+ Y9 p
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    $ n9 i$ w" b# K* A$ z* T
23. 
    ; y3 w3 ^- I  T5 Y; ]' ~
24. typedef   struct{
    9 e8 s& E' |& \2 _$ @
25. WORD   internalPort; //   Port   mapping   internal   port
    ! ^: E/ t$ W. r6 t/ x, [8 p
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;
    ! u: ?9 G, s' n" ~" F
30. 
    
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. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    + w9 y- M7 u- h  S0 \9 x
41. bool IsLANIP(WORD   nIP);
    - Z1 x1 u/ g' P; d- `* h4 P0 t
42. 
    
43. protected:
    6 C' P9 S3 I! D3 L8 B* n  b
44. void InitLocalIP();
    + q/ I& x+ D- b2 y4 ?2 b$ x
45. void SetLastError(CString   error);
    
46. 
    2 S3 z2 k6 _' i2 d
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ' f0 G7 |1 u& k/ X
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    . C4 w9 l7 `9 J
50. 
    # o' B4 ]4 h* K6 Z
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    . c( {7 g1 I! ]" ^
52. 
    ' ?, N  I' ?- ^. r* z1 L3 E
53. bool Search(int   version=1);
    ; c2 e2 l% ?7 G
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    4 [$ [1 a: Y! T. V) `
57. 
    + c' p$ ~% u2 l4 P* S% [
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    2 [6 }) [7 `: F- [, }* C
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    + k; M& j4 p0 m7 K' O. w) P" t5 _
65. CString m_friendlyname;
    * P5 {- ?. t& |* \# i4 Z
66. CString m_modelname;
    
67. int m_version;
    - ?% a* W: u2 o; F: _' |
68. 
    
69. private:
    * S/ w( Q) r+ s+ L  u
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    * g* s, J, l1 k% E/ l* J8 h9 {
71. 
    # T. r% F0 Q4 V( p' w' o5 E6 I
72. CString m_slocalIP;
    2 N+ j: G, F8 e
73. CString m_slastError;
    . X6 J' o% `7 F8 T$ L
74. WORD m_uLocalIP;
    * C  F) l0 h. X! }. T+ f  b
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   % j: A3 |8 z+ K' v" U: `
4. #include   "upnp.h "
   $ |; Z! ?+ r& W' @! B  }' R
5. 
   1 `  n0 _$ K* V7 j$ O# F
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   $ O6 ]9 j& c7 F1 C* ]+ `0 |. |
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   7 P; M! v1 y2 v$ W
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    5 p3 l3 j( O, C2 ~" K) D+ G1 j
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ' P' Y, A& ?$ X/ s) Q" h
15. 
    , |7 z* @; v4 |# ]: x* y, A" W$ n
16. const   CString   getString(int   i)
    + F8 h, K* e( b1 v6 C9 T
17. {
    
18. CString   s;
    ' c7 v8 E3 u) l0 Q% O1 E* M
19. 
    ( W- C* z( ~- e' \% e& b2 P0 W
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    9 C5 h" `6 O! \! p
23. }
    / s6 k! _  y5 t* X5 G
24. 
    ( K3 C% {5 u% H
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    5 ?  C) w3 v9 @  \6 m! G; B. G
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    $ ~, T2 h& M; _9 |5 t4 e  M
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    / ?8 N+ h& q/ D6 J5 ^* P+ `/ y
31. {
      i* V% B- N( N; s) A: k
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    4 P( E1 H& t/ N- L+ `# E
34. 
    4 `/ s+ H; T0 Q. N0 i" f
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    * @/ Q& n* q! ~4 Y7 b9 \
36. {
    
37. char   buffer[10240];
    4 u9 Q" o$ G' w5 v5 A1 q
38. 
    
39. const   CStringA   sa(request);
      ?( ~* m5 f" h8 `6 K1 L
40. int   length   =   sa.GetLength();
    " F% g# @% j  g3 i
41. strcpy(buffer,   (const   char*)sa);
    + E& C" `8 `1 X8 U9 q9 n
42. 
    $ s- k5 d* S/ g: S0 z
43. uint32   ip   =   inet_addr(CStringA(addr));
    ( |, F7 d8 c" W: g# L7 C
44. struct   sockaddr_in   sockaddr;
    ! W; w. E& L6 K9 K
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    4 t! K' B& Z7 @; E6 P
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    4 K# ^9 D4 e$ z- t
50. u_long   lv   =   1;
    ; g) v6 t; o- q) [+ D; u4 U
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    % R/ R& X% h8 D) t- w
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    4 O3 o! [+ |4 y% C. Y
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    6 H- q8 j% o. O5 K" e& m" M* U7 d
61. response   =   CString(CStringA(buffer,   rlen));
    - A# I$ B! |3 i- J4 m# c
62. 
    % c7 f2 M9 X; u! o" \3 d6 N3 o
63. return   true;
    1 B3 r" e7 |& u6 `7 P6 F7 w# m
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    + K7 w# V7 v7 s, q5 Y' u
68. char   buffer[10240];
    4 C0 e( O8 S* K. s: F9 r: d5 H  u; i2 V
69. 
    " p; ]8 o0 e2 r' ~
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    9 V3 j2 r' Y6 L
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    . Q2 K" ^' j* ^8 Q7 N
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 I+ c6 W7 Z0 t. `  w. ~9 r
79. 
    # S- ~% Z! b3 J8 s8 K6 O3 Z
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    2 ^8 Z# w# X- r) w' r/ W
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    2 M0 m# \0 H7 P2 a; d# b: v8 M
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    4 A$ f% K, ~% f8 W1 z
88. 
    6 M# A5 n& |* D, V9 c3 J
89. result   =   response;
    
90. result.Delete(0,   pos);
    4 M$ e' [5 t: F- k) i
91. 
    
92. pos   =   0;
      |. `4 a- I" i
93. status.Tokenize(_T( "   "),   pos);
    " Y, @/ {, l2 [  s" a6 T
94. status   =   status.Tokenize(_T( "   "),   pos);
    ! T+ Z; o4 N/ r  d8 d, ~/ W8 U
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    : \9 C7 Z1 ~/ h. K4 D  [! T  m
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 h- B3 E' K. K0 ^2 C9 j
100. {
     ! U4 b) @- m: _8 Y) O
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( z5 @$ n/ U8 I' L9 E
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     ) Q3 B, Q2 ^+ T$ Z
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     5 X) |5 y5 b8 _! c! R3 [
109. if   (posStart> =posEnd)   return   CString();
     . W9 R3 J' @% i2 Z8 `
110. 
     2 I! E$ s+ O9 r6 n
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     % ^7 l: ~! k  L. O
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     1 w: [. {7 k8 ~/ ^
118. isSearched   =   false;
       F! [! _8 v% N& }& x* c! T2 \
119. }
     
120. 
     3 Z- ?7 B9 o2 B: A' \/ Y7 R
121. MyUPnP::~MyUPnP()
     
122. {
     2 s% m2 N/ W, F' X/ N
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     5 l% @" S- k& m( l& c
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     % b  n- R/ B8 @8 U9 X
129. 
     9 {" w  L3 ^/ \  M3 S; b
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     * {, P( `: S5 l" y# a
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     3 M3 N& f1 a" Y4 F+ A& F
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
       X3 ?. r# ?% X" K$ Y* J. D! ^
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     % M0 v. F, Y$ L3 X4 ~' l! q
141. {UPNPPORTMAP1,   _T( "service ")},
     , R! w; A8 Z6 U2 M- G/ M) C6 Y; m
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
       b8 i! t' H9 I& a' h2 L+ z
144. };
     
145. 
     / a9 J' V$ |' Z
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     % T: w5 z' B% @, i: Q
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++)   {
     
152. if   (!(i%100))   {
     : O1 V# ^- G# _8 `% W
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; P3 V7 K7 \3 E- F9 z1 K* ^+ h$ L
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     2 a  |2 N( }2 T, ?
161. 
     ) F) r6 a% K2 f" T, S; D
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     4 n) u- ?1 j. |9 J8 B) l5 X6 A& X
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     5 P4 p+ Q  V- R: B" `4 ~8 [6 h1 ]
167. closesocket(s);
     7 u' _, l# z; ?# G3 e# v
168. 
     1 c8 a3 K7 F% x; e. `
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + M: w5 \* l' |# D; B
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ! D$ E/ }' A6 V4 |. p' |4 f, G
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     5 H! X( d1 W2 P5 V) ~3 ~
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 |0 C% t. w9 z/ k1 q2 c; e
178. if   (line.IsEmpty())   return   false;
     9 A2 f8 m1 ?& C! C' u
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     & a: W% y0 v; e( t6 U5 |/ ^
181. if   (name   ==   _T( "LOCATION: "))   {
     4 X1 s4 J: X0 n/ M  Y* h5 o
182. line.Delete(0,   9);
     
183. m_description   =   line;
     0 n3 j! w( L0 m3 m! X
184. m_description.Trim();
     
185. return   GetDescription();
     ) B% }- f) z4 C& b& w0 D0 `
186. }
     
187. }
     
188. }
     
189. }
     6 h* }" ?. r6 d# Y' \& O
190. }
     
191. closesocket(s);
     " ?7 G  z" a- B: a* a& P
192. 
     
193. return   false;
     & Q1 r! ?4 t1 t* e( O, \! `. D
194. }
     

复制代码

天云上人

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


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


2 i1 f3 i& M8 M% I9 Y7 O- K#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
6 h* D/ h6 G7 c  i        TRIS_TRUE
};

- @0 }' A( E( ~3 Q* K
, X; ~8 _0 C- l/ Henum UPNP_IMPLEMENTATION{
" p7 x  h7 c  h' c+ U2 e! D+ r        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};




( @- K( j' X# W2 cclass CUPnPImpl
{
public:
        CUPnPImpl();
( D& E0 l$ k; a0 }) e        virtual ~CUPnPImpl();
0 ]4 _. _- H! [& }1 Y. R        struct UPnPError : std::exception {};
        enum {
! V2 x& t" |- B5 w                UPNP_OK,
                UPNP_FAILED,
! D; w/ F8 X" y$ z& N& [; {                UPNP_TIMEOUT
* ?" q" f$ D! N' ]        };
- X& r+ v$ ]0 N- @
% l' ], U, D- B$ x
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 v7 {% c; ?5 E. H        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
7 U. ?, C; O5 ~. [3 t" w3 V6 [        virtual bool        IsReady() = 0;
) _& L, D6 P! A, s; F2 I        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
- M# F! O7 ]. g! d3 ~. d
# d5 F) h6 b( P, Z1 c$ O  F7 o
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 ?; a0 H8 ]0 ^        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
$ A7 g- I- O* G/ m  eprotected:
/ H, o) u" G! A1 f        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
# f1 q3 w  t! [5 M. c4 h' d        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
8 s( `1 P0 k( O; L: c
2 @- u% O( v5 Z. z3 n
private:
        HWND        m_hResultMessageWindow;
$ H6 F" Q) o4 D6 K0 G% R  p        UINT        m_nResultMessageID;

6 n8 C8 ?9 R& [9 J' H  B
$ l6 M% ?0 L- A# u  t& z};
! U6 M& ~% a* M# b: G

6 Z: X2 n/ K' i! e* Q+ V) d! y// Dummy Implementation to be used when no other implementation is available
  a* f8 X' ^1 B! L, Q8 Eclass CUPnPImplNone: public CUPnPImpl
8 G# T* W8 Z: m7 S  E6 C{
; ?; x, e7 }5 [1 O4 Hpublic:
; F6 x- `; }$ H8 e4 C& F        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
! Y8 T+ z2 ]; M! o0 F2 Y8 o# L  @        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
( O* ~5 Z# F, f' |! L        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
. A' ?0 y% |1 I' x        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
" k! Q3 W: o& T, ~* z. m  B//下面是使用windows操作系统自带的UPNP功能的子类

. C0 U  C  ?% g9 @2 U. G! b
#pragma once
#pragma warning( disable: 4355 )
$ F: w0 p& ?) k, ?2 W# }, r

6 o( P# j5 j* `; g4 d& L#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
( y+ R9 r) r( n* M! M#include <winsvc.h>
9 |& {* P/ o/ E0 t' w+ |' [! \( E

#include <vector>
#include <exception>
#include <functional>
/ L4 e5 S3 x3 f$ Z2 ~: _- F- U1 A) K- `
! t. B) b- U# j% T% f$ \


; d; ~# K4 Y7 W, _! L6 vtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
1 V* ?6 C0 w5 m3 @typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
* |0 U4 ^8 l1 Dtypedef _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;
9 E6 _4 X! a* W; Z0 V- \typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
8 [6 r+ k  @% f( f  g
+ ?: T6 ]/ ^* B- F3 K
% b0 ~1 g; @4 R1 G6 D5 Ktypedef DWORD (WINAPI* TGetBestInterface) (
/ A4 @- ]. ?4 Z; h( J& {  IPAddr dwDestAddr,
- E. P* c1 \! N8 B# ?; A. W  e  C* t  PDWORD pdwBestIfIndex
);
; K! D" T3 U" \# j7 H$ q
1 ~) h, R( n9 e. p
8 p1 c* s% ?! v8 r. z& r2 R! Ttypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
# a. ~2 @9 U5 B2 g5 x* v/ ~7 W  PULONG pdwSize,
8 p6 O! \# r# k3 c, u  BOOL bOrder
);
; B- T8 `. N4 g" H
9 w  N% p' E7 ?2 C# M2 @5 |& x
, H5 m1 a! O- r' \typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
* S& X0 w* B- b/ L7 _& i
1 M1 Z9 w% b# h+ B9 K+ E
CString translateUPnPResult(HRESULT hr);
* P# Z4 c" b8 F( dHRESULT UPnPMessage(HRESULT hr);

, B9 A) m- a, L# b$ E) `
5 m- h5 i/ d$ Pclass CUPnPImplWinServ: public CUPnPImpl
{
9 P  x& y, C1 }8 Q/ z        friend class CDeviceFinderCallback;
  Y- o' n$ S% b4 A- D: ?        friend class CServiceCallback;
- H* a* G! ]" U" D$ p+ b// Construction
9 t5 l$ H; F' j5 J) ]) vpublic:
        virtual ~CUPnPImplWinServ();
; T, K3 A* F+ r8 V, r' c& i        CUPnPImplWinServ();

7 O7 }( a- {2 L2 l3 \& i
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
5 g1 L8 ~0 I5 a' f9 t        virtual void        DeletePorts();
, F$ p6 B) n1 u' U        virtual bool        IsReady();
1 V# e5 |. h. _- k# `        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
4 ]1 A* J$ g/ [0 ~

        // 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
+ L! n' {. T. X7 m        virtual bool        CheckAndRefresh()                                                                                { return false; };


. q) b7 _0 J7 ^protected:
# o1 R% t/ E8 d6 j        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
( ]0 ]% L$ A. Z* q5 u: r  l  \        bool        OnSearchComplete();
        void        Init();

' U. ]" z. {/ \
        inline bool IsAsyncFindRunning()
        {
# f: V, {8 z7 O+ }( S: q7 t# P                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
' h5 J8 l$ `, l# O3 u8 a7 [                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
( u1 p6 O1 n' U. E: c) w                        m_bAsyncFindRunning = false;
                }
8 ?$ }1 M$ }+ J                MSG msg;
3 m7 u5 Y) R. `% u5 H+ K8 G7 Z                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
: _/ \$ k2 q1 W- ]% M8 H                        TranslateMessage( &msg );
5 o* x$ y5 ]/ C5 m                        DispatchMessage( &msg );
+ Q( f: v3 u5 M( T                }
5 g9 Q- h3 i: S# j; a/ ~                return m_bAsyncFindRunning;
" |6 g7 A/ y! X9 e4 T        }
0 @( B! [, E( t* g

" @! o$ R9 ]7 P2 c3 i( L' K% q        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
8 g5 x* n( q7 I& x, q' W        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! F" n5 d. n/ P+ }3 Z' J        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
$ Y$ D1 q9 @$ y+ D        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
+ u, O; f5 W( |$ X& Q        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
- f+ s* i8 V4 f- `7 {+ D" y3 E
; f& _0 {/ N3 A6 p  T) W) R
        TGetBestInterface                m_pfGetBestInterface;
! `5 u0 I) `0 ~3 E8 ^) l7 G        TGetIpAddrTable                        m_pfGetIpAddrTable;
" r9 F1 U( X7 G4 ]% p3 B4 Q: [+ |        TGetIfEntry                                m_pfGetIfEntry;

% z# V9 d( ]; A% o; j* v/ D# u
% U9 N( E% b, J+ f# @        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
. `& ~; O. C# W  V% H+ Y                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
# g5 H* f; O- V6 J                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
3 Q* k; n1 D+ t4 w5 ?                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- V4 ]! r+ O1 `3 E! R) _& R
! }. |+ [$ C3 x- s! t0 L
$ y0 a+ c' r9 L) v7 ]                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

$ i; [. v; z- [; ^
+ V2 e; b( a) Y  E3 ?$ k                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
2 A8 c: i0 v" G! _' u3 B       
3 K1 X+ Q0 u/ ], I. T        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
- ]6 M% x1 H0 L, y9 ]        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
# d3 i, |, i* [" O& k        void        CreatePortMappings(ServicePointer pService);
$ E: f( n0 ~. }% k; H. m1 f, C        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
( q4 F6 S* D) U, Y4 ]- Y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
8 I/ Q$ m! y1 J6 n        void        StopUPnPService();
1 L5 B0 Y" M0 @0 ^8 D! r% G) T3 N

' ~' h$ e$ p- Y$ h5 B) e        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
) k& R( o9 u, D# I        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
2 w0 P2 \" A: c/ {- s* ^- x. z& P% j        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

; f0 |+ M9 l0 T
// Private members
private:
& p7 P( k9 M" Y        DWORD        m_tLastEvent;        // When the last event was received?
0 j2 j+ K' S6 I+ H. _* x        std::vector< DevicePointer >  m_pDevices;
- O' F8 W- ]3 I. {7 c0 z& `) m        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
# l, z" k- y! V6 c        DeviceFinderCallback        m_pDeviceFinderCallback;
4 n. l, ~4 _5 `& E5 Z0 e5 T        ServiceCallback                        m_pServiceCallback;

5 J( A/ B) ^& ?8 }* o
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
. a/ k6 o! {# \1 M/ R& f: b1 t        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
9 D7 f& a* c0 `        bool        m_bADSL;                // Is the device ADSL?
4 s& e& S# m6 h7 p. O  U$ K        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
/ y5 k; i. p& x8 |4 y* i$ ?" |( l' h3 Q        bool        m_bAsyncFindRunning;
/ w+ I: ^: O% O6 C$ ]        HMODULE m_hADVAPI32_DLL;
8 {* I/ q. I! E5 h2 \$ I  q        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
/ r. S$ F$ p+ c; `. V( s2 {1 g  u. j        bool        m_bServiceStartedByEmule;
6 S5 U, z# A' M* S+ e6 u  M2 e        bool        m_bDisableWANIPSetup;
: S0 W/ a, h% u. X5 f        bool        m_bDisableWANPPPSetup;
+ L' h' g3 a  a+ L2 v' n

7 B/ z- O" ^9 b) R2 ?};

- r( Y. F& v2 I& e0 k' j
) t: {) r8 ]7 N" D// DeviceFinder Callback
class CDeviceFinderCallback
* N4 U: V9 c4 }        : public IUPnPDeviceFinderCallback
4 K; o# j1 G' p) H{
! S' ]% x* Z0 l2 M( Mpublic:
. s/ F3 Y4 g; G; P) v        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* g7 P8 j, W6 }* C1 R! k" p- p, H                : m_instance( instance )
        { m_lRefCount = 0; }
2 P$ k1 r: V. L$ b/ g4 I
, u4 r* i/ Y/ B4 m
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
/ h4 o% C1 z* V/ e1 u   STDMETHODIMP_(ULONG) Release();
- g8 c" s/ ]$ ?" J

  h( P. G. B  f# T% E// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
4 x5 |+ k1 y5 ]  E        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
# Q# p# i* F7 _% u! i0 X4 X        LONG m_lRefCount;
};
4 b% [- L; l$ z' {7 t( Q

4 @$ P0 d: U* w1 j, {$ [0 R. D// Service Callback
4 {  Y. f% L0 s  _& ?" ~class CServiceCallback
2 P; O$ v. I7 z6 w8 s3 A  _        : public IUPnPServiceCallback
( J. q9 e2 z( X& ]+ _! \) G# |{
, u4 g9 b; b, p( w8 R0 ?5 @6 }- wpublic:
7 f$ |# N( Q- N9 w- Z4 P) N+ k+ [        CServiceCallback(CUPnPImplWinServ& instance)
$ M5 @  ]( K& s: [' p; t. |                : m_instance( instance )
        { m_lRefCount = 0; }
& f3 z3 `2 S" E  Z: A7 Q+ \   
5 Z7 s: A5 J! N   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ v5 V- x$ [4 f5 L: W6 v   STDMETHODIMP_(ULONG) AddRef();
& c- |6 D1 b4 b6 ^4 k   STDMETHODIMP_(ULONG) Release();
, O1 Y6 n6 d3 i8 s  k

// implementation
& z- k; D: G+ Q( k( E1 Q1 [private:
8 C7 u( d# ?2 E9 G- \        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
& m# d& ~; x. P+ u+ v        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


" T4 x5 {$ V- h+ c- |private:
        CUPnPImplWinServ& m_instance;
1 [& P0 P& L5 W: a        LONG m_lRefCount;
};


4 g& T7 `& I' b) O$ L/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
7 K) Z2 x2 _9 u4 l4 `: KCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
: g& ]+ ~0 v0 c1 w4 M: DCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
9 s% \2 J% n' B- m8 ~5 FCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.