UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   8 p! \' m3 V$ I7 z3 V$ m$ g, v  q
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   / O* Z# ^2 b$ k' b: J( U
9. {
   - d- R. e! |3 _0 ~' i
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ; K/ @+ f+ t3 [: b; b0 e5 C/ [
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
    ; j0 _6 \$ p! a3 P3 G5 _) C8 u6 ?
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;
    
18. 
    ! w% N+ F1 q5 l, O
19. typedef   enum{
    ( i5 K, d/ C+ l, e5 }7 M
20. UNAT_TCP, //   TCP   Protocol
    0 P3 I- s0 [- H. r6 W- s0 G
21. UNAT_UDP //   UDP   Protocol
    2 p* S1 N- N1 [+ u/ D5 {" X
22. }   UPNPNAT_PROTOCOL;
    4 ^- q0 n. e/ y' j* f
23. 
    ) f- `: q( E& d7 c) f3 V
24. typedef   struct{
    6 T1 v! m- R  C% o: p6 L8 C
25. WORD   internalPort; //   Port   mapping   internal   port
    % D% Y" a7 |6 f
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    % H1 b2 X  S3 p! q8 U* B
28. CString   description; //   Port   mapping   description
    # F2 P. D/ P: Y
29. }   UPNPNAT_MAPPING;
    5 N1 C4 O" N  Q* w% M
30. 
    " l4 v$ x; w  I  _0 k
31. MyUPnP();
    & o* `4 J+ Y9 r& ^* t% P: V
32. ~MyUPnP();
    
33. 
    - X3 b: E0 l0 b4 M
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    8 g# C$ H& K( X
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    % s' w/ u/ M6 G0 v2 A: R" W
36. void   clearNATPortMapping();
    
37. 
    ; B" X! N5 h6 E* ]0 X0 o
38. CString GetLastError();
    / r) w; v2 N/ r  d
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    6 A/ N7 T1 x; I& b4 o* o# d* q
41. bool IsLANIP(WORD   nIP);
    * {1 U( X1 o' x5 y4 E2 ]
42. 
    * g: p5 T' b8 k2 D
43. protected:
    
44. void InitLocalIP();
    0 q/ J5 z; d7 J4 z
45. void SetLastError(CString   error);
    
46. 
    3 B* M# n9 p! P( p, _8 t- H. y3 g
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
      b7 {) B  t5 ~' g9 Z' Z" V9 W
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ! n6 W% l" u/ Q
52. 
    / ~3 n- j- G# m, s
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    - P( |8 S2 D) R4 g  ~
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    2 S0 d  p# v% u; I! L
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
      Z( }5 x# b( g) g3 X, t4 Y1 B
61. CString m_name;
    
62. CString m_description;
    % Z# ~4 H  [! y3 A
63. CString m_baseurl;
    
64. CString m_controlurl;
    8 K! ?9 f5 l2 ~5 M  _
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    1 g, M" |4 I# \- _- v8 Q
72. CString m_slocalIP;
    * N+ F$ x; W0 r% `1 `8 @. G% Y% ^
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    % n" t& y1 S+ |& C* R. W
77. };
    & Q) k2 R! E0 R- X$ w$ [8 n6 b
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   . _+ }- I/ |$ Y! m6 @% ]8 U! X3 l
2. #include   "stdafx.h "
   
3. 
   " s, A2 J  h6 S" p4 f/ ~' C
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   9 L3 l2 {1 [: Q3 b$ Y0 k2 E
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   & T( f/ B0 p( ^/ y
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    6 J- _! W7 Y  z" v& J5 P) z% ?3 |
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    ( `: a, |# @2 z  w9 k& c+ L
17. {
    - x, @8 J+ y$ s8 T' u9 g
18. CString   s;
    & q% \6 u7 c6 y3 ]
19. 
    9 e0 {. n7 I, E' g+ S6 v% z
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    + Z, v8 w, T( g3 q& o6 U. W
23. }
    8 j9 V3 i+ S% T2 ~' M
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    , _, ^5 y4 t! p% Y) s
26. {
    + @: u7 l3 _* q) b8 T* K( B8 x
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    0 M  S0 T5 d3 e% s2 W7 i
29. 
    ) Z  d* R, @9 p2 p- E% L' r* B
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ( v$ I6 Q6 k4 T+ K4 ]! @% S/ p
31. {
    3 B( n" [' `4 F
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    6 s1 P4 |. c3 t0 y, _$ H
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    + q7 ?$ S1 }1 g& ?% N& b# F
36. {
    
37. char   buffer[10240];
    - J' v7 o! T5 l5 a9 J0 M
38. 
    
39. const   CStringA   sa(request);
    ) \# B8 o$ R+ H3 F
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ( Y8 q5 y% V" a6 Y& J
43. uint32   ip   =   inet_addr(CStringA(addr));
    / p- h! z/ e. v
44. struct   sockaddr_in   sockaddr;
    ! T4 j7 o. J" `- H
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 E0 h& b; u0 n) U) d! ~
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    % _: D3 J2 A8 Y* o( k
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    3 W5 e% R% ]' B
50. u_long   lv   =   1;
    - q! n/ ?& Y# S" _) F
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    4 _; o& G, e) T, n
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ; `7 v2 E7 {2 `5 ^3 W8 \  j
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    2 A% l; Y# n8 D
59. if   (!rlen)   return   false;
    ' L) F4 q, F# ~/ N8 r, N9 H/ {% u- S
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    6 E4 c( T/ b0 @
62. 
    8 V; u% m8 K5 t" o, B( I2 S7 y
63. return   true;
    ( ~4 w: Z2 E  H; e- b: I* h# Y
64. }
    " q% O" u% Q: W6 V2 B7 [9 M7 b
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    5 Z, ^2 M# G" l7 `. @3 I$ B1 b
68. char   buffer[10240];
    
69. 
    ! [& W. f  l7 `* R! p  n
70. const   CStringA   sa(request);
      W3 ?; {* L* ^6 G' k4 \, S
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    9 R# h" F! t; h- n" ?7 B
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    7 ~6 R$ Q/ {# J: }
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    8 \% @  n' S( r' s$ _
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 i. j. O1 {! O' b
84. {
    - O5 v4 E$ J. P# S" w
85. int   pos   =   0;
    7 }3 I9 _( D+ ?' I; {# v3 @
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    % E$ r3 {: G5 ?% G( U; A, Q# M
88. 
    8 w( }( q  P$ u# X: {
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    2 J# Y% h5 F4 |) m$ d  `
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    7 @9 V+ v+ Q; ?' a& `2 V
96. return   true;
    * T2 B9 m) f5 X7 X2 T8 m
97. }
    * R: v. X! H! r9 Q
98. 
    3 z5 B$ M+ I( ]- k& m/ _6 K
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     ! Z8 z* K) q( Z
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     + l$ n  m3 f6 \- Y0 z" b
110. 
       ]% I9 Y5 \" o+ S0 s
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     * }/ r) }5 C7 i6 L) V0 G- B
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     * R8 |  e, v/ A
116. {
     * ^2 L( m& Y' k8 ^) u
117. m_uLocalIP   =   0;
     % }* V0 t9 z7 j. b1 J
118. isSearched   =   false;
     " W& E* c! P% N! G# c( p' C0 z8 s& Q
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ( f' T% _4 \' w# J
126. search   =   m_Mappings.GetNext(pos);
     7 b+ {" a, S" H3 a  {9 Q; J
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     ; X; |3 V7 f* G) O
131. }
     
132. 
     ; x/ ~* u$ K' a0 x$ c. f
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     1 {, w8 P* d% z) O
138. 
     2 y- u. Z0 U' }, X3 ?0 q
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     2 O' j9 {# f' ], ~/ x
142. {UPNPPORTMAP0,   _T( "service ")},
     8 K9 }& N5 }6 T/ e9 g5 s- o
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     6 Q1 }& T$ a" {2 g' X
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     8 Z' I8 }5 @+ J4 ^+ S. B
149. 
     
