UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   : e$ z/ q1 e& w$ x# {# D9 [
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   & p% g6 K5 b. {/ }
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    0 }; g- k( |7 {* @7 F6 z
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
    + L! y/ n1 [5 B) @# K9 \5 J5 k3 c
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ Z" N  H0 v! D2 c: c/ N  P5 g
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    $ `6 \, O9 e; m0 h8 R5 @- K* Y
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    * f* H0 X. Z! H6 _
23. 
    
24. typedef   struct{
    $ E' o! S$ R# x' ?  S; i1 I
25. WORD   internalPort; //   Port   mapping   internal   port
    & a4 B8 s6 |# F% D+ \+ Q/ b
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    7 ~% C0 I1 ]5 S6 o/ o$ N" I
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    9 Y; z: z, `/ }- F. }
32. ~MyUPnP();
    , D# j0 U7 e7 w) p( g
33. 
    / b5 v+ Q: b+ T; y3 ^0 Q7 m
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    6 I, q0 ~; i- ^4 |4 {* l
36. void   clearNATPortMapping();
    , `1 c$ [1 H) r9 _2 ^$ R
37. 
    , b8 j/ d: ^0 g0 E+ y2 E' E
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    - [; z7 e+ g9 E
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
      I) i. K' }/ D# z& |+ t9 v% Y! F
48.       const   CString&   descri,   const   CString&   type);
    4 r& |7 L2 }1 a7 t* u
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    5 v* X+ i5 _0 A- A9 G+ Z- b- x! A% E
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    / [5 `$ R( h' j6 j
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    - L, M) q1 t1 }! v8 I0 y
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ( N" p3 k( [; `
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    * K) ~' p" N+ z& F, F* e
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    2 v# v. L( C6 f" P
72. CString m_slocalIP;
    : b8 a+ y' d6 p8 [% @5 |
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    5 t% k* p8 y; ~
75. 
    . \' R4 _( l2 k3 V% I1 c
76. bool isSearched;
      O( g* w* l: k" |% ]# A0 R
77. };
    " o' e# Q4 z/ n5 P5 T
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ! Q  k' D7 n$ G0 N, }; D: ]
2. #include   "stdafx.h "
   
3. 
   ( d5 I( w5 l8 V' ~
4. #include   "upnp.h "
   
5. 
   % |. o% d& f! Z, ~% R: A
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ; P7 C( G9 Q. K1 J0 Y, z3 T" _
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    5 B. ?9 s# ]$ d2 j% I: d  J: x, {
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    5 Q$ E) r) f# h% [
13. static   const   int UPNPPORT   =   1900;
    ; A* \; _6 w. T+ N
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    * |: ]" T/ t, b; b6 R. W* g: w) W% N
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    , c9 Q7 P* I* o( V& X! F
19. 
    3 D$ L7 v: k5 ^* z: G
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    ) ~' r2 I, h3 E0 {" s; D4 o4 F
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    8 z) [' Z. L$ r/ p7 O' N
28. }
    1 h6 g5 X) u" ^& V
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 `8 E4 N5 p8 ]5 m
31. {
    ) ?: K2 Q! Q, b' O4 |
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    1 u; x/ l/ D# Q# n
34. 
    4 P" r1 o( d- V4 p& U+ C5 L4 _
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    % U0 |$ y: S1 p1 X/ P; W7 A
36. {
    
37. char   buffer[10240];
    
38. 
    0 V* T2 K, ]6 T- o2 p' v
39. const   CStringA   sa(request);
    $ R3 Z. P4 t) Q6 e
40. int   length   =   sa.GetLength();
    2 w6 z% j$ p3 D9 v6 F
41. strcpy(buffer,   (const   char*)sa);
    # S- R7 a1 O7 r& V! x
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ' h) h$ z2 Z! [1 z0 L- J
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;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    6 b) w+ C$ u+ m# x$ Q+ _- t
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ) t* U" i( r% h# z, l: \+ d* \
54. int   n   =   send(s,   buffer,   length,   0);
    & W" ?3 A+ h) {% J2 x
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    5 A: n+ R; n+ H
57. closesocket(s);
    , z- }) i: C+ W0 Z6 D
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    8 ^/ j! J9 I9 {3 k( l' @8 C
60. 
      j' }" D! G# [, y' f
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    6 x2 r0 f" R; N) I! w% @
63. return   true;
    
64. }
    2 N& @1 A  w1 f, m( E+ d7 {# x
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    1 L- H: W* h6 `% j9 M+ ?  K2 ?7 ~
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    9 Y0 u* D) {- {6 M9 G4 m
72. strcpy(buffer,   (const   char*)sa);
    ) V$ z! a1 J& \% S  j
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 @: l  c; X$ [6 b" u
76. sockaddr.sin_family   =   AF_INET;
    2 |( U0 t3 \, l2 U
77. sockaddr.sin_port   =   htons(port);
    : X! i7 S" M% I
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    4 q9 d3 b' I$ g3 N1 i6 }: P
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    , x; b8 F" ?) R* A! _
82. 
    ! C0 j2 ?" X" u1 D3 P0 R0 j
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    9 A6 w/ L+ s9 P9 S, E9 }+ S
84. {
    
85. int   pos   =   0;
    
86. 
    - W  b* B; |: ]
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    " v4 D' \7 {' k, m
88. 
    
89. result   =   response;
    0 H* s! r- v' P. r7 [1 o3 V
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    # L' p4 u& t+ f3 A8 T4 T- E
96. return   true;
    
97. }
    ( p$ Y5 B$ u* ~" r0 d
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     " e" J% W' ?, m! t, y! z+ E% q3 U
101. CString   startTag   =   ' < '   +   name   +   '> ';
     4 T2 h2 X5 C9 ^% T
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     . F" t: \. i4 a' m) C
103. CString   property;
     & r5 p1 ?$ y9 d
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     / G$ B6 G# L% a$ b2 E0 M
108. int   posEnd   =   all.Find(endTag,   posStart);
     : ~4 c" ?0 w. q' u$ x, l
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     : b# V7 J, F: w6 U" u4 [
112. }
     " X/ l" _( e% ]* ?. N
113. 
     
114. MyUPnP::MyUPnP()
     , A, Y) @; U  O2 h8 D4 I
