UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   / F5 g: P9 ^2 O3 k+ Z
3. 
   
4. #pragma   once
   $ b8 z* P$ n; j- I( c
5. 
   ' V3 N8 P1 C& m8 i1 d* R/ u
6. typedef   unsigned   long   ulong;
   - J" n3 a9 y  |: D2 `
7. 
   
8. class   MyUPnP
   ) p# L) X$ m: k* _% @
9. {
   . E, b6 P7 H! Q, }
10. public:
    
11. typedef   enum{
    , S6 S! P& f, H3 {( E& f# ]9 i' w
12. UNAT_OK, //   Successfull
    
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
    5 M: ^: v4 S, w5 W$ k) E
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;
    & {+ P2 v/ `9 G( s( K8 f
18. 
    & F/ z: t  c) b4 d) C, ^
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    0 O* p2 T( T# P! }! U! j* ?# v
21. UNAT_UDP //   UDP   Protocol
    3 E+ D6 |7 Y0 y9 c! r7 }( ~
22. }   UPNPNAT_PROTOCOL;
    + O, i7 a  n- K  g7 t7 ]% Z
23. 
    5 q6 I2 [. z$ z& T
24. typedef   struct{
    8 t! o4 w1 C9 e0 K
25. WORD   internalPort; //   Port   mapping   internal   port
    ( T( {/ [; m& A- m5 v' p- K# B7 A7 @
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    . @# ~% @& D* c) V+ V) j
28. CString   description; //   Port   mapping   description
    9 b% Z- v6 I0 o. C7 L* g0 u6 i
29. }   UPNPNAT_MAPPING;
    / Y- u* i+ w3 T
30. 
    
31. MyUPnP();
    $ O$ {; V- J  g) a8 ^
32. ~MyUPnP();
    * v& e) |' e- `5 o3 A
33. 
    6 V; k* o4 E; w5 \, N
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    : H6 G: o3 u9 D; D0 [
37. 
    
38. CString GetLastError();
    $ }3 O8 A8 ]- n3 B& k% E2 l, i
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ) o) ]' a2 d$ r% P8 r0 Q
41. bool IsLANIP(WORD   nIP);
    
42. 
    # l$ [  A, i0 Q! @6 D1 h
43. protected:
    
44. void InitLocalIP();
    4 Q% d7 E, ~$ p- W- B
45. void SetLastError(CString   error);
    
46. 
    8 k9 d. I1 F4 D
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    1 b) A2 p+ t: c* o+ s0 J* o; h
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ; X7 H( a7 L( ?1 G4 `
52. 
    
53. bool Search(int   version=1);
    , \. i0 V5 {. A5 ]
54. bool GetDescription();
    1 o' y! Z! D' ?- `
55. CString GetProperty(const   CString&   name,   CString&   response);
    $ z, _- a2 l9 k( b  D$ d& f
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    3 I' U7 M- ~, e" s/ A
57. 
    9 `7 J0 f9 n: y! g* W  t) F
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    - E  ]( C% o( B: I
60. CString m_devicename;
    
61. CString m_name;
    5 _& B: \' ~+ t+ g! f
62. CString m_description;
    
63. CString m_baseurl;
    8 {" _/ t$ M1 u! u; m, _1 n
64. CString m_controlurl;
    
65. CString m_friendlyname;
    : a) g$ |  L- s. }/ u" }
66. CString m_modelname;
    
67. int m_version;
    - [- l& C& d2 e1 m6 v, i
68. 
    
69. private:
    ; a; S5 z# t. F0 ]# f0 W. c
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    + C2 I- c; K0 Y, B7 q
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    8 [/ F2 G/ o$ @: `
75. 
    
76. bool isSearched;
    
77. };
    + W4 F" M" r0 W9 t; G/ ^+ W! u
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 ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   / ]2 c; ]- Y5 q: p" h. |$ j6 O
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ) w$ u) m$ J- f) a- E* ]/ l% q9 S
11. 
    , g; |- Q, u# P3 H/ U3 [
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    * z% O! a  L: S$ b. _* L; ?
13. static   const   int UPNPPORT   =   1900;
    * C' i6 b3 M& G' F0 n7 V- d+ b
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    & ?; ~3 K- ~" D1 K( B
16. const   CString   getString(int   i)
    4 Z: }6 j# e8 R4 P5 p2 V
