UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   ; p  i, L4 c; d  @" u4 O
4. #pragma   once
   & ?0 q) X8 M" _$ T4 M" a- A) R
5. 
   ; s( t" E% H: `" f& `8 A! v3 r6 ]
6. typedef   unsigned   long   ulong;
   ; B  y5 z0 d: g$ Y/ x5 L: O( P$ y
7. 
   ( E9 N( N$ Q0 [3 x* d
8. class   MyUPnP
   ) n  Y4 D* z6 ]1 _4 O1 D9 j' z
9. {
   
10. public:
    
11. typedef   enum{
    & |1 o1 E! s$ ^! `
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    3 g' J: ^' k' a) U) I( v
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ; m* g8 G2 k( M8 H# o) H
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;
    5 \% d! O1 a+ o& }
18. 
    4 ^/ Y# n5 D& Q+ O9 E3 e+ A
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    7 N6 t8 {3 o7 X8 e! W
21. UNAT_UDP //   UDP   Protocol
    ! t4 g% n3 w! h+ s/ [
22. }   UPNPNAT_PROTOCOL;
    
23. 
    & S" G- _: R9 o
24. typedef   struct{
    3 ~$ H  |$ N# Z2 f  C
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    6 A2 }9 J, w1 r
28. CString   description; //   Port   mapping   description
    9 N- ^0 q1 g; ]5 t* d
29. }   UPNPNAT_MAPPING;
    , e$ G& k# K1 i8 p
30. 
    ( U1 Q" x/ |2 {: _! P/ g* A) p
31. MyUPnP();
    # ?; @2 \$ o# R$ M
32. ~MyUPnP();
    ' p3 ~2 y" m6 Y7 t/ m8 A" V
33. 
    3 t+ J1 f0 z- S" A2 x$ |6 ^; V) d2 n
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    1 M% g  {% `8 L% n3 ^
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ) j" }  Y1 F9 W' A$ C
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ) _' q( M; x  Y8 r& E+ q9 o! o
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    * n* @2 s7 r9 Z" f) D
43. protected:
    
44. void InitLocalIP();
    3 f: u4 c6 s: _
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    , C1 t* o" v3 @4 j/ X
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    4 L& l; `) U- q- C6 Q* S6 t! ]
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    4 w! J+ B2 ~& x' p
54. bool GetDescription();
    + f0 p# _) r3 [6 L$ X
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());}
    / q/ U) s; b- v: x8 r7 X
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    $ Z+ a4 t$ O' l/ m) n' F% M, l
62. CString m_description;
    6 Q3 s1 E3 t8 g. x: E9 ]' r
63. CString m_baseurl;
    
64. CString m_controlurl;
    3 p0 X& ]( H; A
65. CString m_friendlyname;
    
66. CString m_modelname;
    ( U5 _4 ?7 L* C0 z- h# t4 `8 P0 ^
67. int m_version;
    + V6 |3 q3 i$ z  Y) |% \$ G
68. 
    ; B2 e7 ?, N7 T( ?
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    9 r" o8 _2 h& Y( |
71. 
    
72. CString m_slocalIP;
    : v, F  _, j- F/ }
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    4 H  z3 }3 h* ^5 H
76. bool isSearched;
    # c2 d4 ~. S) g
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   # b# R3 k* W! z7 M3 ~5 r8 E
2. #include   "stdafx.h "
   # R+ K$ Y' @. Q; I1 d) s
3. 
   - y& H- O+ p& F  W3 @" P4 m
4. #include   "upnp.h "
   1 c+ V4 [4 R* ]6 g5 R/ v
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   7 ?% w" l* v/ {# {& A& V' [
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   1 ~" ?2 o7 }0 ~' O
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    9 H+ I  ]3 t$ `* J
16. const   CString   getString(int   i)
    
17. {
    6 O; I" _2 s3 a5 `+ ]
18. CString   s;
    ; r+ o0 U$ X2 N( A0 w4 C
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    2 K9 _' {4 f$ m( d) v4 a6 @
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    9 _/ {/ u) O" s+ u" \' F
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ' y0 l: n3 x" {, q
28. }
    / ^+ k9 E, |" t4 v/ N0 ~
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    + Y% R' _- R3 i- m' a
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    & ~- `: L9 c4 J6 X! O9 L& X7 |
33. }
    . t" O) M2 u+ n3 s/ R( r: y
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    + R$ y" S  d5 E* d+ D2 V: e, s
37. char   buffer[10240];
    7 g, W8 p/ z) s. w8 q8 q
38. 
    $ x1 d. K7 u' u, u: c
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    , b/ L" X. a% L
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    % N0 T0 J4 ^7 {. V$ ^, d
44. struct   sockaddr_in   sockaddr;
    ( @3 J5 R8 Z( Z( m+ n4 W
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    2 p! T' `0 q4 a, v- p
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 R# b+ C; C% g; _" @
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , U6 G/ O) i; k) U' O, Y
50. u_long   lv   =   1;
    # N; m* V3 G! {
