UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   . @$ }  H% ?) D; h' I  R7 ^
3. 
   1 U, G: u# A" _3 ]: S3 v
4. #pragma   once
   
5. 
   % \0 C* I, d+ e, n
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   % T; v; j2 n7 M& @+ `
9. {
   2 X- ?" V1 H0 _% D
10. public:
    % R/ Q* R; o1 e8 R* a6 I" e
11. typedef   enum{
    ! y9 T! G$ M) v5 g: |3 U8 d
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    % f$ x6 a# i% B. i
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    8 W( T: i4 h0 t" B. u( j0 ]. b
18. 
    ! H" `" N5 Z) g% @8 G
19. typedef   enum{
    2 {% `& l) E7 s, L/ z
20. UNAT_TCP, //   TCP   Protocol
    7 N7 P  `/ [8 @7 R4 C# X
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    * X, @* W7 e& O. s) G
23. 
    0 ]+ ^7 u$ M* M  t3 e( A
24. typedef   struct{
    * y1 u; E' i+ {9 W. O
25. WORD   internalPort; //   Port   mapping   internal   port
    & a1 H# M- S6 f7 p+ G3 ]- ~
26. WORD   externalPort; //   Port   mapping   external   port
    2 s1 h1 j( b8 ^' i+ k5 C
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    : [1 `: R0 q- P  l
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    8 ^2 L- h! N- O; |# |  w) C% b3 |- {& Q
31. MyUPnP();
    
32. ~MyUPnP();
    / U% ?3 u' [& t4 J5 _* D2 [
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 [. w" r$ O% d  t" ?% }9 h
36. void   clearNATPortMapping();
    5 L! M  }- c3 l  m8 ?
37. 
    . I% A6 Y& i$ X( v  G  u4 w
38. CString GetLastError();
    8 x8 U' i4 ?; j+ c, H
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    / A1 N) k3 H# ^/ [- m- P& x5 E
42. 
    ( T# r% M' ~( }7 k& T8 `9 {* m. ~
43. protected:
    2 ]& f2 b1 Y/ G! f: F* W  e' A
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    9 g5 g2 X% S' g# X/ u
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    2 Y% B, i: |) Q% j
52. 
    
53. bool Search(int   version=1);
    ; k# G- k* D2 H2 l
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ' V: b: u& K  b* o/ }
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    + _1 p( A; B  l6 a$ j5 y
59. bool InternalSearch(int   version);
    ) P. A* w7 L9 Z6 O: F, U3 T* }* u% B
60. CString m_devicename;
    + i3 |! d/ X! T7 [  ~
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    9 t6 D5 F5 E0 y% X, Q' ^
64. CString m_controlurl;
    
65. CString m_friendlyname;
    - E1 }, [4 j' j5 c
66. CString m_modelname;
    ; H1 a; r! U# M5 [7 Z/ g& S
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    / E5 z0 F% i  J0 t6 Y' b' [
77. };
    2 |: N, v4 w2 w1 C5 ]7 A% x
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   : l! z' [' Y: W- T3 Y
3. 
   
4. #include   "upnp.h "
   
5. 
   ! I8 E3 |1 D5 y
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   1 }/ t) h" A; B: Y0 `
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ' {! i) H: T5 M- ~
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ; O5 [: V! U+ m+ y3 n* c
11. 
    - y& {% n5 b+ |$ K6 }+ _2 Y  k8 D
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    - P* @9 O  _+ R, R4 O
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    * E* R) I! L8 f* {6 e
15. 
    
16. const   CString   getString(int   i)
    ( {9 I! Q# b0 s( a2 X
17. {
    " H) h5 M, P- j! l! H
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    * x; B: B* K8 \; k' e/ T
21. 
    
22. return   s;
    
23. }
    & M, e: D! B: y9 P2 Q- z. [
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    4 S# e1 v" V6 X% t4 o* m! G; \
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    % {$ p" ^: H0 [+ X1 c* L! E
29. 
    " U) X; Y/ l: s3 b* k; h
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    4 j8 G" m+ E: N( a8 f
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    * e! q+ |; o2 o
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    8 f0 |+ w, l& _) [( Z
36. {
    : E5 q* L: V, c! r6 ?0 {8 {9 r
37. char   buffer[10240];
    ' h; j) d, R) u  p- M+ ?1 J$ F
38. 
    * K) R( x$ g% ]0 D
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 ]8 q. a4 }3 J$ q0 {
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    0 S: ]4 j+ A9 C: s
47. sockaddr.sin_port   =   htons(port);
    2 W- T. j) V! R' `/ e0 g, U" K- |8 C
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    : J, w( M5 z, O7 W
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    6 k% B# f* b* X) P
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    - {5 p+ l. H- q  C
53. Sleep(20);
    % w2 n4 g; Z0 G5 O( T
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    9 A' ?1 v  w/ N0 f; L! b
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    : R9 J) O  L; t5 c
59. if   (!rlen)   return   false;
    
60. 
    - S- B& F* F. K" x
61. response   =   CString(CStringA(buffer,   rlen));
    ' q& b' O1 J- e3 k, l+ X
62. 
    
63. return   true;
    # @% M/ Z+ `' N2 K% o8 \
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    4 p8 F* J% w7 F1 g' X
68. char   buffer[10240];
    
69. 
    + g8 _* ^, |7 a* _/ L" _
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ' T7 k% S# R1 Q  f
73. 
    
74. struct   sockaddr_in   sockaddr;
    : b- G% T0 k: \
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    . D/ R3 Z) P9 q5 Y  |% O
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    + C# P! `- E4 b
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # g* o* }; T6 P
81. }
    
82. 
    # \' m; |+ R* h: w- Z( k. T
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    # ^6 Q8 F6 }7 ^
89. result   =   response;
    9 h$ T4 K0 s0 A/ H  z& z
90. result.Delete(0,   pos);
    6 O7 J* V8 G5 j/ m' t5 u
91. 
    * o  v" f& `4 ~% m3 p2 Z- D
92. pos   =   0;
    ' c& o: Y  v7 e: m8 M
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    % @8 U0 W3 s. p* Y7 i* ]8 q' t
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    . {7 }3 P+ R  X
96. return   true;
    ) ]1 k. E" j  j' V8 f- H) M
97. }
    
98. 
    + ]8 v; M- f# }  X1 F
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    6 K1 W5 k+ q. N4 E+ G: g
100. {
     $ ]1 o/ t# f8 h9 ~
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     0 r3 l8 I+ o8 \" {! ?5 h6 M  u! y
103. CString   property;
     
104. 
     " {$ r5 H1 `& {# e8 K6 M/ X/ ?# c2 P
105. int   posStart   =   all.Find(startTag);
     4 u3 c# p5 z% l- E
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ( H' L) I0 q5 D, u/ r9 \
109. if   (posStart> =posEnd)   return   CString();
     $ X5 g. ~6 }! I1 n9 |* U3 ~/ Q
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     : @- A  a, S8 n
114. MyUPnP::MyUPnP()
     ) {2 R. v( B8 i; P
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     # K+ C1 F2 I$ q+ v( G
118. isSearched   =   false;
     : R* s' M  @( k( A
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     7 [$ U( ~. Q2 Y' T" Q" T! |
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     ! |2 C/ M- h  }2 D9 c
128. }
     : Q) e1 z/ f5 o% o
129. 
     : k$ s% q+ W6 Z3 k
130. m_Mappings.RemoveAll();
     
131. }
     . |; G2 |% s* M: ]: o
