UPnP

天云上人

/*uPnP.h*/

1. 
   ! ^) S9 P6 R+ A. c; Y' ?
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   ' A; Q7 b9 L9 Y
6. typedef   unsigned   long   ulong;
   
7. 
   9 x: i5 e" s& p& L* u% s  W+ k
8. class   MyUPnP
   
9. {
   7 x: B8 @, V0 I, v
10. public:
    & E; b8 u7 j9 P3 G
11. typedef   enum{
    4 ]% z/ b) D/ M$ _/ o' w
12. UNAT_OK, //   Successfull
      s# x' j6 S3 n) U" @. 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
    
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
    . X) _$ {2 K& ]. \# e* @0 r
17. }   UPNPNAT_RETURN;
    % t' F1 @; u1 Y; k4 }" `
18. 
    
19. typedef   enum{
    " h" z0 T) I, K, ]) H& U
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    & V9 l/ b) b! ?6 I2 F
22. }   UPNPNAT_PROTOCOL;
    . W6 u2 g3 S3 n" [2 l- g$ D2 k
23. 
    ; Z' o* H& |! e1 y) f0 k5 P. d
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    " W: T3 J; ~  O( }
26. WORD   externalPort; //   Port   mapping   external   port
    7 L! s& Z/ }- a' K7 m
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ' X- X0 D" A! U
29. }   UPNPNAT_MAPPING;
    8 }0 E. I; U0 K
30. 
    2 }1 `5 S, U3 e- M# b3 J1 U1 ^; I
31. MyUPnP();
    ' K3 ^; F- `6 C2 ?% D; B, n$ m
32. ~MyUPnP();
    7 X. ^1 L6 D- t0 B- R. w' l7 [
33. 
    & `0 w# w! [; b: ^( D8 W: E7 i9 V/ u
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    6 ]( h" ?' p$ l9 N! l
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ) q  k; ]- D7 a2 k+ {$ k% u: j
39. CString GetLocalIPStr();
    ; ?2 b+ l( j1 k  m( m
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    : z, n0 o* H2 k. S5 n! j
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    ( z2 p, E9 D/ h3 \2 D" d# h
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    + h$ v0 `5 s0 m: x0 b7 a
48.       const   CString&   descri,   const   CString&   type);
    * M8 U$ b# |% ]
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    + ?( i5 x, V6 f
54. bool GetDescription();
    / w3 S; W8 P: d! c% k
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    " {5 R3 e! K8 S) D2 l4 [
57. 
    9 Y2 X- e9 ]  Z
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    3 {% b  ?7 m0 E; m) C9 D& z3 r' k1 Q
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    $ G+ M  H% u6 R1 F6 z
64. CString m_controlurl;
    ; T9 a( @5 L0 N
65. CString m_friendlyname;
    0 p  b1 L% f" ]3 s% `9 E+ r
66. CString m_modelname;
    : S! a3 L8 `& l6 a' Q7 q/ j3 F9 |
67. int m_version;
    9 a; y, j, R5 ^8 g- E/ e% l: q
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    % d. G: n5 Z" n6 b
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    " E: e& ^% V, \  W" _
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   7 E/ p3 d  Q% c6 {7 C, i5 u: N
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ; H4 z" j9 T' o
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    ; X) J" E1 ^* {* ^" ?7 k2 k) _
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ; c& g2 b: Q* H5 N, w9 h+ Y$ g
15. 
    
16. const   CString   getString(int   i)
    
17. {
      _' s# n  j! v  T) P6 W+ C  ~
18. CString   s;
    $ \4 C7 b$ ]/ I$ Z% d( }/ o+ g
19. 
    , f/ n& d: X2 m- b* y
20. s.Format(_T( "%d "),   i);
    : O% C2 S: }4 ]3 l3 f- i3 ?, A
21. 
    
22. return   s;
    / O2 H$ R! Q8 T; X( p
23. }
    0 Y  W4 A4 |) U0 G4 a2 N6 L+ M
24. 
    $ C" E. L& Q2 v& \4 M& G
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    3 d0 S: ?/ v9 J' y( ]' B' G* _
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    ! ]- v9 }4 X$ F# O* s
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    7 u( n6 I# ]2 N0 R$ n# r
36. {
    
37. char   buffer[10240];
    : C7 `7 M7 `( h1 N+ j1 p! {6 V+ H) p
38. 
    - M7 R0 x6 h: b4 x9 }5 H
39. const   CStringA   sa(request);
    , Y' k& |& y& E9 K  W3 B; Y
40. int   length   =   sa.GetLength();
    7 p. R0 o# h3 w6 c7 \% ?
41. strcpy(buffer,   (const   char*)sa);
    % I$ i  K) W3 X; ~, f# U. ]/ Q1 ~
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    + d% c9 j  \4 [! K
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - U- x) W- Q$ s6 s9 ^- E: G" K
46. sockaddr.sin_family   =   AF_INET;
    $ Z: D* h/ h3 j4 k
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    : f  y3 `% J" G+ W1 n
51. ioctlsocket(s,   FIONBIO,   &lv);
    2 N/ r5 E0 I. U% z8 t
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    5 Q! z+ a% \* v* r6 Q! z4 R
54. int   n   =   send(s,   buffer,   length,   0);
    5 O3 B' f! b3 X5 D# z
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;
    # E4 Q' K9 _. r6 B) k' p
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    ) z3 s% d1 B0 `% f6 k* d* F# G% _
64. }
    
65. 
    - a3 k; W- y7 O5 z6 w
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    ) m$ y, E( ]! j
70. const   CStringA   sa(request);
    & g/ \! x9 J' Q6 t2 ^8 |& U6 H
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    # W9 {, y% m" J. f
73. 
    , t3 v8 {' G9 E0 _
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    ! z* ?/ ?9 e+ D' N. r
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    , W2 @/ F1 s: S# Q( h, Y- a3 G
79. 
    4 p/ X7 E8 C& [, X
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ! ~# t4 @2 t" }8 f. t5 d& U- w0 {
84. {
    
85. int   pos   =   0;
    ; `& t) n& n& N; I