51. ioctlsocket(s,   FIONBIO,   &lv);
    2 y* E3 ^* g: ~
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    % D6 r: x* V' M% k+ \
54. int   n   =   send(s,   buffer,   length,   0);
    5 a% z; ?6 c1 V3 C' n' W. [  A
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    + c& F: l+ U7 L0 E' B, M8 o9 x
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    6 ?) q5 U( h8 _# W& z5 R
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    1 W- N% |2 X, k2 ~. b
67. {
    
68. char   buffer[10240];
    5 n1 F5 d" z6 F; ~1 n7 R) g6 a$ S
69. 
    + {! B- E: I6 |+ o4 l8 s- L
70. const   CStringA   sa(request);
    ( Q4 F* j' G( e' f+ m
71. int   length   =   sa.GetLength();
    % Q2 w  @) K6 l. i/ y7 Q
72. strcpy(buffer,   (const   char*)sa);
    ( W0 x, H  D7 O% i: t: ^; w( E
73. 
    2 r0 Z2 Y* A8 U% _3 `
74. struct   sockaddr_in   sockaddr;
    $ t! D& H, c6 [2 `9 H2 F- w3 K
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    6 I* t9 e( \$ ^! t
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    8 N$ [3 V4 {% D
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & [" d8 U8 F6 g" q+ [* s, ~% \
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    - k% z8 i4 a5 ?$ K
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    / E! x0 l0 U$ f+ f* q" F* V
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    % C; M3 D+ B9 j
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    8 d, q: d9 e& v+ ^, m) ]
98. 
    . k/ _& v  x( q6 w+ t9 I4 u4 Y
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    0 `2 {1 @7 N  A5 V2 Q$ N
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ' a9 v% U2 f% P, Z9 [& [  G# p  C$ M
103. CString   property;
     
104. 
     % e& U9 v& w5 G, w& I* X
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     , Z4 v' p5 U! X
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     3 v) G$ X- k+ N& O; D1 Y
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     2 X4 o* d/ a8 A; c  ?. d. ^2 W
112. }
     4 Z# L) J7 `* ~$ D3 E
113. 
     # Z% [5 N4 j1 z* J* p
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ) h; F+ ~8 N) ^; H
117. m_uLocalIP   =   0;
     - e; c& d% U/ Z0 ~
118. isSearched   =   false;
     . K& x0 x1 y  v: h8 m" }% S$ G8 x
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     ' m5 L1 h( h3 Y# L, |% y9 b4 E0 U
122. {
     ! P3 n7 P  T) \# v, p
123. UPNPNAT_MAPPING   search;
     5 y5 {" p9 D. h, L1 A
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ( J6 W3 s. q+ Y  O/ j
125. while(pos){
     , C0 R7 e- s& p9 s
126. search   =   m_Mappings.GetNext(pos);
     2 I+ T# l4 l# `+ P- a! p7 F
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     : [; u, q. F' g: v  \( w6 c0 D
130. m_Mappings.RemoveAll();
     - ^1 H9 X# Q! B: D3 s
131. }
     
132. 
     - B+ Q* @5 X6 z  h3 A
133. 
     8 Y  M' H- p* |* |( s
134. bool   MyUPnP::InternalSearch(int   version)
     , T: M; U2 Y/ T
135. {
     6 R% \% i6 f7 ]) X" u. e
136. if(version <=0)version   =   1;
     5 r8 t: d" x: P* h9 A+ j. g
137. m_version   =   version;
     
138. 
     : w, a2 v; R0 Q/ X9 c
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     / n* F* {2 t* W4 \/ k6 H: a
144. };
     ( y/ _7 G4 |7 M* O/ \0 w0 C
145. 
     3 y5 M# w+ T8 u* x! @" t/ W
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     6 u( h9 l( Q) J4 e" d+ t, I
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     * ]# e* E$ I5 R% T! z' ^, t5 J
150. int   rlen   =   0;
     + r+ |/ T3 W6 z: E5 x& G
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     9 j; \. e' s7 @
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);
     : W$ ~# T  ]: O9 M4 q4 r8 B+ D/ q9 D1 i
155. CString   request;
     & T0 Q/ U. _( r$ q& P  m: w
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);
     % p) a. b7 v* r! Q
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     4 b! J8 x: n3 S7 q4 i6 `1 K: w8 \( u7 w
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     9 [) E; [& J9 s, O  r
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     6 b# W. c# P+ j+ ~+ L( o6 H
167. closesocket(s);
     ' p1 Z, d7 O5 L& ?: O& |+ E
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     ) d. O1 W; Q. Q4 U7 Y; l$ v% W
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);
     ) |$ Q. ~6 v0 L" H
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ( W/ O2 f$ I7 [& n) Q9 z
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     5 b; d: c) u" A- Z3 ^
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ! j% `; K7 k9 W( `+ G
182. line.Delete(0,   9);
     
183. m_description   =   line;
     3 X6 a( n$ L1 \! A  n
184. m_description.Trim();
     ( G( ]4 h  f1 E6 _" `0 \
185. return   GetDescription();
     0 u& _# i1 u) s4 y
186. }
     8 o3 N; I6 G+ K
187. }
     
188. }
       m+ n  {$ {3 h' x' p5 `1 F
189. }
     
190. }
     
191. closesocket(s);
     $ K0 h  H# S/ H  b* i$ V
192. 
     2 g$ u8 v& j) s0 a# h8 ~, O  M: h4 m
193. return   false;
     # v$ ~) k/ h1 T' G- F
194. }
     ) e0 @. P# }9 i( d! X

复制代码

天云上人

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


% P7 }  U& H7 i# x* \7 x/ t///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
4 W- Z2 }1 M& K; H5 |7 B
; k3 n* i2 @) r$ ^
#pragma once
6 ]0 \( _# U3 Y: Q6 G7 a! S#include <exception>
# j+ F0 W$ g8 P3 m/ d0 u
7 f$ r2 B& I# z8 T! o# N7 [2 o3 v
  enum TRISTATE{
        TRIS_FALSE,
5 c# G( k! }9 X        TRIS_UNKNOWN,
. K& {* X  X5 v9 Z: s        TRIS_TRUE
};
7 K' I' J% p- S! c

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
9 V. R0 u2 k# E( ^/ a0 i};

1 t+ z8 n, o! u+ @1 ]3 z  s9 }


' ~8 J/ \7 E6 E4 S, Dclass CUPnPImpl
0 _( C) R; g# x{
public:
        CUPnPImpl();
7 f" I( W9 O8 Z        virtual ~CUPnPImpl();
3 h2 r- a5 T# i0 C8 k/ N0 u& M        struct UPnPError : std::exception {};
4 \3 M* Q3 g1 q' i4 k. F0 a: x        enum {
4 r2 ^  L& S) Q. F9 ?                UPNP_OK,
+ v( @+ R' \& p5 q1 B4 G' Q$ ~                UPNP_FAILED,
- Y6 K8 B3 c1 W& M                UPNP_TIMEOUT
% e4 q5 Q; K6 R- @6 O5 x: p! }* [0 x% a        };


2 c) ^) j/ i/ q' j/ Q7 x        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
' w: \# n( u6 W! X% z9 x# T        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
% O# L' a7 E9 u! p* k( O. l$ l0 M        virtual bool        IsReady() = 0;
7 ]% r: a% @0 x; d5 d' l* @* ^) A        virtual int                GetImplementationID() = 0;
       
% K7 i2 J' N5 Q5 C/ b: }        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
- B" X9 N- K7 k# y# z9 {9 G
1 y- U& |" A. k; y
6 t) d3 h2 }% N// Implementation
) K8 i# S; |! l$ D' d) J( Tprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
& G* K- M5 m- N. ]+ A' d# K5 `        uint16                                m_nUDPPort;
* T0 P0 F" Y' t- A( Z% F! K6 y  [        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
4 [4 g6 G; f* H( c7 l$ Z5 f( I        bool                                m_bCheckAndRefresh;
( C! M& j. x5 G+ V% V: f

  J! S) Y9 u; x. ?- @0 [private:
: b$ r' v0 X8 S+ c' |        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
, j3 {9 e$ l; b1 `
; c) X, F+ @* P- P/ Y  N4 d
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
* |: c& K) h6 n! \+ C! [  l+ f        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
- m! Y7 p. a! h: O        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


1 _# Z+ Y6 ^% K! J( a#pragma once
( |* ?- O6 }) W  i#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
& G3 z2 h9 u: r3 c- I+ n#include <comdef.h>
+ F6 X( k. O# N$ o#include <winsvc.h>

( R5 A$ K$ k+ @( K# g$ x# h
% R+ b( O- b# W7 S6 s; `, ~#include <vector>
/ c7 q  d8 |0 L1 U# f5 b#include <exception>
#include <functional>

, M7 e& v  B' L7 K2 b+ ]7 R* X+ q3 H6 l
& \! S+ H' W0 O# G( q8 Z) B

7 o. g! Z& {" n$ O0 Otypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
- l6 g/ b) [' o, r# m, Ztypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
; R2 s# H0 w! I4 p; H, b; L" ntypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
0 h  E' P3 ~. Z- @typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
. L* ^/ Z7 C7 F* Z3 E2 e- ^5 n8 _8 e" D! Ttypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
' t- _6 G: Y3 V& u: z$ G- h! D9 i  PDWORD pdwBestIfIndex
);
* W! W5 E; g- O+ s* u

! c9 H8 R8 e/ N0 T& m0 stypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
& y: u- |8 y6 w! k( I' M

. z+ F- }8 E5 u" b) ?. a' dtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
" M3 Q: k& B1 z) h% i);

: O; r; T9 _# O6 n+ C
% X, u/ `2 o+ X! N  Y: _CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
2 f4 U' ]1 I) z
8 T) t" j1 F) ~' t2 c
class CUPnPImplWinServ: public CUPnPImpl
1 {7 y3 g6 R1 t% u) `{
% B- o+ J/ ]. T* h5 r) E3 X/ W        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
. T3 _& e0 W7 d/ d: v. `2 {// Construction
7 C- ?! p! k3 f8 _3 l& L2 t1 epublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
1 v- M0 W  _" M# P( X2 U
# h% q+ J  r' _4 W" e
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
3 O6 E8 G# y! l/ T) C# S        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
' j5 D5 c$ R  O( p        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

. [$ T9 k1 g4 ]
0 H% a: q$ A5 D! n, {        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
- h3 g# B; O9 X' i        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
! ~$ q( `6 \) J( E% _* Z6 G

protected:
, K' S$ K2 T, G( U* v/ H        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
# `  N9 H+ {. s: l5 R
  j3 H5 ?6 @! T( a- b0 B
' M; _1 Y' Q+ e0 ]& Q. ]8 t) K        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
5 z8 F2 J* k! K                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
% e2 V$ F, g( H2 e2 f0 R                        m_bAsyncFindRunning = false;
" X. T8 Z( Q7 C: H, {                }
7 F! C9 `8 ~$ ^- z2 Z% }# e/ u* s                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
+ r+ S' x. F) [' ]  E' r% o+ v                {
                        TranslateMessage( &msg );
4 c: g3 G. l. y4 V                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
) N2 D% v8 v% L/ \3 U; K        }
8 ]7 f! ?! z) S2 ~1 k

        TRISTATE                        m_bUPnPDeviceConnected;
$ q( @' x% s4 j8 H
9 K0 ^# V; o# p# k4 P
// Implementation
* D! K& Q( l3 k) k9 w- c        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
% Z9 e# Q0 N  x# K( k! ?        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);
; V/ j& t* Y' ?: t

( Y, q6 |# B$ B9 @        TGetBestInterface                m_pfGetBestInterface;
3 N9 d2 ]/ {* S9 j5 I; T        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
3 y# C& @# G  U! i& T0 l! m' @                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
5 l7 n7 r- e! K2 B                result_type operator()(argument_type device) const
. ?% v, c5 a% g4 n3 R2 D* J! |                {
# L: k! {% v) M3 R; z0 Q                        CComBSTR deviceName;
+ c) j& D" |4 X, E7 p# `" V2 ?                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

$ [- e" v: d, a* u1 P4 y
/ ?+ i1 ]& ^; A$ q                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

# E8 V( R* d" y' S6 K
8 |! |9 ]6 ^1 N" u" E$ P                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" ~5 Y) U$ p# M: v" g8 ~  f$ a                }
+ Z  n2 h8 W" M2 Z- C                CComBSTR m_udn;
' Q  @, y9 H7 B, F        };
3 y/ W6 q" o1 D2 L5 [6 e0 m: ~       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
) A3 r2 z+ h. F6 x' r        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
/ B' q2 b2 y8 \1 [- J0 R        HRESULT        MapPort(const ServicePointer& service);
) @' j7 Z& L- J        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


$ N0 ~/ a- g+ {) y# C; P; }        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
5 F  {) e+ K' }- p/ T0 v6 O7 d        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);
1 B5 I& Y3 F& D4 W        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
/ U1 ^: [" U$ R$ R- J+ c- c        CString        GetLocalRoutableIP(ServicePointer pService);

+ S3 \$ X9 |/ z/ k; b3 `" f, b
: E3 O! _( D7 ~( U2 R// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
! Q8 o' l8 W1 W) ^4 }        FinderPointer                        m_pDeviceFinder;
; V: L+ L7 L  I  R/ ^: a        DeviceFinderCallback        m_pDeviceFinderCallback;
6 A# E5 j0 z: s7 N) J, G  t% \# s        ServiceCallback                        m_pServiceCallback;

# r- B# [) D3 t' c. j$ ?
7 s5 ]# }+ i! i( d8 W7 ^        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
" ?2 G! w0 O+ k# _+ Q/ }, s8 ]        CString m_sExternalIP;
6 u9 q& F, l4 L6 E% C        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) O6 ^! |+ H0 l- R        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
1 |' }/ R3 J( ^0 t        bool        m_bSecondTry;
2 Q: z6 i1 [$ A- S3 h! e* c        bool        m_bServiceStartedByEmule;
/ q1 k- d5 y0 {) t        bool        m_bDisableWANIPSetup;
/ g! b, @3 S1 x        bool        m_bDisableWANPPPSetup;
6 |- r) l5 W( h  _$ L0 a' o

};

& l1 q, H0 x, [
// DeviceFinder Callback
: k& d$ S0 ]5 n6 L+ H7 \& [& u. D$ v/ rclass CDeviceFinderCallback
) Z0 N; e1 _, K, V8 }4 J3 z9 `( k7 S        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
( S) D* p" o0 l                : m_instance( instance )
9 V# ?4 k1 c3 R  W8 D        { m_lRefCount = 0; }
; t* a0 X& X0 A6 W% F

) {* P$ Z1 p  M7 A  W* t* j/ t   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 e2 i; r* I" Z! p   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
0 p2 |( H- R6 ~- P9 H6 G

, J- y% P: F: f, D$ I. d9 Z% G// implementation
7 C: U* T1 E7 T7 R) Tprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
4 i' L, C% Q0 f, m        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
8 q% M/ b! ~0 p: f* J7 D: v        HRESULT __stdcall SearchComplete(LONG nFindData);


2 T: W; Z/ K/ S' a! R% ~  wprivate:
7 L) A: J7 T1 w* ]) S        CUPnPImplWinServ& m_instance;
: l, v  y) k/ z        LONG m_lRefCount;
; E. l+ P: \6 _' M# \% o: x  o* h};
2 {$ D0 `4 H: u' f7 f
$ f7 o+ w2 S1 s3 g* o
6 C5 [2 J$ J1 ]+ m( p3 Z5 q// Service Callback
class CServiceCallback
% N( c. y- j8 K) P) M3 }        : public IUPnPServiceCallback
{
/ @# J* F% P, I! b6 s8 l) c5 z) [) _public:
        CServiceCallback(CUPnPImplWinServ& instance)
# o6 H( c9 V/ A- T                : m_instance( instance )
2 g: |9 _2 Y, e1 n  h        { m_lRefCount = 0; }
   
7 M# X1 ?' ^1 s   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
& G, |3 E/ |. H9 r; f3 I5 c   STDMETHODIMP_(ULONG) Release();
; q6 W+ W$ ^% Y7 u* ~- n, ^0 L
8 ]3 J2 h: K$ c9 w% [' M5 @8 o, C
// implementation
, K" [& w& o3 v: y0 gprivate:
" e0 ~+ M6 x- h" |$ X        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
: w/ s. [7 M3 I8 j
+ K) K, A* A  U  {! t4 S
private:
& o/ s0 @/ n; ]* G8 f1 e/ n        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
. s: G0 K: O, k- g3 l; j" O6 S};
! u+ r& K/ ]( n; u+ P9 f- X: R
+ Z$ S. I/ ?- h
6 A) q- J. m& c! \: p/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
+ O* B; G) Z, Z4 t9 C% dCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
7 L' R2 S8 c& A' b; aCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
/ i# Q% D" v2 e# E. RCUPnPImpl::DeletePorts删除端口映射.