UPnP

天云上人

/*uPnP.h*/

1. 
   5 _- i; m3 L  h2 G) ~* `
2. #ifndef   MYUPNP_H_
   8 B8 q' D9 y$ l) M& j" L8 U! Y
3. 
     n# ]1 E" u, K; C6 R
4. #pragma   once
   
5. 
   # Z) e# S6 u, P8 I8 d8 O; U
6. typedef   unsigned   long   ulong;
   5 J: ^9 m  ~4 o7 ?6 g) i8 A
7. 
   
8. class   MyUPnP
   
9. {
   ! {9 d. g" F  K$ p
10. public:
    % I/ S& @, H, G) J1 W* ?" T
11. typedef   enum{
    
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
    - Y$ S$ h$ v. p. `* _
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
    ( F' W8 T+ I: S) ?) R
17. }   UPNPNAT_RETURN;
    : B/ y) K4 _4 I! Y* r5 V
18. 
    1 {2 p2 \3 r' y7 w$ @: d; s+ A
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    / ~, O/ m* ^' c0 T' {8 m6 V
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ) B1 O0 v/ y9 ~$ \' g! b# l; v
24. typedef   struct{
    : e/ U( o- C$ D% k9 h. G
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)
    3 k/ V  c# w* P: l4 j3 v
28. CString   description; //   Port   mapping   description
    4 M& S5 x0 ^+ Y6 O
29. }   UPNPNAT_MAPPING;
    ! l  w" e2 t( j5 c9 _  e9 g
30. 
    7 s9 h; ]/ A6 X7 q' b
31. MyUPnP();
    