17. {
    
18. CString   s;
    
19. 
    0 o" Y8 i% {4 }: o
20. s.Format(_T( "%d "),   i);
    - X1 z+ N$ @8 m, e8 c2 M
21. 
    3 ?- P2 e1 E* o# |2 X' V3 c  b
22. return   s;
    
23. }
    
24. 
    % E6 N0 |7 U5 M4 f4 w
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    8 u% U/ x& s3 R/ E4 ]0 p
26. {
    ! o* j, ?; _/ n! w: G$ L) K& F
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . E: P( P( k* o+ ?  u6 e; y8 A% F
31. {
    . d0 S4 a* N. ~
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ' {, b$ L  B, E2 P& f* \8 Q5 [
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    6 ?6 i- s3 r- \- w4 V! o( C7 d
37. char   buffer[10240];
    
38. 
    7 @# _# G5 ?" [# @& m, |& O$ y
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ( j8 b" }; A+ E) T9 P7 f. H
44. struct   sockaddr_in   sockaddr;
    - ~- }. u5 W' [& i1 b' R* K
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    + U3 _/ z1 I8 f8 ?6 D. X9 _9 {
46. sockaddr.sin_family   =   AF_INET;
    4 \" u) _2 u0 `! z
47. sockaddr.sin_port   =   htons(port);
    - a. J; m) V1 E  M2 |
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! A: m' ~$ @# y2 f/ P, q
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    5 v& u1 H( |, v  s# `3 g/ C# Q
50. u_long   lv   =   1;
    - ^7 I& x( T; I; p5 a1 W, |. \! e2 N
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + r8 f! C9 E4 T  s. ^
53. Sleep(20);
    ( D. f2 B8 q2 v6 }6 c# O
54. int   n   =   send(s,   buffer,   length,   0);
    1 N0 Q# _! J+ N" S9 t
55. Sleep(100);
    0 F, ^8 C% v/ O2 ?$ F! ^
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    0 g; s9 m0 `% L! _, m
59. if   (!rlen)   return   false;
    5 O9 u' K$ n' Q& G" N
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    2 h2 P; U4 `! j' G: [
63. return   true;
    
64. }
    + j* z4 u5 M4 H% j
65. 
    7 k) A) X8 r7 n
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    2 P9 j; m$ m5 f% I6 p3 n) R) L! b
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    , f& L) Q* @: I" L. u
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    2 z1 f- E0 M  h
74. struct   sockaddr_in   sockaddr;
      R. _. a& S9 b6 c
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    ' \/ r; W2 u3 ]! o' ^2 L, t$ ^- e
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    5 ~  X8 O+ V/ ]0 M" R, _! ]& ?- O& l
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % |! y' x% x% t- W4 d/ b4 h
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    ; A5 O& o' I+ m) t# B
90. result.Delete(0,   pos);
    ( D% w; J& u: E1 v5 ]3 g1 U& t
