UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   3 e$ P( w2 J' j7 R) H
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    3 X5 Z0 f$ p7 M3 K6 U" a+ v7 [: }
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ) n9 y/ o& C" |8 \+ I+ D
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    : g6 S7 M2 L) f
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
    % f3 C: d3 J4 z
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    # s6 r2 g0 J' Y; }
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    5 C1 ^8 H" S& }% ]5 Y. r7 \/ I
21. UNAT_UDP //   UDP   Protocol
    + P% i1 r$ |) @' |; L" X
22. }   UPNPNAT_PROTOCOL;
    3 L6 `2 c7 {2 j  O) e4 G+ |5 |
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ) S+ b  r9 k6 ?$ Y) J: v
26. WORD   externalPort; //   Port   mapping   external   port
    ' Q* O  ~! [- ~$ r) d* ]6 ^* T8 R
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    9 R9 E. u  X# |. {* R
29. }   UPNPNAT_MAPPING;
    1 O* M- J- x( k
30. 
    
31. MyUPnP();
    * v" T4 {' \6 N" P
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    $ v0 G: k( o7 [  I1 o! m) {
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    8 d2 H4 C6 b& M, W
40. WORD GetLocalIP();
    0 X+ z9 R7 u' c1 n( O: y1 y; X2 Z
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    6 N9 O1 h$ O$ ?  r/ h! X
44. void InitLocalIP();
    1 S) y. e( B7 |! S
45. void SetLastError(CString   error);
    
46. 
    
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);
    # R9 W. d" @2 J9 x' J' B% ~* h5 y7 H
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    " e) v$ j" Q+ E: K+ x* W8 E
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! q/ W9 X. V$ p* m8 Y7 ~
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    3 p" r- `  s" r% N0 e  R( ^
57. 
    1 z2 d8 ?( K7 F- Y& q6 C- k- O; B6 @
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    + t" X& w; B- \  ^$ u
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    . p4 P0 v; a; Q4 Y; Y& e$ }
61. CString m_name;
    
62. CString m_description;
    4 w) p$ x  k* D4 e( H8 r; r
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    1 |) T2 z9 D( H: |3 b
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    + z2 G: i/ N# X6 X6 a* y+ l# a/ g
75. 
      H. w9 I/ t% H; O: G. I, n
76. bool isSearched;
    ; S, ^. q. H9 G; n) _# d- G
77. };
    / \! z8 l1 x4 B' T. l) t1 l
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   9 V: r! e$ T' }* P
2. #include   "stdafx.h "
   3 s' Y- k, F" R2 i
3. 
   3 u" Y& h& Z4 G
4. #include   "upnp.h "
   & W. c4 b# [. C  O$ {& @
5. 
   - j* k8 m+ k( \8 k6 @7 [
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   2 X% B2 ?  Y) w2 a
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    . @9 l! u/ f# g# `  c
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    4 b1 A: {  d$ Y) I/ C% Y" Z6 h
13. static   const   int UPNPPORT   =   1900;
    2 W- }# i2 E# N- T# O3 z, ?
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    $ W, @7 E$ y+ a  c
15. 
    
16. const   CString   getString(int   i)
    . y* m# y4 f: R/ N, A6 `4 \: N