150. int   rlen   =   0;
     ! p, d, |% H/ ]7 P! ~
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
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);
     
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);
     : V9 ]; a* _" m: n( O
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     , i5 c1 C, @1 ^! m3 F! @
160. }
     . r) S" r7 |* M) i) n; G( x3 @
161. 
     5 J3 x; i& Q. M
162. Sleep(10);
     6 I2 u4 X* E  K+ H, I7 G
163. 
     
164. char   buffer[10240];
     . `) _5 X/ A! t; a4 O3 A: @- b
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ; v+ T/ n; v/ Q! K4 d( T
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( V6 D9 K2 j) a1 _- \) L4 b/ ~
170. CString   result;
     , S9 a- W3 v! X8 F" Y6 g0 d
171. if   (!parseHTTPResponse(response,   result))   return   false;
     4 s; l1 H- Z. O9 E# V
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)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     - C- l" n: d* e0 ^  n
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     " t" @7 O+ F2 o' E, y5 A6 Z& o
181. if   (name   ==   _T( "LOCATION: "))   {
     4 N- @) L% s* r6 o- M* h
182. line.Delete(0,   9);
     3 E/ q- U- n. ~8 V; `0 }" x
183. m_description   =   line;
     
184. m_description.Trim();
     / F2 J5 J* {. d
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     1 w- X- `7 c" |2 N$ r# n9 y
194. }
     1 z; @% y* E, I9 S! ^

复制代码

天云上人

以下有关upnp的接口来自emule，
4 ]1 F2 H) u) t& W  }) v( g
+ ]4 F3 S9 W- y& }; `
3 W+ ?3 z7 v2 Z" v' R! }4 d///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
7 d' W' F* \: E% m* e; s' q: H: g

" f. z# t3 {8 ~' @7 V* l5 Y#pragma once
3 c. I3 |; W, E#include <exception>


  enum TRISTATE{
0 D  L8 n2 U/ C* \2 T% M        TRIS_FALSE,
' w1 b! W2 K; W6 |+ D4 @        TRIS_UNKNOWN,
/ J, ~* t, _" R) v0 G+ a% C5 U4 I        TRIS_TRUE
};


/ @. f, L! ?* |enum UPNP_IMPLEMENTATION{
* N* h% Q6 u& ~! n0 e        UPNP_IMPL_WINDOWSERVICE = 0,
5 E9 F# e2 i9 {$ V& Z% U        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

7 N9 B( r9 J- |5 Y' C
" A5 {4 \# `7 l" t0 A; O: X
/ d1 M: g: U. v& B+ Y6 U( Y+ }
% Z7 Z+ X; q$ E+ jclass CUPnPImpl
' q2 u# ?5 x. C5 T5 U4 ]9 m{
public:
5 H, h! E- k5 _* Q3 ~( X" e- q        CUPnPImpl();
9 A0 Z+ X! S6 I        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
( |! ?. r; j( j) Z% O/ _        enum {
1 q1 A5 Y5 _$ B- n, D                UPNP_OK,
                UPNP_FAILED,
" R) a9 t$ N/ ?, v                UPNP_TIMEOUT
        };

2 @. C: S# q4 a" b+ s/ B6 G+ v2 @
3 v5 M8 D  W" t3 D. u6 D% Q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 J9 h: g7 t% y5 A) u        virtual bool        CheckAndRefresh() = 0;
% N( v# I8 ?+ l) ]: H3 t        virtual void        StopAsyncFind() = 0;
6 _! p, ~/ U1 e. j% w$ k8 r! s        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
0 v4 [: ~8 j! N: q9 T       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
3 V9 W3 W! [; P5 U, i( }3 t
$ I6 Z6 [$ U3 j; i+ X
% U0 {+ W8 ^, w5 e& m7 D& t, P$ ^        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 I$ o: R, O, ]6 L. v5 T5 M        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
- A9 W# }: H, B+ \% p; i
' i2 M& ~+ n; W9 _+ D8 r: {
// Implementation
protected:
# F% q- L+ `6 [' `* B        volatile TRISTATE        m_bUPnPPortsForwarded;
/ O: W* d+ _9 A' E        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
8 n8 q, L6 S% f3 O1 S  H        uint16                                m_nTCPWebPort;
* k) K' G3 ]2 \& ~) L        bool                                m_bCheckAndRefresh;
/ u1 B+ h, O* q7 }6 ^- M. [
- x3 a1 ^1 O( x  _. S% f- j
0 E" p) ?( R  M) k" Y% _$ c: Fprivate:
        HWND        m_hResultMessageWindow;
- L6 U8 P, U: ^        UINT        m_nResultMessageID;
; ?  u  V3 H% E# w

+ c+ {: q9 Y8 E. ?8 ?  P* G};

- x5 P0 V5 Y  j" V4 M: w
// Dummy Implementation to be used when no other implementation is available
) l7 R2 r4 E$ l- Q1 pclass CUPnPImplNone: public CUPnPImpl
7 O9 `) Q! Z' j8 d# {6 V7 s{
6 \0 s5 C  u! _4 ?' m  y7 @- q0 ^public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
+ p/ v; a$ Q  M2 W+ y6 m0 c# E! d        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
/ ?+ Z! b6 u0 r( l3 t9 ]        virtual bool        IsReady()                                                                                                { return false; }
  [. D4 [, q" R/ t, n2 i        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
! c3 m8 Y1 c1 ?8 s: M3 k
+ E  X+ i% h0 J7 Q6 i, v+ A7 m, |
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
/ g* M9 ]' _+ F4 w3 l, B
( \, `. w5 l2 @4 T) {) r9 @0 O
0 T! B1 O3 c  c# Y% q/ W#pragma once
#pragma warning( disable: 4355 )
% z8 c  g: o3 b( t: k& f1 f
- I' r. v7 w& g/ ?
' q- a# c% `! G' [/ K7 n#include "UPnPImpl.h"
7 C) R* Q! e* x" r$ v7 h& W#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


2 ~$ ^- z% {3 t0 g. o+ L#include <vector>
. R6 B! Z& v8 Q2 ~/ J& R#include <exception>
#include <functional>


3 _$ ^+ Y2 z7 q8 q, D! E" t8 `+ S

$ @0 _2 k" F0 D. Atypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
; u4 y% r' m) s' l* Z, f' z/ ^typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
3 ~# i: e# i6 K7 ?typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


/ N8 B  f; n" Z2 Z1 \6 ntypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
, Y0 `8 m$ A2 {0 @& v  PDWORD pdwBestIfIndex
1 C5 p) ~5 C# e" m2 I1 a4 {);

  S/ O0 V; {5 o
& P( m1 p. ~# F# P6 [4 Ytypedef DWORD (WINAPI* TGetIpAddrTable) (
' \$ T& ?. ~; J& M. N7 j  PMIB_IPADDRTABLE pIpAddrTable,
$ d! i9 K/ k3 B7 Q- N: Y& I1 L. Q8 h, h  PULONG pdwSize,
' e! ?& v% p: W2 W7 A8 y  BOOL bOrder
);

* K1 l6 F' X. e$ L7 Z
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


* V/ z1 Z: B2 i" nCString translateUPnPResult(HRESULT hr);
% s/ y0 x, \: X# d- J& nHRESULT UPnPMessage(HRESULT hr);


& S1 c; y1 ], h. a  C5 h" ^class CUPnPImplWinServ: public CUPnPImpl
  Z$ \5 [* l& m9 ?2 j{
        friend class CDeviceFinderCallback;
( T" ~+ }/ c2 ?$ D' g        friend class CServiceCallback;
  d, k2 P) m2 I% i// Construction
; Y* n  k- M- V8 T3 {public:
' o( Z" w4 i8 ^( E- l# |        virtual ~CUPnPImplWinServ();
9 S+ m& @- W, o; n: I        CUPnPImplWinServ();
, Y* Z4 e# c  @

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
& e  b  Y2 o5 F7 G! v        virtual void        DeletePorts();
0 w6 e! f: D$ F# y        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


/ w0 @% x* J5 K% d2 S& x        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
9 X3 l: @/ o" d& c" c4 S0 E        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
: p% }7 a9 `6 w9 W& G3 i        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
7 s% ~& U* \% f1 b- f/ `; m, ^  z. ^        void        Init();


        inline bool IsAsyncFindRunning()
! N4 F2 ?( i  o. a        {
/ R" C$ p/ S. e                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
, n+ c/ k2 v7 N1 J  K                {
5 y7 r; H% _/ c8 N" [& c                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
/ m* o6 ]$ ]" w+ P                }
! ?+ G0 N2 G4 c- }4 c. Y                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
2 t) P3 A7 x, R+ z# F3 G8 G                        DispatchMessage( &msg );
' a6 i& ^9 J2 O# }. [/ o  b5 `: _                }
1 \' G9 e/ `( s. i                return m_bAsyncFindRunning;
        }
4 i5 s3 V( E# G2 }6 J

        TRISTATE                        m_bUPnPDeviceConnected;
$ p  x& q; ]/ y' y1 R
5 V' b/ T+ ~# _) R5 E0 J
/ T+ _, @( ], Q// Implementation
3 N8 n) `. U& B5 F( W        // API functions
& L/ A$ T) O: d$ p        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
) D: X& F" x8 ?5 q        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 ]9 ~* z! I" a6 A; p; D        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
" B+ ]& H  b9 M. X        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

( p6 M% Y* B7 m& Z
        TGetBestInterface                m_pfGetBestInterface;
- C3 @( D$ j0 a& q9 b( Y$ E# @        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
8 u. C5 |0 `% q3 V, z, o7 v
0 }$ T) `" h: E& K
        static FinderPointer CreateFinderInstance();
$ W5 _( K6 N9 b+ R  x        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
6 D4 H, f- e( g                result_type operator()(argument_type device) const
) K! Q( K* U. ?6 f                {
                        CComBSTR deviceName;
: O' [4 {" M2 L# F$ e                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

5 A. H8 A/ N& v7 [% y( m
                        if ( FAILED( hr ) )
9 G4 M9 A) G6 u$ X2 j8 H, F" n                                return UPnPMessage( hr ), false;

/ P" [- S4 r4 j6 ]
) T) F. Y( _3 U                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
( ?2 [+ p3 B' N. y        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
2 l6 u: I$ I7 L. g  q; {# h' G        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
" s( J. `4 h1 ?6 n! ]" d+ y# ]0 t8 u        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
+ g$ E9 s6 R6 R% N: k1 e                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
, p) B; y! o& s! J" S# G1 \
8 F; ?) T+ B, b' ]* h* ?9 [
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
8 s  j1 M) C9 n" t        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
" S8 f  i4 v3 u) @: n* _. Q        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


* z- D. r5 i& M" I2 a// Private members
private:
( i, M) E  C. E1 p4 n& Y        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


% ?9 f. I  E9 |& R        LONG        m_nAsyncFindHandle;
7 |% n5 p3 u: L; V, v        bool        m_bCOM;
: W* ~8 Y8 `* Z/ p% s        bool        m_bPortIsFree;
% o3 l" ?. Y8 x1 q        CString m_sLocalIP;
        CString m_sExternalIP;
4 {( |! m7 X. r" z5 ?( W0 X        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
3 A# y0 z' [- N- C9 J: @        bool        m_bInited;
        bool        m_bAsyncFindRunning;
7 [) S; I5 U4 ]0 _% D4 t! f        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
" u' C  `  L. m8 L1 w/ ~' l7 w        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

2 c/ y7 Q1 A3 G+ J% @# s
$ ~% i" o; @# d! @) g  @};
  k# j7 q5 G) Q

// DeviceFinder Callback
class CDeviceFinderCallback
$ R% M5 N% P5 n0 C) L9 G        : public IUPnPDeviceFinderCallback
{
5 B, u5 k8 N0 T+ [9 u( Mpublic:
; L* n) D) u8 L4 y% j" R! `0 O$ I        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
6 Z1 l; `8 `9 C9 p1 o! m        { m_lRefCount = 0; }


" K* F6 C1 |- H! N   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
+ F$ b- F) W% ~   STDMETHODIMP_(ULONG) Release();