132. 
     
133. 
     ( h. K" S+ M8 t. u3 }/ {
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     1 l- {6 s7 @& j7 P* Z& x
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     2 a5 u4 ~/ Q& V( \- `
138. 
     
139. #define   NUMBEROFDEVICES 2
     ( I8 H$ z. f9 Q
140. CString   devices[][2]   =   {
     : J! T0 k: I; B2 h2 B' V4 f! S
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     , f. x+ |4 O2 I5 P" _
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     6 a; t1 F+ @% N0 J
144. };
     & }  r2 J4 J5 p! ^6 G$ ^; O2 N4 _) r+ F
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     " @+ d! T  N% ?& j6 r
147. u_long   lv   =   1;
     ' O4 v6 C* Q9 h$ H
148. ioctlsocket(s,   FIONBIO,   &lv);
     , Z# B4 d% O4 S% V$ ^
149. 
     8 K/ M& {2 i, p# J1 k& ?/ d3 z
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     4 g$ [  l! j, Z" \% s
152. if   (!(i%100))   {
     - G( U' ]4 a, t7 ?; s
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , C  ^9 ^# W# u
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 "),
     ' x. T) ^4 P1 i# H3 d' A
157. 6,   m_name);
     ' G3 _' K( l' |* p4 l; B; l) l
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     6 \) i* J# z4 @
159. }
     % m; i/ ^- Q2 [4 b
160. }
     $ n& D1 U# ]) j0 z- I! u+ R+ q
161. 
     
162. Sleep(10);
     2 x. N, ]; k" J9 @
163. 
     7 i6 J; ^# g1 j
164. char   buffer[10240];
     / d2 J' p0 {  D9 P7 m$ C+ V6 G2 k8 O  M
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     8 x" f' e- H# f% W- z
166. if   (rlen   <=   0)   continue;
     1 i7 p' W" i2 b. o: P" H) O$ A
167. closesocket(s);
     2 ]) [, W$ Y% V/ W3 r7 ?: ?
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     $ G: v) d) h/ A6 x: ]2 m! u5 Y) l
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     7 o" i# Q% x: I: R% d# V
176. for   (int   pos   =   0;;)   {
     + \- q; K% |  A+ ^
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     $ e# W3 p+ X# E0 g- R
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
       A) O: J" x( I( y( x; z2 d9 m3 g
183. m_description   =   line;
     7 I5 @2 I1 L6 C/ H  r
184. m_description.Trim();
     
185. return   GetDescription();
     9 m, b0 R  Y& f6 G
186. }
     
187. }
     
188. }
     - h% L" r* _; ?& H$ Y$ ~4 z
189. }
     & m# o9 E( l; v
190. }
     & i6 J6 p% J# p- @: }3 n6 l
191. closesocket(s);
     / W1 `( O0 H  ?
192. 
     % C% |3 |& ~1 a* H. j4 C: U- C
193. return   false;
     
194. }
     ; y. @2 ?4 T/ G( T! o3 m- _

复制代码

天云上人

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

$ d% \: x* Q" c1 O% V2 Z
1 y: W4 C# A- t2 l- O0 a9 A///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 }" U. v8 I' F' [

#pragma once
#include <exception>

# M; C- h2 Z' |% W7 C' G6 C
  enum TRISTATE{
* n5 ~0 P# P7 G9 f% m9 J        TRIS_FALSE,
        TRIS_UNKNOWN,
# Z/ k: M2 L5 }& k$ K/ t        TRIS_TRUE
4 z8 b% p- D6 N6 T1 N};

+ J9 U; R( u4 Z1 g5 l) d5 r- V
/ i& v" H/ ~( \" [; kenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
, s: F0 t: q. u! z/ v3 o        UPNP_IMPL_MINIUPNPLIB,
) ^. H9 j7 A( z& U( y        UPNP_IMPL_NONE /*last*/
6 E9 L- O; ^- _/ o. j" @7 O) }! f; z: [};




class CUPnPImpl
2 z9 v' Z' _7 T$ s0 ]& n{
public:
3 A' m/ _9 i: C        CUPnPImpl();
        virtual ~CUPnPImpl();
; b7 h4 D! L) K; |        struct UPnPError : std::exception {};
        enum {
+ B$ i" q9 q6 u1 B2 O, T- F, j                UPNP_OK,
0 u7 p2 N: p1 l' F+ }( Y                UPNP_FAILED,
1 j/ z( J+ |8 G8 c! F1 P0 g# N' D0 |                UPNP_TIMEOUT
        };

# M* J# C) c$ L5 Y
2 o% ^. M- t# m+ w/ \6 T) ?& q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
; I7 j/ R$ P; X7 M" f        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
9 [' w, @1 }! c8 A       
8 Z+ N& U  M  m( p: b# B- ^        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

( a9 P8 x0 I& v+ E8 A
' p& x: h& M$ V5 I  V        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 S* S" U( r+ y7 N7 |& V2 C8 W( P        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
$ _) X8 _+ w* |. Q; C        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
, B1 K: H5 ~" Y7 q% `

// Implementation
, r" v3 J- u8 P  ?$ m' Lprotected:
0 v4 w- n. {3 B7 g$ |4 }        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
; V, n/ {9 _. h( ^* [+ h        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
0 l0 j" c7 P+ a+ l  n$ t. p        bool                                m_bCheckAndRefresh;
" U. Y: l4 B6 l4 V
$ _, L8 h2 s% L) a
" e6 O5 O+ Q* @8 I+ E: f* Aprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
& N0 p# @8 }% F6 A9 b- Ipublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
/ y( M' C* f  M6 H) D) B) g        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
9 A. e  D7 W) r7 w: K6 k- P        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
/ B" t7 ?8 _& j9 }6 T$ ?

/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
, N& B( {; f" q' M) }

  ?3 S( [+ K; x9 b# G#pragma once
7 [# {# n- p" F5 q7 }4 G. F! M#pragma warning( disable: 4355 )

# }2 ^1 M9 y* ~) }; V
#include "UPnPImpl.h"
  L3 G1 T4 x6 ]$ a#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
7 T$ D& O& p" K" b. x#include <winsvc.h>
/ y' W. ?0 T! b2 [
" x2 D4 H8 O' B2 z) V; e* H
. |. t5 x, V6 \0 s#include <vector>
#include <exception>
#include <functional>

$ V! ~9 u  F' m" e
7 l; w* B; q1 U& L+ j# L

8 U+ N9 ]8 N& ~8 R. g! ^3 `0 J: ~typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
/ u% O% m: s' |5 C8 Ktypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
* o$ o  l1 p' S& n; e% X& Qtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
* [: B: w' O$ S5 A1 Q

$ l& e5 `. p) y4 W$ G  M% Z6 e" Z9 Etypedef DWORD (WINAPI* TGetBestInterface) (
  B( b& F0 I: Z$ L- D( y7 I  IPAddr dwDestAddr,
. {4 k# M% L, ]( x! F4 a  PDWORD pdwBestIfIndex
: j1 \7 i  }, \( B);
- @7 r1 H  @) x+ G1 b! v6 G
! t6 J/ y6 o5 h5 c( E
5 H3 _5 c6 _6 @+ o9 Gtypedef DWORD (WINAPI* TGetIpAddrTable) (
& G6 M8 R7 L) ^+ Z- T6 q  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
" y) R2 O2 J9 P7 A  BOOL bOrder
1 q  y# D. B8 Q0 ?4 b0 L2 l);
* C; Y: z. s7 n( V5 ~* a+ O- ^
( m& B0 [$ h/ B/ w% g: n
/ p. D) H- G  Atypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
3 q/ f/ Y& e8 d' ~* e; ]
; n+ `5 s, }- u1 m" q
2 l* w" {$ a& p: l  B: c' zCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
- {+ q! l. F. R. n1 S

class CUPnPImplWinServ: public CUPnPImpl
{
8 W% f4 B% Y# v, a: C        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
" h( e3 P4 p2 R  n1 u5 M6 r. ~4 f% {        virtual ~CUPnPImplWinServ();
1 S- q9 w* S4 d        CUPnPImplWinServ();

# j. u& p# H/ d. x/ c  s! l8 M
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
8 F: D% I/ b% Y$ B+ J        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
& R6 ^" _( q2 T) n$ {        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


8 x9 }+ t* w0 B) {( {( b        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
% F1 z  Z2 X, u3 C) {6 T        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
. ~, Y1 G4 H  Y8 ^8 f        virtual bool        CheckAndRefresh()                                                                                { return false; };
3 A8 U" E$ ~$ I4 L5 t

5 y! a' Z" D- o; I+ aprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
, n1 k& T+ }( q) Z8 E) E& b& r+ X4 P( d1 _        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
8 P7 @3 f8 i* X) L

8 R& u* W7 j  f* L        inline bool IsAsyncFindRunning()
3 p% G- i$ r. W/ [% z$ {& i        {
: ]) i! Y8 {' d" u6 Y% H$ U                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
$ A% A% A3 v7 i5 {* g                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
9 C& _1 F1 P8 Z1 k3 D! `6 j                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
- B6 Q0 M9 L$ i5 C, P2 o                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
7 w+ ]- U0 e4 n" J                return m_bAsyncFindRunning;
        }

6 _, ^# |6 O" S
/ t1 B! s( H; i1 O" K, u4 U; h        TRISTATE                        m_bUPnPDeviceConnected;
3 C6 Q; U4 y) ?, P, I
, f" N4 V5 u3 G$ v
// Implementation
        // API functions
: k+ j4 r) s* h6 d) O( {! z        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
+ ~1 R0 Z& a5 J% k# z( r        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
5 j/ |+ _" Z+ a$ Z        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

2 Y( P+ S1 h  q5 L
+ @( x* c! ~2 f4 @        TGetBestInterface                m_pfGetBestInterface;
# ]4 A  w% h: W' _& X3 B        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
1 M. C5 {( K& n4 @5 ]+ R

$ ]- O9 g: [; J$ E5 G: y        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
! Z3 R8 ~, K8 q2 T* `0 d        {
3 a" j, e& Z, `                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
5 ?0 T/ h1 H' m* y* h( f                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
6 t8 S) f. F; S! ^0 ?

$ `  Y5 D2 k1 Q% w  o9 P                        if ( FAILED( hr ) )
& i0 R6 r  B5 z2 f3 f                                return UPnPMessage( hr ), false;
/ V, A. |4 {+ _( \, v

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
7 e3 `4 E9 R" q7 {: Z3 k                }
6 v5 \0 I! V  K4 T7 v                CComBSTR m_udn;
        };
' e8 h9 b0 B; z; J0 o, t       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
! k7 a* q: J) N1 l" D2 V6 @+ j        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
0 \( r& R; A# r3 J5 ^- c6 W$ ~) A% S        void        DeleteExistingPortMappings(ServicePointer pService);
& ~  [. |$ k$ }8 k, _+ Q$ ^$ P6 E& l        void        CreatePortMappings(ServicePointer pService);
8 z& M0 n. P8 q        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
" w& N9 q) D. Y0 p        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
9 y0 u% Y1 k  D- x; d1 m4 Y                LPCTSTR pszInArgString, CString& strResult);
0 f/ t3 H2 V7 f9 C) @        void        StopUPnPService();
/ y! v9 m7 q, K* }) K& E

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
' K8 t5 J9 p2 x: f/ ~" ~$ J6 y        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
* z7 h2 T2 ]# H: w7 w2 ^! a/ G        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
" {; R8 z/ ?4 Z$ e" B. {        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
* E" a3 a- z8 d; Cprivate:
( [$ i& a/ V3 H% t# p: C        DWORD        m_tLastEvent;        // When the last event was received?
/ W2 E: \" F  P3 q) N! \5 k# E5 I        std::vector< DevicePointer >  m_pDevices;
1 r$ \/ i1 V, [- l        std::vector< ServicePointer > m_pServices;
# S& Z, {6 X$ I, Z) K5 u        FinderPointer                        m_pDeviceFinder;
" J  U4 [, E8 \% W; S" U% ]  r% r6 @        DeviceFinderCallback        m_pDeviceFinderCallback;
5 s5 S+ |7 `( |" u& n        ServiceCallback                        m_pServiceCallback;

6 J& s) y% o( @
3 U; E& e; t% U4 N- Y3 b        LONG        m_nAsyncFindHandle;
  B9 u+ ^' Z3 D1 f  V: H        bool        m_bCOM;
+ H. p, \  p8 }4 R3 B8 q        bool        m_bPortIsFree;
, U- h9 f- C0 c        CString m_sLocalIP;
5 k. A0 e  x5 n0 J8 M, Q0 P        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
5 ^4 f) c2 C! I/ Q# I# o0 h' T        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
( A) {' w2 a0 _* b1 B' j: `        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
& I& E% L: e  B% ^# }7 W0 p' v        bool        m_bDisableWANPPPSetup;
9 _* q  c/ D$ i9 J1 a8 n( |

};


- Y1 ^6 I7 ^9 l" j// DeviceFinder Callback
class CDeviceFinderCallback
* \1 G$ j2 [% Y+ c        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


  Q; c' |6 a3 |- @   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
) j5 h, n4 c" U) [5 y/ W$ A9 F   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
5 a" a, E* @; ~+ p

// implementation
1 y8 J& ^( r4 [- t, n! Oprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
* Y, D- a# w: J5 }        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
& u; R9 ~' Z4 j& b; e        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

' U; V  ~9 a  |2 {1 w. `' c( f
+ o' U+ f) p. V7 X; q+ X& l5 l// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
+ C  F) W) g. Q% d, |! [- ?4 S        { m_lRefCount = 0; }
+ R& f4 u& p5 n# ]   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# V1 Z) R6 j8 l9 j8 x7 s   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
2 a& a) M- A8 {$ u/ c  z

// implementation
- n: G7 d5 Q: A! tprivate:
4 j2 y1 L# F  f7 b- U$ [        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
  _7 H1 Z6 r  B6 i2 V- g        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


6 y" P! U  g6 P* h& W" d4 a: kprivate:
        CUPnPImplWinServ& m_instance;
2 L6 Y7 D4 Z7 y" W        LONG m_lRefCount;
};
! l4 U5 _$ f, ~+ @
8 Z: C# I' R8 X$ n5 t
) }4 {5 d6 h5 M8 F/////////////////////////////////////////////////


: e5 s9 ?/ r& K4 c' Y使用时只需要使用抽象类的接口。
* {7 R  L4 n. Q( h3 W% K5 O- b8 r( ?5 g' YCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
6 O; @% H2 F  x% J  rCUPnPImpl::StopAsyncFind停止设备查找.
1 f9 [6 H/ f& }' Z$ U6 E9 E2 LCUPnPImpl::DeletePorts删除端口映射.