86. 
    ' [; U6 q. D4 _8 {
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    2 b: `1 m, E9 f
89. result   =   response;
    
90. result.Delete(0,   pos);
    9 d) d4 D" q/ a$ S+ h; [
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    ! n) ^+ J8 ^+ h0 C0 u9 g3 M
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    3 I5 I3 Y& ~7 k0 K6 E
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    , }* w. F  F) N. ?8 }$ }
100. {
     * x# G# o3 e8 b$ }+ ~
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     5 S. D% A9 d& H6 u9 T$ j. u
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     5 |4 o. X7 \* B8 E9 Z! U( M! z
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     6 F( u: p, ^. j: q+ j7 Z" J
113. 
     . [3 k9 A- l: w& F$ X
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     " `" \+ p- @  d: Z
116. {
     
117. m_uLocalIP   =   0;
     ' t, n( b: s% T5 B% ?
118. isSearched   =   false;
     
119. }
     
120. 
     3 h6 P0 \- v$ N7 Z# Z  Z, w4 j& \: Y
121. MyUPnP::~MyUPnP()
     4 g6 {% b' i% ]. z
122. {
     
123. UPNPNAT_MAPPING   search;
     7 |) L  v. }8 ]) f0 l
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . l/ O3 V4 F( ^" r, v: O
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     , Y, F3 _! a" k9 w0 Z, s
131. }
     ; F3 t$ R( L7 b
132. 
     + S+ n" s$ t9 U
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     - U, K# D9 I+ u' w2 z$ s" j7 n
135. {
     . T& |7 L* \" W& R
136. if(version <=0)version   =   1;
     # S  R1 ?! y* Q$ \
137. m_version   =   version;
     
138. 
     . c8 A  n: A; |/ s7 l
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     0 C1 |/ b$ w- {% X* O6 x
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ; b6 M+ \! I& v$ c1 E
144. };
     ) \( Q' ^" Y# X( G
145. 
     9 L9 ^, h! Z5 K% a
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     ( J. `1 o$ ]$ i" T$ b3 O
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     1 N, x6 Z2 y9 Q% n, V% z; b% y9 w3 R- w
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     * i7 Z: F" X4 S3 k( G: B
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);
     - E% y( |) a7 |) L. b
159. }
     3 Y0 K4 Y0 q0 K
160. }
     / z5 Q# S  H; c; M; ]' E! f1 z# ]* e
161. 
     , _! U" e6 L# ^7 g5 t* o& H  r
162. Sleep(10);
     / ^+ u0 `/ f- V  R5 |; a$ @! ^
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
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++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     8 q1 e. j1 N' J! g
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     % n6 L) @/ F# J* L7 t; A
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     + |$ {: o* z* x& z/ P
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     8 n7 m  d5 P  S* x8 R
187. }
     
188. }
     
189. }
     
190. }
     $ m& F' @! z1 j8 b8 D3 p; a
191. closesocket(s);
     
192. 
     