115. :   m_version(1)
     " Z0 k* `" t3 W% L7 x6 \
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     0 [: \9 _- q3 e: Y5 s
119. }
     6 F7 M" M7 H0 G1 j/ |5 P
120. 
     ' m. E1 f6 d. t2 i8 T: ^
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     3 p* G3 ^! s/ {: q
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ( m0 l, x/ Z1 y; q4 `
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     2 y9 ?! j3 ?7 g
130. m_Mappings.RemoveAll();
     1 ]8 N* x, ^4 c! s
131. }
     0 i/ P. T4 A" I# p8 j
132. 
     . L8 F" c4 ?% F3 U
133. 
     $ _& T3 R0 I) T/ ~
134. bool   MyUPnP::InternalSearch(int   version)
     ' g) X! r  E) J. h( ^8 `
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     / v" a' X# [) }* h% a( |4 [$ I
138. 
     
139. #define   NUMBEROFDEVICES 2
     & _9 r9 d( A) h' r4 c' K- O
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     ; A# y& u; U1 }5 h9 Z
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ' j$ e. q5 H7 j  x/ F0 {+ e8 e. v
144. };
     ( [9 n# m" G6 Y0 B$ D
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ; M8 ~/ ]# |$ @* _# F" Q. h
149. 
       y  B6 Y4 I. q! w
150. int   rlen   =   0;
     ' f0 D; d, p, Y' P6 T6 D
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     $ Q$ r/ y% u7 O, E* q6 b! M2 @, r( ^4 o
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     * C0 h$ O$ o$ ?1 S& K0 N0 Y
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     & k; {# f& i6 G: h( X
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 "),
     ( n& N6 c/ p+ r0 V/ f5 R# |* G, s; b
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ) l' \, V0 q/ ]. t3 c; k$ |0 j
159. }
     
160. }
     
161. 
     ! Q" }7 b9 b3 A+ g& B2 ~/ ^0 {* ^
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     6 u6 y% u7 }5 ]4 M# h1 D! O
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     & ^& H% d+ x. H& I: @
168. 
     7 ~' ]6 B' v) X, g8 d% L4 G: c  k
169. CString   response   =   CString(CStringA(buffer,   rlen));
     4 K, m3 H! q8 \2 a
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
       H) G& z+ V$ X! F' g3 \
172. 
     ; Q- k$ X: P( F4 O3 F" g- v
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     % u4 o3 w3 Z5 y. o* `) f. `5 [& v
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     : j  W% c1 }& j& ?
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     * N' _% U: e& W) i$ S! B4 z1 o! A
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ! B# o. s- Z5 ]" W0 f9 V
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     # L3 w- ~- m! o$ W4 [2 o7 H: t
183. m_description   =   line;
     
184. m_description.Trim();
     & Q/ X4 h) r. q$ P  S  E