17. {
    1 j/ U3 E$ i2 f- m. u
18. CString   s;
    ' q9 a5 r0 ~) F5 J, R1 t5 Z( b* J
19. 
    2 Q1 [  r3 d; F' ~; f! `
20. s.Format(_T( "%d "),   i);
    ! T2 D7 U! f. t2 f" N
21. 
    # O$ b6 ~; v! u3 d6 n: K1 A
22. return   s;
    # V! V, @, Y3 ?% ?
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    . q% \0 V! J+ {" M, J
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % S* I8 x$ i: g4 a2 P
28. }
    
29. 
    4 r1 N3 o; l$ {
30. const   CString   GetArgString(const   CString&   name,   int   value)
    0 ?0 L/ Z4 Y( K5 _
31. {
    9 V8 I% R$ U9 Q& [# U. ]9 M
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    9 N9 w& x6 w! a
34. 
    # ~4 m, x! f( d' X3 M( F+ m8 N
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    . m" v4 L4 t5 f* e. }- e
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    . {, ~* c3 L/ i1 m  ~5 S
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    / v$ u8 }5 x* s% M
43. uint32   ip   =   inet_addr(CStringA(addr));
    7 B  b3 A- J# L, N7 v
44. struct   sockaddr_in   sockaddr;
    0 N6 f" Y1 L" ?: R$ A
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 ?+ D# J0 Z/ l
46. sockaddr.sin_family   =   AF_INET;
    ) C3 c2 f$ }8 X, l/ o# W
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    " m( N' [  x1 L. g
51. ioctlsocket(s,   FIONBIO,   &lv);
    % |% \8 h1 J) L2 T6 x5 ]0 l
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    - J2 B( Y' J9 o8 I8 ]5 a: F. @& ?
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    , {4 ]% }( j0 o7 p! S3 U9 S" u
55. Sleep(100);
    & [! F2 w4 l6 i4 B; Y
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    1 z; t% M( j/ n; B+ ^0 M% B, A
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    " h& U, ]7 d1 I+ u: x
64. }
    
65. 
      L& b( d, p0 L+ L7 s, ~6 t
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    2 H0 y7 `0 @6 F. q3 Q6 i& R
67. {
    3 h; }) U' Z5 F8 T. W$ w' ^0 y. N) M
68. char   buffer[10240];
    ! n0 i- ?9 c! t  Q5 G& h
69. 
    
70. const   CStringA   sa(request);
    . b& F  F* }2 [0 u% _
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    # L0 `0 M, ]% V1 q8 W9 @3 ^$ R
73. 
    
74. struct   sockaddr_in   sockaddr;
    : z! X* m5 C* |" d) U
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    . p/ n& G0 x8 y# W% m  ^+ f
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 Q. b) _1 N/ y
79. 
    ' o0 l/ j9 x- [. v
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + M7 t# s) u: L7 n6 F& g- G5 G* R
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    + L1 w# f7 ]  j
84. {
    $ m/ U2 j; O1 u  r* w- e, a
85. int   pos   =   0;
    
86. 
    & Y( {; ^. S/ a! {( i9 I
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    5 n% w: F9 z$ y
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    9 |' q4 g6 g  t( l- O# e" j% }6 e2 G# D
94. status   =   status.Tokenize(_T( "   "),   pos);
    8 o' N2 S. n/ [  H/ ?" V
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    & {/ I" [/ ^3 Z0 c7 g' w- t
96. return   true;
    
97. }
    
98. 
    & q8 y- K& Z* y) {/ G' }
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ( R+ y; d& C5 @; i, W
100. {
     9 F) D2 d) \3 M
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ; ]! D, {2 n" Z: K
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ! Y5 t' t: V+ ^2 Q4 b
103. CString   property;
     " m8 N( [/ M) \2 t+ I5 _* r
104. 
     # I! v2 i2 K2 u
105. int   posStart   =   all.Find(startTag);
     7 V8 E+ I$ s/ H
106. if   (posStart <0)   return   CString();
     8 O* a% n2 ]9 U; R% j# h' N
107. 
     * |2 R( \6 M/ F  X
108. int   posEnd   =   all.Find(endTag,   posStart);
     ; m! e2 m0 M+ v' A4 e; ~% v0 R. S
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     ( C5 S% w5 R- B% J$ f2 J
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     2 _& B* F; R* i0 E+ E' R
120. 
     ; x' V: e% n$ H' D) R
121. MyUPnP::~MyUPnP()
     1 {: g% {3 k' n) x: `
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     5 u  F. g  S6 Q) I. Q
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     0 L5 o5 _1 t# w7 N" I$ `; m
129. 
     
130. m_Mappings.RemoveAll();
     ( x8 z- m+ B6 I1 N* l
131. }
     * m  k6 L! {6 z. j3 I. `" @8 v
132. 
     
133. 
     6 n& F4 {: M& O2 G# ?- p7 b
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     . U: L! ^* C1 R: j' M" l. F( @
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     . U$ y3 R* r( F) ?
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     7 J- P9 ^4 ]' o# I7 W. i
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ) X# P% R* J6 e; S4 Z
152. if   (!(i%100))   {
     ' [6 |0 P) S" B4 j
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     : a, d/ t7 d' V- H" A6 p
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);
     5 R+ }0 E6 S. _+ k, f
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     0 N; a' N& H, {2 f
162. Sleep(10);
     
163. 
     . b3 ~5 a4 @$ ^- R9 s
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ) @( M7 |( L% L
167. closesocket(s);
     
168. 
     9 `+ E! P$ o5 X! o. M. d' l
169. CString   response   =   CString(CStringA(buffer,   rlen));
     6 l: e1 D+ m6 Q  K, `0 f' r
170. CString   result;
     0 n# ^4 F$ Z- F5 c# v9 @, S
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . u' i" a7 T% C; B+ W& ~
172. 
     + m/ A! e8 o' f. ?2 ^) L: o4 w6 ]
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);
     7 O( {6 ~1 S3 c0 B& I: y2 a5 H, ~
175. if   (result.Find(m_name)   > =   0)   {
     + _8 f+ J% ~# b7 b% M
176. for   (int   pos   =   0;;)   {
     - M: D/ Z6 M# m# f. J( o; V
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     / X9 Q. Y( m# [; Q: O" n
178. if   (line.IsEmpty())   return   false;
     $ n2 S% G; M( ]0 v! F: c  \
179. CString   name   =   line.Mid(0,   9);
     + F: \- @+ n6 q( v. n' J2 K
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
       h6 S$ k$ I3 o- r
182. line.Delete(0,   9);
     : j0 S5 ]* Z# J: @, X* Z4 o; Q
183. m_description   =   line;
     7 J+ V9 C* o/ o( f( t
184. m_description.Trim();
     0 l2 d" p% r) L; v
185. return   GetDescription();
     " s2 ?4 P$ K" `3 f% u4 E
186. }
     
187. }
     2 w. r# G4 y, u* ~+ Y9 ~1 G1 O) Z/ R
188. }
     
189. }
     " Q( I1 t  g0 |9 v$ `5 j( ?
190. }
     , h( l0 c6 A( ~2 R6 u
191. closesocket(s);
     5 a* E" q: B! h# K
192. 
     # u+ c" W" V2 e/ o- s+ [6 y
193. return   false;
     & \, P/ N1 F) E% W7 E
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
+ A9 u, i# G0 \! F% W8 |) ]
# h' @! Y" ~# T, N2 C2 o* ^; V
; A) l7 h, p8 {0 H% J' j///////////////////////////////////////////
) b# o' j  L# f# \//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 M7 z3 ?. s, w( y- d" r4 o
/ e. _) j' x/ F0 p! z6 _
+ J' K1 K4 B/ ^" ]#pragma once
#include <exception>
2 `& F0 E3 @; Q- C- a4 j- |: a( t
* O  V5 ]1 ?4 _" u* k
- y6 m" r+ z" m; Z  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
. F6 V/ }% y/ ^, i- Y+ P2 F. Y4 S};

! I! z6 \: s) p, g0 W
" v2 x" e1 j/ f* Y9 ^9 W; M, ?% t6 W) _enum UPNP_IMPLEMENTATION{
4 w) F- b$ M, j+ }        UPNP_IMPL_WINDOWSERVICE = 0,
, ~4 D9 e1 R  g+ ~, o3 v5 a) r        UPNP_IMPL_MINIUPNPLIB,
6 j  K6 F4 S3 n3 e: h8 `        UPNP_IMPL_NONE /*last*/
+ B% i) A" K1 C( y/ j8 X};
. o9 Y0 c9 s" T3 i& v& Y- l



class CUPnPImpl
{
1 T* S' A0 q2 }2 Cpublic:
( T; r/ [( M1 h" r        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
  a# E; C8 e$ M! [2 }; D8 Q                UPNP_OK,
1 V1 K1 u2 _& G                UPNP_FAILED,
                UPNP_TIMEOUT
) i' ^) |! _2 e# ^  I* ?' x        };

5 k( Q- A. W# f/ G3 |& e9 f
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
0 Q& W: ]/ i+ A- e2 ^+ y- o1 `' i        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

6 `, r! r8 ]# m9 e2 J' j
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
4 W6 m; l$ R0 G% P        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* U2 x5 x! A( u. J5 `2 a# Y        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
9 c: Q1 ^8 y$ Z9 C        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
3 Z6 s- c3 M$ W        bool                                m_bCheckAndRefresh;

& Q* S; }& Y/ u7 k6 V( X
( V2 _  x* F  D/ L& \& ^# P  jprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

3 Q) Q, L# C# m( I/ w( M
, n: A- Y8 k2 _};


0 e- X* F& B9 @, @  u// Dummy Implementation to be used when no other implementation is available
8 d/ h3 d5 [/ @* z5 Eclass CUPnPImplNone: public CUPnPImpl
{
, Z$ ?  k/ a  P2 t4 v6 Npublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
8 V$ r, Y7 p7 E9 `+ U4 Z% M  @7 ?# |        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
8 [, N$ \9 `4 w+ b  S        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


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

3 v. R1 R) P/ ^0 n4 b
8 }7 w0 L$ H2 m) L+ R/ b#pragma once
#pragma warning( disable: 4355 )

( ~" J3 p7 R2 F3 n$ ~0 s
* }" V8 T! V- d! B#include "UPnPImpl.h"
4 ^. B0 a( O, w  N#include <upnp.h>
5 Z* T0 |/ C3 k$ p: f#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
: S$ B0 c5 h. k" q( b6 z: B
& K: W3 @/ ~8 g; [( I/ A
#include <vector>
  K- h* E& j( T0 m8 @4 r#include <exception>
3 ^1 O5 B- b2 m* `+ a#include <functional>

" l9 R& C4 Y( }% ^
& t% g6 l3 N0 ?% W4 ]- {6 X

  N8 I2 A$ _+ m/ `% Ntypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
. F, _0 X& b1 u! W- O7 ctypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
5 Z* \1 b; s' n9 q% y1 Z1 i1 itypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
: v- m! @/ G6 ~! b; ^/ T9 E2 Q' r8 xtypedef        _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;

3 K1 s1 {# R& r- }, C0 w
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
, U( m* I2 j# A4 v; s, K8 J0 W
. o6 C1 M( a+ z
5 N: P; M7 k" e; @  ktypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
6 E/ S; g7 c& D2 u5 W% h' z. y  PULONG pdwSize,
  BOOL bOrder
);
% ^* [9 {/ q4 Y" {: a" O
. r/ S+ ^& k+ L; D9 `1 D/ a4 e1 ~9 W
1 B+ R3 Y: u8 E- c% s! Wtypedef DWORD (WINAPI* TGetIfEntry) (
  @) T6 Y  W5 c% s  PMIB_IFROW pIfRow
);

, z) t1 U1 _0 B
CString translateUPnPResult(HRESULT hr);
; B+ n% A1 B0 K7 _! }: _* N. s" ?HRESULT UPnPMessage(HRESULT hr);


) U5 E7 k9 I8 f6 b) O% hclass CUPnPImplWinServ: public CUPnPImpl
  A/ ?, W% \+ a8 a3 E& e7 y{
        friend class CDeviceFinderCallback;
- |. J8 b. O  W" H) L        friend class CServiceCallback;
# c- v3 q4 \* \6 ?0 k# Z) E// Construction
' z; c( t1 D9 G0 k, X2 ]5 Xpublic:
* i/ C! N5 W7 K- m% Z3 c+ i- s3 T        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
: b) v: G$ \8 z0 U
- q, _6 ]! K. U
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
1 J% w( T# T# G& {+ d' n* d        virtual void        DeletePorts();
( ~% f; I# h% s1 ~8 z5 m        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
/ @! X9 l5 C4 W) l8 V" D! Y2 _
5 \  f$ Z% |5 L/ B3 {
$ ?$ G4 u8 n6 O) S/ D- ^        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
% z4 H9 x& m/ p5 X4 ~        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
9 N5 T+ k/ ^: J; m+ h9 x: l+ C

# ?  R2 P- ?% p* i; f# ?protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
1 e- x7 m! q' L& o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
5 \: D- r# c8 u5 j- M        void        RemoveDevice(CComBSTR bsUDN);
1 q: O2 n+ }1 R4 H2 Y  P$ M/ [/ m$ D        bool        OnSearchComplete();
* a% j2 _4 `. O        void        Init();

4 W2 ?5 K1 ]3 G0 g# z
        inline bool IsAsyncFindRunning()
7 ~  Y! C  b* K7 D9 X8 p        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
) g* a1 J2 \# X2 D                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
# o7 I* z) r# a: i2 Y! b                }
- _% G, m9 s! |/ p                MSG msg;
7 g3 v: Z% b  V: z# q4 g                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
4 M0 I" H. O1 U8 k: v8 r1 F* y                {
, Y. f# e4 P0 A5 U/ `* {; [: A* _                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
. Y3 w5 r7 v* ~1 I; b                return m_bAsyncFindRunning;
        }

% f6 w9 g6 i" ~+ U7 M& h- q
9 [% P1 V; y, m. x: t( ^0 G4 ^        TRISTATE                        m_bUPnPDeviceConnected;
; g! Q5 \; n& y4 x$ r
4 i8 I: J2 a) J/ s: K' a1 f2 U; b
1 l/ }+ e; W# g: x  r* t3 H// Implementation
        // API functions
& S$ l7 p8 K/ k& |) m: A: q        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
+ B$ J2 {2 Y3 s9 N0 F* z        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
3 c' t1 U+ _3 M4 t        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*);
6 W6 l* r3 V5 y$ l$ U! N7 j        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 W( K4 J- y. S! Z0 C1 c

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

( P7 X4 P. ~0 b
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
, Y3 s( R$ d# T- C                {
- ?3 [1 a; Z, Q4 q# [* b                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
+ V9 _* M; G5 L! Q/ e8 [; `
, V* a- @3 J1 d" g" \% ~2 F! Z2 ]
) a' m! F9 M/ a. V: ^3 C                        if ( FAILED( hr ) )
" R& ^0 F: ?, ^" l- G/ N3 B                                return UPnPMessage( hr ), false;
1 Y- K% K9 p, |. N9 `2 n8 Q

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
* O) N' O  @. q/ U+ G4 n, O1 X                }
                CComBSTR m_udn;
% z* p) T  D9 Q. W7 r: n& o        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
  `& r& ?. K% d        HRESULT        MapPort(const ServicePointer& service);
( z6 W. Q" i- t7 o: \- h( d        void        DeleteExistingPortMappings(ServicePointer pService);
) X( s3 t1 K1 o& G        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
% {4 ?  \, c- P* S( B                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
+ J: W- V; j; b( u

        // Utility functions
, O" p4 z( v4 T. @/ |2 w; m8 w( Z4 t        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);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
" w- R/ ~8 a5 X        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
( W) f. T/ ^5 k4 ]& o9 Q, u% y4 g        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
8 C9 L" O. s: u( h$ P$ c' S        CString        GetLocalRoutableIP(ServicePointer pService);

: Q2 n, t( _% u
// Private members
" q- K: X2 _6 G, f& kprivate:
5 H# r1 P, e3 r( q6 N3 l9 I4 W1 L        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
0 j5 [/ A, {8 p( {* |! p9 U( L        DeviceFinderCallback        m_pDeviceFinderCallback;
( M) x! T) Y: S6 a( N+ }0 c. U$ p        ServiceCallback                        m_pServiceCallback;
  l  |% E8 h: d9 z

, Q9 ?: u% L6 J        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
% k! ]5 e& r8 Z# h$ ]1 W6 A( P5 I        bool        m_bPortIsFree;
# z  t) b, ]+ X" Q4 K        CString m_sLocalIP;
, O0 r6 V: _: a1 L! Q        CString m_sExternalIP;
* ~) a* w" G- y9 p        bool        m_bADSL;                // Is the device ADSL?
! d2 G+ h7 E6 ^' p$ x5 ~        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
2 b$ ~/ _7 O, u. {. q4 o        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
+ J% l9 _( M$ C/ @2 k2 ~1 `6 T0 ]        HMODULE        m_hIPHLPAPI_DLL;
4 L* `! B' t" X6 ]0 h        bool        m_bSecondTry;
. c& I6 _& q0 N: W        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
' C2 n2 |7 P: E/ M/ q, e; E        bool        m_bDisableWANPPPSetup;


};
9 A. _% d' E5 j1 v/ ?
5 K  z/ \- ?5 {; ]# H- [5 I
// DeviceFinder Callback
- B& M4 R  C1 ?$ g9 ^% ]# ~: J' [class CDeviceFinderCallback
* {$ H1 F9 y* V* ]# L        : public IUPnPDeviceFinderCallback
! M) V; E3 f, u! O{
) a5 c% G3 G2 ^; k: l4 @- Lpublic:
: j; h2 r5 z: A        CDeviceFinderCallback(CUPnPImplWinServ& instance)
  y5 X/ j; k3 P- l                : m_instance( instance )
        { m_lRefCount = 0; }
2 p# n# L5 Z# x9 L) v/ x
6 U" n- r; Q5 a# k. s
" \: J% _+ r( g5 V6 L3 [2 P0 }( t   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


+ Z2 Z8 G& b5 M. H4 Z, n& w% p// implementation
1 X5 t: q. S3 S& T2 H% _# rprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
% n- D. _! q5 m        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
1 G8 m& k" w7 E% p0 W9 h        HRESULT __stdcall SearchComplete(LONG nFindData);

  v, b8 v' T/ }/ y  E* g( M
: A2 w, P7 S. o, ~0 n+ vprivate:
, i- X' A7 `# `  a( D        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 B% x  c3 C0 z2 w) n; U* E( ]};
5 g3 \( E7 q2 E& D' D" g( B

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
# |( {- R3 A6 `public:
        CServiceCallback(CUPnPImplWinServ& instance)
' U/ p6 C; u7 ]9 W$ y                : m_instance( instance )
        { m_lRefCount = 0; }
1 f: u1 D: L! c   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
* i1 S% o, h% [/ j& C   STDMETHODIMP_(ULONG) AddRef();
' m+ W  V& q* @7 Q9 Z; x  d1 B   STDMETHODIMP_(ULONG) Release();
1 ~! D+ Q( m5 I# ]" f! b

// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
8 l' Y; {9 m/ S( Q% V        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

- E. Q' d! T/ L# b- G- Y
private:
' P4 I" f: i% V        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
0 E3 P6 k4 _: j( Q% Q};


/////////////////////////////////////////////////

' M  n; I7 @5 B
0 \9 p- k( G( v) l$ e3 n使用时只需要使用抽象类的接口。
' ?# m* D: K; E$ Q* }6 ZCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
& r7 z  B1 n, |3 G$ H" f* ~CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.