193. return   false;
     6 j+ r6 E' l8 p) t6 r
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
1 u9 [. ~0 b/ `  F1 e: Y
. Y, b( D; ]( r
///////////////////////////////////////////
. F. L& |. w( @$ U2 p//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
- [1 A' ]: X/ M8 u& Z: j
8 z7 |. v4 i; P3 _$ S
, a" _% `) y, r#pragma once
  }' S  ^  W0 h1 e4 [1 a, O( }#include <exception>
6 v& |% A1 l8 Z# m
9 K7 ]( V; k" s
4 C0 h' u1 I* l  a! a. B) A) S2 a  enum TRISTATE{
7 e& R) M1 a' t% V2 f        TRIS_FALSE,
% s+ B. z3 D6 {. a0 B        TRIS_UNKNOWN,
9 F/ |2 X' W  e2 Z- F* L5 L) |        TRIS_TRUE
};


/ X) M" T$ Q+ Q4 a; E- ]* r9 N3 penum UPNP_IMPLEMENTATION{
$ a. p' C5 Q9 o; d; A  y        UPNP_IMPL_WINDOWSERVICE = 0,
2 V- @) U' F6 |0 _        UPNP_IMPL_MINIUPNPLIB,
2 [; ]0 X4 D& o2 P7 i; G: ^        UPNP_IMPL_NONE /*last*/
- m' l1 S! n, V( Y6 a5 {};

) T8 R( ^+ \9 S
+ ]* d, |) R! J& G* B) E

% M( F: h# v3 @9 w- r5 R" `& ?class CUPnPImpl
& {/ o5 a  t: i$ ~1 ?- \1 j{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
2 n6 Q6 ]" c! u: i                UPNP_OK,
                UPNP_FAILED,
3 y* V4 O0 q! Y& r7 ]' `6 |                UPNP_TIMEOUT
. _3 y. N, t4 i4 {; w( l; V        };

& R2 F7 O) k) s# n9 M
% @  A  P" E) }! E/ k8 M        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
4 L! U$ P. n2 r4 \4 P        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
) F' X+ {$ v) ^1 Q9 U        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

2 n/ _) |6 H& M0 Z& {  e4 U
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
& I/ k0 R2 {# q$ x1 L( c1 ?# w0 L        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ ]. u' j# H  ^. F  l7 f7 a5 C        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

, f+ y; A/ S- z. t# {$ ^
9 k& y( L, r$ E( ^0 G3 Y2 |// Implementation
1 x; I: @) s0 c3 A' @3 a; Q% cprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
& f& e3 {) F- Q' K        void                                SendResultMessage();
        uint16                                m_nUDPPort;
& d2 R* ?# Z. C& a7 S& u        uint16                                m_nTCPPort;
: R1 U1 V2 B! ]6 }6 x  P! g) R        uint16                                m_nTCPWebPort;
% C" c% [1 A0 f+ E) `4 W# J        bool                                m_bCheckAndRefresh;