32. ~MyUPnP();
    9 C5 ^) ^4 t% x' J4 s
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 U" J2 g" c; V% _/ i
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    ; X) Y* A9 Z+ n- v2 |, F
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    0 E4 k  S* L6 [  Q+ E  i/ P; v
42. 
    
43. protected:
    ( H- m1 U3 X  I) P
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    ) N! I6 }4 u' n( w. }' o' O- G3 B
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    9 j6 t5 j+ V: @  X5 h  c: j
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    : i( K$ l6 {, ~  h( R% A" z# {
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    " A7 C! i! e8 ^/ m
52. 
    & f, U6 t: o& G6 X; E+ ]8 J2 q& z2 n
53. bool Search(int   version=1);
    ! c  `, {+ T; }3 C
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    / O# ]+ f! ?2 s( H$ ~5 S5 m  s7 T
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    1 d1 a5 ]4 [/ V$ N, g% c
59. bool InternalSearch(int   version);
    * X( @5 S% C& V
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    3 i! J9 @7 N# v
64. CString m_controlurl;
    ( K+ M. T. S4 G; w1 E
65. CString m_friendlyname;
    . [; A. o8 B6 D* ^* F& L0 a5 ?' r
66. CString m_modelname;
    
67. int m_version;
    / X. z1 L# ^- h' e: x5 F5 h
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    3 X% z0 w0 @( f4 ~, S
74. WORD m_uLocalIP;
    " P& M: n/ a, X% v9 e; t
75. 
    
76. bool isSearched;
    6 s7 L: b% s! C+ T, b4 z+ \
77. };
    , l* y5 }- l( `6 B
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   4 B+ e3 L4 O. n. i0 `, g
2. #include   "stdafx.h "
   
3. 
   ! i/ n  i2 q6 ?/ P' F4 v
4. #include   "upnp.h "
   
5. 
   . a. P% g0 r7 t" `# J6 {
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ; W& d; Y, @1 m! y1 q9 D8 z
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   $ E6 K# Q) d" r. w" w$ ~5 N
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    1 x+ ~9 ^2 G' p
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    & A: S- x! ^3 O  f1 x, C$ Z
13. static   const   int UPNPPORT   =   1900;
    . N" h/ c' [- f. i3 r) h( T
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    8 M' l- W' o5 {/ L
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    ' {5 z5 \8 a( a+ E4 ]
21. 
    / K5 r) ~$ V7 j. B" u& ~
22. return   s;
    
23. }
    
24. 
    . p) z) c9 R  W
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    5 W6 ^9 J/ {3 `9 i) O# `& m
26. {
    % z2 S! h) ~/ z9 Z, T! K9 d" D
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    , W% l" D7 b( d- K
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 m0 t) i6 @% u* `7 Q0 D
31. {
    ( g9 k7 {5 m/ P% v: L! d
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    " m" X/ }4 K6 B  F) ?+ o7 ?
33. }
    , e  v$ [- e# q, s) [
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    4 j5 K9 f' o0 o3 b5 M+ _
39. const   CStringA   sa(request);
    4 k( \- p$ T( c# W1 ]" t: U
40. int   length   =   sa.GetLength();
    2 `& ?" l1 j  f: v6 h
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    8 J8 c( A4 |+ a: N
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    2 y; c: [  d( o+ _& T
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    & ]# R" t9 e8 h* Q" O( P5 P9 p
50. u_long   lv   =   1;
    6 o# u% O' x2 I- ?- \. ]
51. ioctlsocket(s,   FIONBIO,   &lv);
    & w8 R) {( ^  \3 C0 Q+ x# f
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    7 {& V$ g! H3 J- |; P, R8 G6 j; ?
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 A+ g: O* ?" X5 o- j2 F( g
59. if   (!rlen)   return   false;
    ( z$ p& H9 K4 t9 S' K  W
60. 
    ) M/ N& Y. U% M+ u# \5 R" c0 d
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    * B) [7 O8 l8 F1 S0 {0 q% k
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ) R. l' J8 C: Y" I' `& ^. l8 z6 v
68. char   buffer[10240];
    - t" L1 [0 m1 f' K& @
69. 
    / {+ d0 ?1 D+ ]6 j3 J
70. const   CStringA   sa(request);
    - F9 a7 B$ E" l3 O9 p% R
71. int   length   =   sa.GetLength();
    : s9 i7 P, B+ [: z
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    " s) x( n$ z: D* O7 `: i
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    5 h9 D+ u" j, d- u3 k* T5 S* q
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    7 n0 _/ D% w8 s$ D* {6 p2 x% Y  e# V
85. int   pos   =   0;
    7 D+ O/ L2 D4 s/ G. ?9 E* b
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    4 y  p2 z  n/ A/ a
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    - z5 z9 d* h- d/ H/ W1 S+ O
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    3 a9 V9 n7 a/ U) G- f
96. return   true;
    
97. }
    
98. 
    ) ^9 G3 L8 v' v6 G1 g+ h
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     0 a0 R. n& N6 I& }8 `# G- ]
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     : x5 t" }0 Y) C: R$ _( y
106. if   (posStart <0)   return   CString();
     , N/ B- K8 z( @- k0 K+ {
107. 
     ! U/ a9 M4 |3 {% F: ?
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     / n! z# i3 e% G  n8 l. a
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     % A8 ~; }  O( _5 V# y: `0 K; t
112. }
     
113. 
     - S3 i+ D/ L$ k$ q! h
114. MyUPnP::MyUPnP()
     ) B* ^" Z# {2 S& l8 Y" F8 m+ X
115. :   m_version(1)
     / p7 i# c% J' K. C; Z/ x& z3 ?
116. {
     $ B; a0 v. p2 s6 ?# R5 s
117. m_uLocalIP   =   0;
     ; w8 r0 v  K! K; o
118. isSearched   =   false;
     
119. }
     ; t# l2 @( i; h
120. 
     6 J  x. C" D4 A% @! r
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ! H. |" {$ ?& O- a) W  w+ t$ @
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     ' [( g. D- q* z  ^- x6 {0 }
130. m_Mappings.RemoveAll();
     2 l4 C. U8 _( }5 ]
131. }
     