91. 
    ( H8 m2 Y" u/ z6 |, y8 F
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ' }- V; m8 o% M$ M; V( V8 n# l
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     4 c, c# p4 e% E, q7 ^
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     % ~3 @7 ?( a9 y6 J) z3 A8 M
107. 
     0 E: T5 x0 |  t. P/ n3 F
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     , P$ u5 c7 c7 c' J1 t* b
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     : Z! u" N7 k1 q' t8 z7 I* F0 v" {1 b
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     ( a. h% G2 ]* Y5 X
119. }
     7 G8 J  C  D( s6 o# b% K9 `) {
120. 
     ! X% m! R! X5 W0 I4 u0 S; n  p9 m
121. MyUPnP::~MyUPnP()
     + \2 w: x# ]# _2 d: I# t  G8 k
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
       L8 {' V4 [2 F
125. while(pos){
     ( z" c- O* T' E' I, W
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     . y* Y* \5 b1 ?
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     : h/ N6 E. U; ?# K  B# u- [8 O) o4 {, e
131. }
     + i& {, s- r: Y% g
132. 
     4 |3 ~5 @& C: G3 ~' C( B
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     4 G2 d2 g8 {+ M% X5 o; ]& e. e
135. {
     
136. if(version <=0)version   =   1;
     ' p. F  Y5 q5 `- [
137. m_version   =   version;
     . ]7 b1 C: q3 c6 {2 {
138. 
     . I. e3 o, G- Q$ ]7 ]$ N
139. #define   NUMBEROFDEVICES 2
     . n, A* W8 q" H! Q: g4 z! ~, F
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ' |6 O9 V2 u+ X7 u
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     6 S2 |1 C7 }" U$ E# c% v) {" K
147. u_long   lv   =   1;
     + N: }+ @! b) O" M: I# ^) v7 p0 o
148. ioctlsocket(s,   FIONBIO,   &lv);
       w8 S/ D6 g8 U, m( Y2 x% s
149. 
     1 W: r3 `1 p0 h0 [$ C
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     , v/ A( A. E1 A7 _9 [! R
152. if   (!(i%100))   {
     5 j/ f. ?" H1 F9 I
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);
     + Q% x( {/ \1 S  h# m# J
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);
     ( u1 r( v# \: i: k
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     7 @2 ]. x( y1 i$ J& h
164. char   buffer[10240];
     - S: t8 p  c) L
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     1 n6 K" d: T# r- u4 A
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
       Z" V; [/ m7 |( _  j
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     # |4 U) Y: n" t! G# j4 C9 [' k
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     & ]1 A9 B7 p4 }+ w: F0 j
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     3 D. O# P9 L8 Z1 Q( P& o* t
180. name.MakeUpper();
     + r2 A" V9 v. D( i! y
181. if   (name   ==   _T( "LOCATION: "))   {
     ' P5 U: Z# T  Z7 c
182. line.Delete(0,   9);
     4 K% `' u/ w& @/ B; A
183. m_description   =   line;
     
184. m_description.Trim();
     1 ~4 G! h# O; g' ]$ F
185. return   GetDescription();
     ! z6 Q  X0 y/ j! R( b/ j0 l
186. }
     % Z. J/ m1 b. ]& U5 d. ]) N
187. }
     
188. }
     
189. }
     
190. }
     & b3 B. _+ Q5 ~( m
191. closesocket(s);
     
192. 
     % B9 @7 X/ S  r5 H3 H
193. return   false;
     
194. }
     ' g! [+ z  p& R3 K2 B4 n5 U

复制代码

天云上人

以下有关upnp的接口来自emule，
% X2 R8 n$ V5 S2 `

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
* f/ }# c3 ^. R3 m

& m0 d7 O) |# j! }#pragma once
#include <exception>
5 E+ V4 m) i6 q; t% R
8 J3 W3 P3 W  i; z& `" I
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
. J' i2 ?+ E4 A/ `

enum UPNP_IMPLEMENTATION{
9 h2 V' Z: i6 @  ^2 j3 W  j0 C7 B        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
& [- @8 s8 E6 P) d& z        UPNP_IMPL_NONE /*last*/
8 V; w/ S1 m" `};
, n' D+ K- |7 s1 J( J; f



% G! \' U8 q* P  Lclass CUPnPImpl
9 Z3 U. [! c; o: [{
public:
        CUPnPImpl();
- |' c3 j/ d: V# n0 k        virtual ~CUPnPImpl();
2 c& M, i. e& m2 J) H        struct UPnPError : std::exception {};
+ T: O9 P0 K$ r0 b8 v5 S8 Y/ z        enum {
% r# z" |7 T; g3 _6 L5 ~* u6 ~& N                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
0 b) U/ o  Z7 M* v- b0 Z4 C8 C# I

9 t6 V  `( v& y' v$ k        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
- n" P9 L1 A$ F7 O" i) B        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
- t  b" `# H2 B& {& h- M& U* b        virtual int                GetImplementationID() = 0;
# d0 @0 U" v" J$ U% g: E1 U# y& V       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


% t2 p/ G: }* h: ~( i        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
+ E, V2 Q; o$ m5 U; P        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

3 G" Y. {4 x; _
// Implementation
' O, @: f6 f! E$ t7 N  s; f9 Lprotected:
+ s' Z6 X4 _, T; O4 B        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
& P7 S) W1 r9 S) e        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};