! F( H2 z1 p( F0 h; q" G
0 G* B+ y9 F5 \$ qprivate:
        HWND        m_hResultMessageWindow;
3 [! ^  E5 ?8 X, G/ R2 O6 p        UINT        m_nResultMessageID;
2 i" V- F3 G+ s$ b& T' O

};

1 X/ m8 L0 B- \
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
+ Q# [+ a5 t( K' e  P3 l0 ~{
: \, W  N- E$ t% Y5 a8 ?7 w9 n4 u7 opublic:
- b. h: O. m8 X. m0 M* P        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 f- \: i2 c4 K9 R+ n- F$ b; x        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
1 G% z# E- \# D7 d        virtual void        DeletePorts()                                                                                        { }
3 E2 {9 E, G/ N# ^0 s" J5 `        virtual bool        IsReady()                                                                                                { return false; }
" {2 Q) g7 x) _) ^. ]* C        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
- y0 N1 u  l* {9 z, {: {6 x3 L
( [$ i- r0 \$ \9 f. z
/////////////////////////////////////
+ L. Q* |, y4 m1 H& e) \//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
2 y) N' Z& ]* M; P3 ?$ a7 ?, H& z; K#pragma warning( disable: 4355 )
6 Q0 T7 Y& q- v( l4 H4 S+ N

) F4 j, t9 S9 D4 C+ z" x#include "UPnPImpl.h"
#include <upnp.h>
# V0 T" [, x7 B, T& E, Q* D, ]#include <iphlpapi.h>
; g( F% k% B' M8 V- @#include <comdef.h>
#include <winsvc.h>


#include <vector>
$ V9 h# `* F$ ?2 k$ w#include <exception>
3 ]7 u1 L! \3 ~- r1 \4 e  F#include <functional>
$ W" R0 c& _7 r2 k7 e
% }- K$ M; l& {8 R; }: V. {& @
2 ^/ w5 p& W; L; @9 c, J
  Z' X& z; w: p4 B: s& B
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
2 t! ?4 ~1 i9 z! u1 j1 s+ P/ W" xtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
& f; j5 ]( c4 Qtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
  h! d5 {0 Y! s& A+ z5 Htypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
- d( X& l* T$ u% t# o4 g& Gtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


1 r8 h) v6 Y  V% O( X3 \typedef DWORD (WINAPI* TGetBestInterface) (
1 a3 D# f9 ~, v  IPAddr dwDestAddr,
! G( d& r, m1 g& t# W) z  PDWORD pdwBestIfIndex
);
$ Q0 m6 i% h' }, F" C: O5 i

( y6 V9 ~" L4 A+ Y) {7 ~typedef DWORD (WINAPI* TGetIpAddrTable) (
# c" I, Y1 i' S9 l6 d8 D! p7 p* _  PMIB_IPADDRTABLE pIpAddrTable,
4 ^/ N+ d: z" y  PULONG pdwSize,
; C" a5 r0 w' W/ x' g" T  BOOL bOrder
);
6 S9 Z& o. s' o# j

* R7 q0 D7 H0 F, vtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
* i6 s) r6 m# y& ], w/ p- a" h$ u);


5 m) T" m2 V# i8 p; O) `# a  RCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
' O/ s% d8 x6 z# D- ?

class CUPnPImplWinServ: public CUPnPImpl
8 w0 O# C/ {) [1 F{
/ Y2 W# }$ d. o: [. \+ W        friend class CDeviceFinderCallback;
, c; K2 A, r0 S1 V7 F        friend class CServiceCallback;
// Construction
8 u1 _5 ]7 k; h. a0 fpublic:
        virtual ~CUPnPImplWinServ();
7 d9 v- \" P+ k        CUPnPImplWinServ();

( O' q: c9 Y5 _2 N. w
' o4 g6 ?2 [: U2 _! Q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
4 K; J3 j% B3 n( t) M3 ^* w7 u0 V5 e        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

! ?* _3 m" v8 f( {, \
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
1 x% B* t4 V( s        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
4 D  H3 r. M6 A) W
0 w; R$ x. z9 T, H7 X" a
protected:
' o( Q1 o) G: F( @; ]' y        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
6 r9 n6 K4 X7 X) n) z8 ~9 P        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
9 u* g! T3 U+ g7 u+ H  g        void        RemoveDevice(CComBSTR bsUDN);
1 j' t; ?: X7 A2 _  a  t, X$ e- q* c        bool        OnSearchComplete();
        void        Init();

0 {2 T: r/ M* v& N
        inline bool IsAsyncFindRunning()
        {
- ?; S" \. M  ]9 e7 A                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
8 e" h# S2 {+ \8 g4 B                {
3 v! F/ u2 i% ]. q1 {  O4 |# R                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
8 O4 V1 {) A  j# P4 S& ~  h! M                        m_bAsyncFindRunning = false;
. o$ S: Y- J* M) W; L, i                }
5 C6 I2 W  d  P$ g; Z; O1 t                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
: g" q9 D" p3 c$ G; }+ T5 L; p+ m                {
: S% g7 D# w7 p                        TranslateMessage( &msg );
4 `' }' t* D! O: F8 |' C9 o                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }

+ D" z% \' J8 O( r' M: `0 j! f
        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
% d) V$ X2 Z" J7 d+ {        // API functions
  v% c) h& P, Z/ e5 Z5 `3 b0 Z( {        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
0 Q9 s: x; Q/ H0 A: j& `6 I        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
0 I& X6 v! ]) \1 t        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*);
7 `$ ]+ o* o, [4 X" X        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
) [: R% c! _7 {0 G$ N
+ |  ]. x  f" ~) X- g! g, a0 U
  L; e0 Y; B7 e1 |5 A        TGetBestInterface                m_pfGetBestInterface;
1 ^5 D+ ^4 E' p, T! p        TGetIpAddrTable                        m_pfGetIpAddrTable;
9 g/ |) M0 S- @6 k$ X        TGetIfEntry                                m_pfGetIfEntry;
% v  \% f: K" V4 v- C( I4 E
& a. a( G8 B% \- O5 ^4 _: T
* {2 O/ a( `; ^; o0 T# m; i        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
+ ~4 V9 @* I( Y1 t' Y  l1 U' L+ ^        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ o5 O: ~  [% A2 a                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
! O3 G6 A  W, u+ E' e  R) ~% Y                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
8 r, H$ _0 Q. E) E, t                                return UPnPMessage( hr ), false;
' ^( F3 k# g3 E- B1 I

9 n; y' p6 q' Q. B0 Z                        return wcscmp( deviceName.m_str, m_udn ) == 0;
' C2 d" y; O2 v: K                }
$ n9 I! R3 Y" e* w) S4 E$ Y9 l$ D, b                CComBSTR m_udn;
( m( O/ Q$ @$ L* F- s& Y) ^" p: q& b6 V        };
5 f1 G9 n! e" L- b2 b: H       
( K5 ^6 U( |3 M7 P7 _9 D7 [        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
  s7 ^. o. O7 X7 m6 P        void        StartPortMapping();
2 w( K+ f" t7 M        HRESULT        MapPort(const ServicePointer& service);
: T2 S& i8 J9 l% s4 D0 R8 L) V        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
7 h. }9 G. D6 w1 ]        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
+ S. {0 i: P( e                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
" B2 g9 ?: O$ w. F# P$ r+ b# `
+ ]3 N1 W" `/ ?) V* k2 ~) s8 X' j
        // Utility functions
" ~- `* F. P% J+ \" z- J  l        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
8 k1 Z5 w/ W; ]1 R        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# i: @. V. w4 x        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
! p* }, ?. f: S( Q' {- k: X/ x( H        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
" [5 {/ r5 J# @8 b/ I- i
8 i9 D5 h3 u: n7 U
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
% T  F5 m* K4 P2 N! u. n        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
6 E% ?2 P7 e, Q        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


) I7 Q1 A0 w; k1 P        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
) a4 c1 X& N2 v6 s6 L        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
, h# O9 n# u7 \/ ], o7 P        HMODULE m_hADVAPI32_DLL;
: L( l* B4 u, e. K4 }1 b        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
% e0 {) G$ k! W) E: d- r        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
) w/ T0 Q9 X0 i/ {        bool        m_bDisableWANPPPSetup;


0 j5 ^+ B, E: l) V};
4 T1 ]3 H" h+ g) T" i5 {+ Q
, d( g0 m0 }' A/ E
// DeviceFinder Callback
3 s* X9 x/ n5 x( r8 J3 aclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
5 S* E0 m* b! T7 c! n{
( q; n2 }6 u) upublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 [# ^( L; G; E" ]& I, L' X                : m_instance( instance )
% ?$ N' P  f+ N# A        { m_lRefCount = 0; }

' T0 U4 K& K0 u
+ e4 f  ~" c8 \  H   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

5 g6 e  F9 s' d( q" Z. q7 O
, v: z3 Q+ [) ~8 _% B8 o// implementation
/ x* `8 ~4 |% N. e" Dprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
# N9 t8 A7 Q" d* Y        HRESULT __stdcall SearchComplete(LONG nFindData);
, X$ S7 m* d* L# }/ t- X

; S6 u% K: l  c5 V) _private:
        CUPnPImplWinServ& m_instance;
' i! Y5 v* [8 u/ C        LONG m_lRefCount;
) M3 W9 Q! J" a6 t  L};
; {' j) O) G. h  c& |6 j2 [# I
. s0 W8 j7 Z  v
$ G6 @& D" f% {, X: G) o( f- l( Z// Service Callback
class CServiceCallback
. ?4 l7 Q  ~7 y0 R- _) w        : public IUPnPServiceCallback
. |& k  f6 R0 E' W7 _{
! Y4 ?- I# |3 o; V3 vpublic:
5 i2 {; ~! f# C        CServiceCallback(CUPnPImplWinServ& instance)
5 j# F, b9 H$ v8 q# T# E' E  y                : m_instance( instance )
. D/ B# ]8 e- \) |5 k        { m_lRefCount = 0; }
   
8 q6 Q0 U: S: Z9 ^% F   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  l/ E  ?$ ]6 M   STDMETHODIMP_(ULONG) Release();
1 y/ H! o" q- c) j+ m
. x0 ]# F0 C& S$ E. J7 K% ?
// implementation
/ g5 P' J) M- @; g6 \2 _private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
" ~% S% I' @* H) I/ r  Y# m  p" S5 O- M        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


- P! e& T& m) l' R2 aprivate:
# W5 L( w; g0 \7 m0 i7 X- N        CUPnPImplWinServ& m_instance;
; _# x( U# y% u/ J8 \* i# q        LONG m_lRefCount;
};


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


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
! V1 l: c. `6 j3 ~; q  L0 c0 |' wCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.