* d* W2 W2 c0 a9 g+ z
' W8 n, Y( r$ f  E; \- W// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
& c. B  V7 q5 C/ \( v5 D# G        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
2 |- `  C9 [  ^7 ]' U; T* l7 F        HRESULT __stdcall SearchComplete(LONG nFindData);

- N# ^) i2 Q! }7 r: C+ I. |
7 F( A+ [: ~# [' |4 d" g! Bprivate:
        CUPnPImplWinServ& m_instance;
( k# W2 r" U; f2 D        LONG m_lRefCount;
};
: X# n- N/ ?1 K0 }3 y1 [

// Service Callback
class CServiceCallback
% ~# d  P" m/ @        : public IUPnPServiceCallback
+ V7 M  ?; `$ ?4 [: e{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
) [  ]- l& f0 c                : m_instance( instance )
        { m_lRefCount = 0; }
; B* [. O4 O2 J6 \, D, p# o   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
' B" U4 L# L: V" K   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

0 K& i* G# h8 O* y: U7 k
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

4 N# o$ C. |" V" Z0 |9 z( ^( p
private:
# A3 l( R6 @# F/ v        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
1 K' a8 s; k: d6 U3 L};

8 [0 t9 k( G2 k9 i2 r
/////////////////////////////////////////////////

# C4 a# V$ w3 c. S. E- K' z
" ]( Y/ `) K6 Q5 K  ~# f1 [7 c使用时只需要使用抽象类的接口。
9 f  l9 t5 k0 h5 Q* ~% k# zCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
- S3 ]' [. M: G) i5 i. vCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.