132. 
     7 n7 Y  ?/ d7 _. U, ~
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     " v! @2 m! y! g
137. m_version   =   version;
     7 Y; j/ B4 X' r
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     : R' G4 X" R% Y! k& u
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     6 x5 R! i7 D( q( E' R7 {8 n
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 e6 S1 P# m7 Z; V' K2 d
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     , I+ h$ J  ?! Z- y% F, k: L
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     + z9 l& C' k/ o
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     7 N) u+ _! T5 Z4 v7 N
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     9 b% e4 O4 ~9 l& E. |2 U! x* K% D
155. CString   request;
     2 ^$ `; y! i6 I: q$ z8 r
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 "),
     7 q5 `4 U& V. j0 K! h. V
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     1 }  n0 }5 z5 j$ b
159. }
     $ ]/ k0 m: X1 p) q5 A3 O
160. }
     
161. 
     - _" G* _9 ]2 f+ d
162. Sleep(10);
     
163. 
     . n/ b) i7 Y( d) o2 ]
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     8 m; q$ l8 {7 ?* x
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ; h; K* @, W. n4 u
168. 
     ' z: m& e8 r5 C/ P2 |" t( \
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( |3 Q* L' u& j6 v) G7 }- H( s
170. CString   result;
     3 s( F- z% K2 W1 i) r
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     1 U/ J, Q' m2 d$ H% C
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     % r" f& v! t- e4 b# `$ x  u
175. if   (result.Find(m_name)   > =   0)   {
     " z4 `% V) j/ H: }$ o7 Y5 p
176. for   (int   pos   =   0;;)   {
     ; C/ P! I: ]# g
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();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     / o+ M$ [% B% C2 T5 y" M
183. m_description   =   line;
     ' s: d/ G% k7 a2 B; r4 n
184. m_description.Trim();
     . y7 `/ ?1 Q+ m1 A
185. return   GetDescription();
     ) T  M* f7 e2 c' a0 r6 o
186. }
     
187. }
     
188. }
     ) `! H" L) b: l2 E$ f% N
189. }
     ! k; v+ L/ w* ~  N
190. }
     
191. closesocket(s);
     ) ]) r4 N, `# Y
192. 
     + L$ q8 v6 @4 j5 q
193. return   false;
     8 G+ G" X" g, \0 h* G1 J( ]
194. }
     

复制代码

天云上人

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

; }; F% I& [, Z) E* v9 L
+ k9 B- C( P% H, J8 B5 c) ~///////////////////////////////////////////
5 u$ y* T+ \5 _//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
- Q% f  i9 }: K. k
* M+ m+ ~- a5 F% g* ?0 D
#pragma once
& @. ?( s1 @: w#include <exception>
% \  x" T  @& e0 r2 z
( U4 o: [2 b% F- z0 t+ i1 K; `
; d) H( _, ?: }( C7 ~5 W  enum TRISTATE{
) e+ ^: E$ m2 g        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
- p" N+ b, G+ ^4 g% Z# R; ]' o};
( V, X! B% j# ?- Y/ @2 {( ~$ R

) ]5 e  [9 Z  n7 z/ `  jenum UPNP_IMPLEMENTATION{
/ |' A, g1 m$ m; ]3 ~! N& g! b        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
- m$ p. Q2 o$ P+ w# \        UPNP_IMPL_NONE /*last*/
};

+ a, e) y) I0 X0 l5 r% u) u! s
! D. j/ h: N# H, g. Q: Q4 q7 |. |

3 z+ ]- E  a; p# j4 Oclass CUPnPImpl
& K4 R$ H! O' ~) h; ?. y& j{
1 @4 V+ W4 R/ }public:
! ~9 H9 O3 ^+ C. Q8 R+ B        CUPnPImpl();
        virtual ~CUPnPImpl();
' o$ d* \+ H  w        struct UPnPError : std::exception {};
        enum {
; V  Z. b& ]3 _1 ?5 m. w' U6 Z                UPNP_OK,
4 W. F) R! b6 r6 f3 I) ^                UPNP_FAILED,
% @& N- d8 Y6 }. F/ w% e                UPNP_TIMEOUT
+ J1 V. T! a% M+ i1 \# {        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
8 O0 Y. G. b2 v- C$ f3 P  A        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
% t/ Y3 Z, R# K& j" o' Y        virtual void        DeletePorts() = 0;
: ~0 ^/ e0 x- V' g7 W        virtual bool        IsReady() = 0;
9 s1 `4 m  B  q9 G& V  e5 S        virtual int                GetImplementationID() = 0;
7 P* o9 X" t! X8 {6 D) |( c       
7 G( \& e3 z- x/ ~% t8 j& |% H        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
" q1 n. J* V) U, |3 r: d2 V        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
2 W& {) V/ z" M* P+ I6 V        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- c6 _: j1 \* h/ h3 J5 R" h        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

" d. J. L8 G2 d2 L8 i
. m3 K$ b% m0 }! @2 j' L* b0 [// Implementation
5 A+ O/ V% ]( [7 O5 @3 N; iprotected:
3 f/ y) ?- n* v+ d; V        volatile TRISTATE        m_bUPnPPortsForwarded;
9 D5 T& m$ |# G& U' E* {        void                                SendResultMessage();
, n* ?" M8 o/ S; d$ s        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
: T  r1 }: \6 f2 K' f/ A4 N        uint16                                m_nTCPWebPort;
7 V+ g/ v7 W9 q* p. w  w0 R& _        bool                                m_bCheckAndRefresh;
1 O& s3 w) k1 k
$ L- c. R6 Q; x$ I; }2 p' b
8 ~/ x, [% o. I* eprivate:
        HWND        m_hResultMessageWindow;
( ]- {- Y3 o$ g. x        UINT        m_nResultMessageID;


2 Y$ y  v1 {6 r# W* W};

" A" `/ s9 Q2 S2 ^; z
' t* w6 \! J! M9 m. @! U, H5 x// Dummy Implementation to be used when no other implementation is available
9 C' r, c7 M+ a( B! Zclass CUPnPImplNone: public CUPnPImpl
' P& W  D+ I9 |% e5 g{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
9 e. K; f. d; D/ s' G        virtual void        StopAsyncFind()                                                                                        { }
' f2 Y7 B' P  O2 I" Z; P        virtual void        DeletePorts()                                                                                        { }
8 e* X2 H9 o) a* R8 {        virtual bool        IsReady()                                                                                                { return false; }
: |% z3 \- \9 I/ a- o$ Y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
+ S/ ?! R1 `: m1 X% }
9 G7 s  Q5 D: x' t2 E) N
2 n( A+ `" t  }4 S8 ]/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
/ s+ s+ @7 S3 D( `5 Q: |) Q
' K2 l! k8 O! P1 J& Q
#pragma once
+ w& a' M' t$ D8 R: `- H#pragma warning( disable: 4355 )

7 W" D5 J  p$ q2 V. r
#include "UPnPImpl.h"
" I$ p  L1 F& U. d0 t! c- ~#include <upnp.h>
' S  H! S4 Y" O/ a1 h#include <iphlpapi.h>
, H" T& j" f$ |! o- B& K& `#include <comdef.h>
: t' ^+ w" u- g2 G#include <winsvc.h>

* q' V+ ^1 @: g+ m% O% n
#include <vector>
1 l9 E5 e, ]! V/ H5 a#include <exception>
#include <functional>
: @. |* [. k: r7 Y2 G4 v


' i7 v+ O$ C7 m+ e
7 Y) [, [& _7 e; H$ ]typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
; s; p5 E9 u0 s' ltypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
' r# ^% P  I0 Ytypedef        _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;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
$ }% j& _# ^/ b( `9 L& ~

6 O4 ^3 G2 H4 m+ s# {) d2 o: X! @9 otypedef DWORD (WINAPI* TGetBestInterface) (
( I# g9 G+ u' O% D0 h- w( |  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


, d+ L7 i, H# Y$ z9 j% }  utypedef DWORD (WINAPI* TGetIpAddrTable) (
# p! D5 [5 ?- @% R% k; X  PMIB_IPADDRTABLE pIpAddrTable,
. u  O. ]! u6 G4 c; {% l( n  PULONG pdwSize,
4 e; U/ F2 @5 o$ ?- W$ {9 e  o  BOOL bOrder
) u# l8 x, n4 e: r$ j: Q+ h);
1 l# j" `2 o& [0 H8 v9 x$ c

/ A7 N% t( W$ g7 T8 c6 J4 W6 wtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
% ~+ B  _) j+ ?
3 J2 e! G0 p! [; x; c/ w" K2 l
$ _: e: I4 T1 O  ~1 ^" \4 BCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
/ p$ e$ z2 y; G3 K1 E. ~) q# f- N" G5 \

9 [: ~* {2 J! z8 \! K% aclass CUPnPImplWinServ: public CUPnPImpl
4 B' `  w& f2 \$ k* {' Q# ]{
        friend class CDeviceFinderCallback;
0 s3 G4 |5 f+ z* V' t, k        friend class CServiceCallback;
) M7 F: V) _. }8 h% w0 M, |2 n// Construction
( M  S. `* w: ^+ ?" Kpublic:
        virtual ~CUPnPImplWinServ();
9 M& H7 V1 _. ^) l        CUPnPImplWinServ();

. K& B- i) e" E; H
5 ?4 N2 h8 h0 z7 U. q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
0 l/ b2 c  K# W8 D$ o! T$ A        virtual bool        IsReady();
9 q% i7 T9 Q# N        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
' Q% a" [) p! F" b: h% w
+ L% f# [. L# p" o/ ?3 g' h; h5 h: g/ Q
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
- L7 }! x# O' G3 d, |        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
; q. V) H3 i9 K8 d: q$ h5 K8 C9 g        virtual bool        CheckAndRefresh()                                                                                { return false; };
0 J6 Q4 z/ Q* T

protected:
4 g/ c1 E$ Q+ A) l' A        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();
% V2 u% W$ d6 S+ |. S9 w) U; E        void        Init();

) t' t! Y; |; r6 `+ C* B! e
8 e" K' Q4 u8 q& g6 B) H$ E        inline bool IsAsyncFindRunning()
! y8 H3 D" R/ x* s* K6 |6 V2 h        {
. m7 o3 `  a" C7 {% j                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
  X5 Z6 G% }% n, Z& M" x' a                        TranslateMessage( &msg );
! V" a9 D: ^  v3 C                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
: ]' S$ Q' Y+ K+ R  s9 d  g        }
4 ^/ V/ _- }, W. v) B/ }
* S$ x9 D. L& n. J; h5 u
9 x; r+ v7 B9 E0 b1 i        TRISTATE                        m_bUPnPDeviceConnected;


/ x* S  ^5 {( \, \8 D- O* b// Implementation
        // API functions
- y4 ^! @+ B9 V( Q        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
+ e6 Z, v1 D0 D9 s- a: ]        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
  E( \$ B2 n, W7 [* l9 |2 c        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);
! m" S: W. L& p4 Q  H

# s; ]2 v. ~' C- k: p7 {# H        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
" h* {* e, u: H  I3 I
4 q' W# q  u( t& D* I9 u6 h
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
3 @/ q9 a- J' T7 g5 d& [: k        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
. W8 }& q$ J5 C/ _' D9 }6 G" o* ]% b                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
; M! x% x$ {4 D* f) A$ b+ L

( ]2 e! Y9 C# |! F6 g0 a6 {                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

: v3 G7 w  S  D/ ~" O
: x1 O3 x7 S7 k) D3 \9 k                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
" s; d- S% t' v! B       
* E1 S# e: U3 t. P0 f* [        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
. B& P4 e4 m4 x$ X$ K, u' L% Y        void        StartPortMapping();
, G/ y, m9 X2 k! _7 g+ D9 m5 R  P* W        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
" t: y5 ?: p/ s! _4 b$ b! o        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
, d" N+ F: s% K' D3 y; x2 R                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


1 \* Z: w7 z0 W# `  ]        // Utility functions
; Z8 v9 v  p- C: S2 a/ z& I        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
3 B- c1 e' Y# R4 x. L3 Q+ V        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
& H  T& E9 q7 r4 S4 S  ~" e        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
) v( I1 A. M9 w! R. g' P        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
" Y' d" F1 L4 B; K! v        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
0 N& R, J6 j9 D1 |7 `        CString        GetLocalRoutableIP(ServicePointer pService);


6 d* U) Z0 Z- Q3 \1 V// Private members
private:
- R+ p, G" b: x/ M        DWORD        m_tLastEvent;        // When the last event was received?
" d2 B7 V6 k, t        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
$ ^  n- ]: F5 ?  b" f- h        FinderPointer                        m_pDeviceFinder;
5 [6 e2 u3 {- u1 q) z0 l% e& e        DeviceFinderCallback        m_pDeviceFinderCallback;
/ m( y' R$ ~. Z) \        ServiceCallback                        m_pServiceCallback;
! x) D! `; m& A

& `! b6 U# H. ?" u" k5 `- Q        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
' C- L: R$ I9 A( _; K3 V$ ^        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
# P' {4 D  H5 F* \; N        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
( ~! B2 z; F. E1 ]        bool        m_bSecondTry;
- W: c# @1 d! P; D7 }/ P        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
& U/ k8 L) q& B+ c  c: e* H: Z% w4 K        bool        m_bDisableWANPPPSetup;
6 K/ `6 e  v8 g3 t

};
+ G  _3 o* s3 T0 c, e7 r
4 g+ D* z9 f7 s- Y
// DeviceFinder Callback
8 w$ z* X* B. V+ ^& Dclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
4 E4 G3 I: w6 ~1 d* S        { m_lRefCount = 0; }

, c6 Q! @) G4 }
1 n& `; w- Y8 `4 ^! ?! q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# F6 A9 q  K; J   STDMETHODIMP_(ULONG) AddRef();
) }' [' m5 B* ?; M8 E& L   STDMETHODIMP_(ULONG) Release();


// implementation
$ a- Z# b! ~  k) P5 n, F+ n- u% @private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
0 a6 ]6 O4 b0 y3 ~& e        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' T  C% I8 j+ X$ E; e        HRESULT __stdcall SearchComplete(LONG nFindData);
8 r2 P" u  U2 c

' s& w( R# }2 @1 O$ xprivate:
' F4 n1 W9 x2 e0 _- s0 O        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
1 m; j, ~( l  F};
% c+ Y0 a: u$ K
' N2 y+ }3 ^" }2 y  t2 k: k2 s
0 a# o) H" S" e  v. q& R// Service Callback
class CServiceCallback
3 v  l# u6 C/ B! g, R# R        : public IUPnPServiceCallback
: k( e8 B7 I, ^; ~{
public:
6 z' z9 e8 j. R% ^6 j/ h3 k        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
! ]) w3 b9 {. l        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
% e5 L5 U* j- c   STDMETHODIMP_(ULONG) AddRef();
1 L# C+ _& h) X8 ~: ]; n+ z   STDMETHODIMP_(ULONG) Release();

& M1 E; Z' |/ }# m
// implementation
+ ~! l, {" n2 K+ `private:
- E  D5 X4 v7 g' ~9 Q7 Q        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
2 B) T# T* J6 E        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
& b0 l/ w: }# n

private:
% u7 I3 G, A" W( Q  W4 ^; r        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


3 J+ _7 I0 j' }3 m1 W! ~1 H/ }4 i/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
3 \! D( D; k9 I* Q, j1 t* T# {7 aCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
8 X3 p* [3 R  w& F, H, K# H! ICUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
4 c. t/ `3 Q3 S/ {, RCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.