- D5 ~% @$ F+ |. {+ w
0 h  o, C5 r+ a2 P/ ^! d0 W) v( d// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
/ L) m! p$ B3 B! j% `* A8 Z5 O- ^2 W{
5 ?# K: v1 v3 ?4 opublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
) e! q& j7 E6 x! |( s6 p8 |};
; V: i! ?( R. V% Y: P! B

! M% E+ p0 ?) D' |1 c6 r/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
( P7 X, U1 F2 x: t: q. m

#pragma once
; o& r' \& k, |& g# ?#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
: J; Y" E$ p& N' A#include <upnp.h>
3 n5 ]4 H5 v! g$ n4 t5 g$ p& s#include <iphlpapi.h>
#include <comdef.h>
  f% `" ]. F" p. j) I5 m#include <winsvc.h>
9 g- O" ^. B% ~; _
2 ^* J& s  @- s! _, u
#include <vector>
. l* o) M1 F+ D- f1 ~& d, l#include <exception>
#include <functional>

( u. x; m8 G9 x; L3 \
# U  g+ P5 b/ ?0 v. Y

7 R1 X3 @# O6 gtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
9 k0 X. x7 N7 `+ otypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
3 @0 G& Q/ T! J" s2 Utypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
/ h: Y. v* X# _typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
8 y, C( X' s7 s6 }( W/ ttypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

( t7 B8 N0 q* R/ o0 D; }  V
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
( I) G( k" \9 C' z2 [  PDWORD pdwBestIfIndex
);

  s! D" D. C9 }/ |
typedef DWORD (WINAPI* TGetIpAddrTable) (
9 Y2 }9 x0 g# w. ]. T1 |  PMIB_IPADDRTABLE pIpAddrTable,
" O4 u. @' ~/ ?* }$ a  PULONG pdwSize,
+ R* f/ h: X3 a' l, W2 F6 w; f  BOOL bOrder
+ J2 o: k5 H2 s' E  G);
5 H5 w. E0 L: I1 q# a4 Z

% H# p- c( Q: [' K, otypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
* }5 U" k( A  G);


6 d3 e9 |, F2 }" ~CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
( y- i3 l: a* n' d

. X5 u. X) A+ ~5 D8 Fclass CUPnPImplWinServ: public CUPnPImpl
8 ?/ e3 ~9 N1 g2 V! N0 J, Z3 l{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
& p3 m, V+ t; e( {

' Z+ w9 A' x. e& Z) J8 ^- l! S        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
% f" T% A9 p6 W/ m( F! W        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
, A$ m  ]7 P2 K: L        virtual bool        CheckAndRefresh()                                                                                { return false; };

6 R3 @0 v. s/ O1 q. @7 W, Y
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
: p9 d( b4 O7 u4 \  i( l        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
' l! {$ b& A+ z) r: ~        bool        OnSearchComplete();
        void        Init();
2 X& t4 f. r0 G' Y  n- C' s! f5 K

        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
- E2 U  \- P# v. s! I8 ]' \                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
# T8 h1 e8 H* Q$ R$ X2 }8 X6 c                        m_bAsyncFindRunning = false;
                }
                MSG msg;
) F* O( p1 z! M% H0 G$ |5 {                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
: J/ k$ w# B6 E. U8 j; r3 G                {
1 z+ I0 P. @* N6 G1 e/ ^7 A                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
0 O, i+ V- e  Q: ~) E" F. e2 V                }
                return m_bAsyncFindRunning;
        }


) J# g4 `5 ]; o: A5 \+ q        TRISTATE                        m_bUPnPDeviceConnected;

+ H, F/ J& @1 r* `6 h
! W3 Z8 ~6 ~' D  @// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
* \' F" b; w, D( f1 G        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);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


9 L: B: F& ]6 Z  E& C6 E1 ^6 g8 N( A        static FinderPointer CreateFinderInstance();
* m( U, I# [% i' L9 S        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
! K) A* g9 W! a$ ?( W; s( r                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

5 ?5 N$ Q5 ~8 u  Q" g
3 r( U  L6 \* U) P" `/ Q. u                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