185. return   GetDescription();
     4 w" _! O# Z9 U) p: D$ j
186. }
     + @" k2 P# A1 f" G: k' a
187. }
     $ c# }$ J7 Q' f
188. }
     1 l% N3 S/ Q3 e- Q) m" K, _- }
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     % I0 \8 v: d$ m6 k  l/ r7 `

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
4 h8 A- j5 i4 r3 b: {

#pragma once
#include <exception>


. l; d+ x2 `2 g  U4 e% A3 ~  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
# w4 b; g. D. R( f. S& e        TRIS_TRUE
};

) Q3 Q' G9 i$ m. w
. A/ f1 g0 b! [# kenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

0 x! v5 C2 K: R' u6 k$ g# O0 S

, q- v* W$ i) y
class CUPnPImpl
{
public:
: |5 ?$ v1 }  b, g- g        CUPnPImpl();
6 t7 i. o! N7 ~7 X/ l0 {1 s        virtual ~CUPnPImpl();
, Y  y' [1 n8 O, y0 f1 R) d        struct UPnPError : std::exception {};
+ O  O5 N" s- @/ g5 h, y: ~        enum {
                UPNP_OK,
                UPNP_FAILED,
' G, {- S& K. }                UPNP_TIMEOUT
8 q# b0 r2 e, u- z        };
7 w4 F2 A' H0 f* e0 \' Y

4 D) |1 G9 U1 }: k' M  u, a; j" m7 T        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
4 o# ?$ X; @4 ^  e' C+ d( j' J; d        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
: J5 G* M! F- ^) F        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
1 e2 L$ S- Q+ a0 \8 l# y* k( T) o

' E2 |# d# o. |$ r/ G  }6 B" {$ `        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 H) _$ J% x! T" n0 f9 D- t$ I        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
8 y+ W+ v8 }1 B- ~        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