( v! c" x+ I" r, g4 \( B
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
. t, u/ D1 `/ D- Y        };
0 d$ ~$ c# b' M8 r# X7 I- v       
! b; `/ @) g  Y& g        void        ProcessAsyncFind(CComBSTR bsSearchType);
; s7 J" W; G' E  v" Q        HRESULT        GetDeviceServices(DevicePointer pDevice);
  D+ f2 P4 ]) C7 j2 C' N7 L3 W1 Y! d( }        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
8 h" J; R+ C5 c" H4 I        void        DeleteExistingPortMappings(ServicePointer pService);
+ e( y. O. q, t+ E7 C9 {. K7 W  ?4 c        void        CreatePortMappings(ServicePointer pService);
7 K5 ^8 p8 @' X. j9 D. [: ^# B" E9 o        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
# Z9 g; |: M/ S& M) W        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
. I1 ~  N% a2 C* j        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 B* A9 S( a9 X8 p        CString        GetLocalRoutableIP(ServicePointer pService);

  S6 L6 h4 W" r4 r3 W
// Private members
private:
5 z3 @. y+ s2 L1 _: T% r        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
7 Z% h7 j; o: Y  n3 j        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

- S0 f' Y/ }- O& n( j- F) S# U
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
# c' d6 |% d5 k' H* k        bool        m_bPortIsFree;
/ t: p* q: y. Q& S1 R        CString m_sLocalIP;
        CString m_sExternalIP;
$ b0 `* q0 ~$ }& @. e8 c. A        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
1 [& k7 m4 F1 ?        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
# s7 u* N8 p4 E7 r. i0 T4 m        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
- {* H0 D) W; C4 ^  `- z) q# C' @# W( r        bool        m_bDisableWANIPSetup;
' N" ~8 e" [9 y" G* B1 q        bool        m_bDisableWANPPPSetup;
7 G( h. E9 [, d/ j0 }
' |6 t9 H: y5 a; d
};


// DeviceFinder Callback
class CDeviceFinderCallback
' P4 J3 k- ~6 \        : public IUPnPDeviceFinderCallback
{
$ _$ O8 s; o3 `8 Y1 L5 Bpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* n# \1 f2 H9 |9 J/ w' O7 \                : m_instance( instance )
        { m_lRefCount = 0; }
& J! H6 p7 n& [

6 t. u8 B: i9 |- R1 d   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
! k7 c8 v. G/ t, X7 Q9 v   STDMETHODIMP_(ULONG) Release();

8 V0 q3 ?/ J: V4 y2 C
// implementation
2 K+ ?8 i& o% Y/ H$ p' dprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
5 {* p" M' V/ G; H2 Y/ P& l        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
6 y( ~! p# i' b        HRESULT __stdcall SearchComplete(LONG nFindData);
* N5 G7 J8 y3 v/ ^/ E# V; Y& q  A& f# n
# x! t6 L% t7 R! I8 q
! u' h# l0 `5 [! A# |- Tprivate:
% M) T- P8 y5 h7 _6 P; P& D! E5 S        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
! `8 [' I) B& d5 D, y" T};
, v. T4 p7 D7 T

; ^* n( ~/ \5 X% [5 `$ J( T// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
1 U+ ~7 h1 s2 d* ~8 Spublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
0 M0 c. t* W, q1 H  L0 T   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
, |7 ~2 Q$ `* ^   STDMETHODIMP_(ULONG) AddRef();
3 g# E( ?& c/ D1 Q& u/ X; h( C/ ^. g   STDMETHODIMP_(ULONG) Release();
2 i+ ?! F$ p' m. f. o1 H$ i

# B( I: L# {' i7 d/ f0 R; D+ s// implementation
# k7 w$ r# ~9 r" M1 s# |private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
0 t/ F: w0 ?! F        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
1 S7 i/ W  K% i7 H: W- Y' E+ m# {
) p: i& ]9 ]+ c1 T; H% T8 u
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
8 N' D) J6 ^  f* {. i3 _% r

& t8 C0 V4 T- N2 _/ x4 u/////////////////////////////////////////////////

/ ~" f- c; f/ @$ v
使用时只需要使用抽象类的接口。
% B1 Q+ v4 U/ A0 O3 ^CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
, S; A# T8 y% W# A4 a2 G* F9 a3 K7 X: zCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.