* m, p: d5 M: F) @( t- `
// Implementation
protected:
$ q4 C7 z6 l  S" Q6 N8 G        volatile TRISTATE        m_bUPnPPortsForwarded;
4 i1 X3 L8 ^8 d- j( V$ o        void                                SendResultMessage();
( y$ d( i* V! Q% ]        uint16                                m_nUDPPort;
/ Q) K4 W( }, r3 i/ F* `        uint16                                m_nTCPPort;
4 {6 ]# @2 C! H        uint16                                m_nTCPWebPort;
: m/ d' ~( s0 |8 R) p2 O        bool                                m_bCheckAndRefresh;
8 G( T4 b0 A, k
8 |( g( Z5 F! F! x0 V- }0 W. }* u
. k# ^* ~  o% c; M+ I& dprivate:
        HWND        m_hResultMessageWindow;
3 O8 y" d4 Q3 }5 V" C        UINT        m_nResultMessageID;
' d, w1 v- f7 R
6 O. A# o2 @/ A1 _1 S2 j
};
+ T- m# {% n' ^

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
3 ~4 _! Z1 x3 k  E7 J        virtual bool        CheckAndRefresh()                                                                                { return false; }
9 O$ ~( |# S4 Q* n0 j. C; w        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
# U4 m/ e4 q% t' ~1 ]& h: P};
; I: n1 L0 t: ?5 A' o/ |( X1 B
" v0 V# L4 g+ }
# t7 r* t% F9 {4 n# @# s  B/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
6 M3 W" ?, W- i0 [- Y  c: [

7 j1 d; C% B7 y9 S#pragma once
#pragma warning( disable: 4355 )
. y- g3 o9 t# D7 I0 p' q6 `4 v
% t. u5 B8 M& F2 _
#include "UPnPImpl.h"
#include <upnp.h>
1 K+ x2 G8 ]5 d( c! C: H#include <iphlpapi.h>
8 y- H5 w7 `2 o1 e9 o+ A#include <comdef.h>
) G& n+ C. @; T' O8 w#include <winsvc.h>


7 B. X+ \. J8 H+ }#include <vector>
#include <exception>
" o* _# e- ]1 \6 U" l* s( D$ h- U#include <functional>

. f5 g( o& H( C# z4 R


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
* e2 ^# h4 r( \+ H5 z- f1 C, ftypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
7 q- B3 G0 V( wtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
8 O; @/ q3 |2 n3 `6 }2 y4 h6 h* qtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
; V" O$ Z! h; O3 m; h6 [8 }typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
2 o. ^. e4 N5 {# Otypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

1 l  `1 [' w7 r" V' p
% ^% v6 C- B- U+ U0 _/ e5 ]; v" B1 ctypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


* f- d+ ^; G. b9 S, I! Etypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
% q3 J9 H4 G3 X  q- Z" F);


typedef DWORD (WINAPI* TGetIfEntry) (
& e# N/ _3 J4 L2 U7 s  PMIB_IFROW pIfRow
);


, V) ?* O: Z: CCString translateUPnPResult(HRESULT hr);
  R7 o, [3 ?: H% ^3 z' I7 vHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
/ q9 M6 L# t& Z: ]! c! G        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
" k( i& ?6 |/ F// Construction
public:
        virtual ~CUPnPImplWinServ();
0 R: J$ s0 a" T        CUPnPImplWinServ();
( o; j0 Q8 d2 C' V  q
( u4 P% {$ r7 P& ?
$ V5 d8 ^) F, Y* a; q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
) r6 T; @6 {* m* ?        virtual void        StopAsyncFind();
& t* y8 E/ q5 g, O0 e        virtual void        DeletePorts();
  l$ u. g3 x6 H$ U        virtual bool        IsReady();
1 |* i3 Q3 Z& X( B9 _1 X        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


9 V- }/ ?' X; M% Y        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
1 a2 _/ u8 D$ l        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

7 ?9 p9 a+ K. x6 M; w1 F
& x: Z( }$ m5 U$ m3 J0 I: V% g& O1 dprotected:
+ u# b& X) k( L6 \, K5 a        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
# g" Q! g" g+ q; ?# j: Y        void        RemoveDevice(CComBSTR bsUDN);
6 \. V4 R* W9 Q        bool        OnSearchComplete();
        void        Init();
4 ~! g) h$ F4 }0 m- ]& t

# }8 k' b' i: v% s$ a        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
  r1 v4 O1 [7 q1 o4 k                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
$ E4 `* E, P) Q0 Y8 s) b7 ]% p3 I                        m_bAsyncFindRunning = false;
' V. E4 ^/ @$ Z5 g. E                }
4 \) z5 I( n5 q* g6 G  c7 q                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
+ G( u6 Z/ O. w; F' ?! ~7 h                return m_bAsyncFindRunning;
/ z1 t$ U. e# z8 F( W4 Z, W        }
! [* Y; n; e5 c/ ^1 Q

2 _8 S+ I1 [$ M7 p- C( ^* ]        TRISTATE                        m_bUPnPDeviceConnected;

3 I5 ]+ b+ f" m: g$ H: s- i- }3 j: r
// Implementation
! e% k& L" o3 g3 I( A+ p8 `% [        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 K' U: {2 [% R9 e( U1 L) ^  x# X        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
* |! S4 G3 n4 `4 W0 }

3 T# _; U; y  ~2 F1 R4 B        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
5 n8 [( @) V$ u9 U$ M" H

        static FinderPointer CreateFinderInstance();
6 y3 }; O9 W9 a8 y; g( e. ?        struct FindDevice : std::unary_function< DevicePointer, bool >
1 C' B! J; n2 R" Q( W        {
1 Q) ~  n; M2 D9 Y2 x                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
  B) L) h+ ?# y( [/ M: k                result_type operator()(argument_type device) const
) f. T5 Y, u1 m; X                {
                        CComBSTR deviceName;
4 j. i( j9 Q2 Y5 K0 Y                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

! E7 g2 j7 e9 e# d# a" q
5 {( d; ~: E4 F1 Z% f/ u# u                        if ( FAILED( hr ) )
" r; n; O1 h& O4 A- \) Z0 X                                return UPnPMessage( hr ), false;

% T6 |* a; F, Y5 B/ w
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" o& w0 [( `6 c+ ~; k                }
2 q0 j2 @$ g8 y                CComBSTR m_udn;
  V  a- j4 Y3 |* T# B        };
# w5 c/ o1 d0 j' v6 ]' H! c* V       
' t+ r; b8 z+ i9 Z0 x& u        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
$ T! i/ J! H$ L$ n" m5 i  R+ g1 x        void        StartPortMapping();
0 v0 T3 [* c; N0 o        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
9 h" Z: F& T  f        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* R) E' h1 ]$ q3 A1 d. Y                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
4 Z4 y0 m/ ?& I* H  }

9 {2 |( F% |/ y; v1 K6 E        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
# a; D5 S. e$ y; x& m: n        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
+ O2 u6 q/ y* T& E" U& Q3 T/ v        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
3 p6 T1 L9 o/ B5 N5 [8 v( ^- a+ l        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
- x! O! |6 p4 q, }

5 s! m: m; W. C% {2 J// Private members
private:
9 f- C. w! l3 B. q6 O+ @' A        DWORD        m_tLastEvent;        // When the last event was received?
( V& t7 q% O1 s6 k, a" i( d        std::vector< DevicePointer >  m_pDevices;
  @8 E& m- }! W% a        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
$ z# `( s/ Y3 a, q# A

        LONG        m_nAsyncFindHandle;
+ w  h* t2 j/ b! v0 L7 M        bool        m_bCOM;
        bool        m_bPortIsFree;
3 T  S* j- o0 S0 I; x- Z5 x- }  y        CString m_sLocalIP;
3 E& C: k, w  i6 x% w        CString m_sExternalIP;
6 a/ Y6 h3 S# Q7 K" H        bool        m_bADSL;                // Is the device ADSL?
& r, `( j  S: H. l; b5 Q        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
) q$ E# K- e7 t( Z) R3 W: ~  v% u        bool        m_bInited;
        bool        m_bAsyncFindRunning;
2 [$ g( J9 \' M        HMODULE m_hADVAPI32_DLL;
# u" J/ T5 V2 q8 C8 ]5 P6 l        HMODULE        m_hIPHLPAPI_DLL;
. u, D  Z. L# ^" @# n# R2 s! E        bool        m_bSecondTry;
* o5 E+ d% B& A* j4 T        bool        m_bServiceStartedByEmule;
% `$ E- Y8 h( w, u1 V9 V9 A2 P+ J        bool        m_bDisableWANIPSetup;
9 }# c! s6 O9 S% S/ t2 j" K: A        bool        m_bDisableWANPPPSetup;
( d7 w4 G- U4 g+ N9 r) S

4 j! M+ P1 I) C' j6 g5 l: p9 v# ~};
) [* J9 \. x8 F9 m0 h' U
" A, O- t' Z# V
# q3 ?, E& p- g& e4 O1 C// DeviceFinder Callback
3 r8 {% ?- A- ^8 {% F6 Iclass CDeviceFinderCallback
' H  {5 B, G* q2 B" b6 A9 J        : public IUPnPDeviceFinderCallback
4 l7 k7 V' g6 g( Z; y4 a$ t/ y{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
  O/ K: H' C1 P7 q( g        { m_lRefCount = 0; }
$ S: z& y! {% C: i2 v' b

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
; {9 q( v# f+ m# v0 e) [) g   STDMETHODIMP_(ULONG) AddRef();
: Z9 }1 X# v2 l3 m   STDMETHODIMP_(ULONG) Release();

4 m7 W4 U9 Z4 M# C2 y. w
8 I5 {/ u) Q& d// implementation
1 e# \8 I4 {6 ]/ aprivate:
8 C! `- m5 K+ T$ W7 ~        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
* F: F7 Y$ w3 c9 A: W4 y6 Y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
  D4 x6 Z5 J# K" Y8 ^! L

private:
* @5 a2 \) K( l4 h% a: x        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 M1 q- t  d: X3 ~7 D/ p};
" p  w3 H9 r" B% r) m, x# `* S

/ w* J; U; g. a# m4 `# ?/ o7 ]// Service Callback
2 C1 n) Q" `) `class CServiceCallback
% \/ R" k& p' s1 [* r9 A" q: W# U3 E        : public IUPnPServiceCallback
7 r! e& @# |5 k) X{
public:
) C% W# I' T+ h8 j0 R        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- O7 r: x: ^) E( q   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
  Z% d; _* M+ v) r3 G
: V9 n2 P+ c8 j2 y* ^& n
// implementation
private:
4 }2 C0 f7 K/ W9 X        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
8 K* r# h1 E) Z  i8 C

' o% ~# U/ C% `! J" dprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
2 r6 s$ M! \6 o+ W1 n# ?( ^
" z2 k9 b6 D" x' q$ v2 v) z
' {" e6 n- _$ E6 `( W' T- O/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
. z0 }% x7 m3 l$ RCUPnPImpl::StopAsyncFind停止设备查找.
0 A0 t% q3 ]. d$ b" @) b4 xCUPnPImpl::DeletePorts删除端口映射.