UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   & @$ n5 J7 O8 t4 j' @
4. #pragma   once
   . x/ M, X6 g7 \
5. 
   - Z. C( V; v5 {& ~6 C/ d2 l
6. typedef   unsigned   long   ulong;
   ' f% L! R7 O% k2 Z3 O% J
7. 
     X2 J1 F# P% X0 K$ G5 N
8. class   MyUPnP
   0 d' Q7 J9 a3 U
9. {
   
10. public:
    
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    , M5 ]5 H, M( ]; ]
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    4 ^% Y4 B7 ^6 \8 f- P% z
17. }   UPNPNAT_RETURN;
    ! x, i# \; l- Z' g6 n$ e
18. 
    . C! M# ?9 J  r* v( {+ {8 C- l) M
19. typedef   enum{
    $ q0 Q9 e0 b+ {
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
      c4 ]0 @3 o2 ^
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    : k: p; \) U3 \
25. WORD   internalPort; //   Port   mapping   internal   port
      H4 a' o, c: u+ \7 |6 \
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    1 W- u9 j% y* x3 |
28. CString   description; //   Port   mapping   description
    1 r+ h& n) ]# \
29. }   UPNPNAT_MAPPING;
    " J; @; M6 h: x  X/ F7 E( N& s
30. 
    
31. MyUPnP();
    ( f8 x0 O$ Y' ^: x
32. ~MyUPnP();
    8 ]0 t' u8 y% v- E& L
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    6 H  \' q3 B: S' @$ z1 m
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    ! p4 x* ~3 c/ o+ D
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ' K! H8 _1 G* ~4 k8 G7 K
41. bool IsLANIP(WORD   nIP);
    
42. 
    * R/ Y$ s' v* ]8 ^1 C; g; O
43. protected:
    . J# h2 }7 o4 F) X4 Y6 k
44. void InitLocalIP();
    " Q- o* Z/ e3 ?2 O. q* r1 B
45. void SetLastError(CString   error);
    + y( R: K6 k( v) Z
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
      z6 G! k- F+ @/ c% f% Y( u+ Y
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    2 J7 ^) v7 [6 W! P3 b
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    3 E6 Q7 \6 Y4 G) r) F  t
53. bool Search(int   version=1);
    
54. bool GetDescription();
    * E# e& R& h9 ~  R4 m
55. CString GetProperty(const   CString&   name,   CString&   response);
    * Q5 h) I+ j8 F7 R8 W
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    $ b' l+ z+ f& A$ V- _# [
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    - i+ h( m) {' d( b( v0 J2 ~5 g  J6 n
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    / j7 w  a/ B6 w7 s
64. CString m_controlurl;
    ! P& N  J7 j) o! w  C1 d
65. CString m_friendlyname;
    0 m0 l7 R4 p1 L9 e# P- q' A
66. CString m_modelname;
    * ]1 s2 x% \1 V, h* _# T
67. int m_version;
    + r3 f: x/ u: e+ `7 J
68. 
    4 ?4 W4 p- @/ i: L0 q8 G% x7 O8 G
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    # o  o, V; t$ X5 o+ J- I1 {( X
75. 
    0 z4 D/ {/ T% S* a: z
76. bool isSearched;
    
77. };
    3 a1 y, a  n6 Z
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   9 P# O% r  ?  U5 o$ `
2. #include   "stdafx.h "
   4 ^7 ~$ b& h5 ^9 L5 P
3. 
   - L4 G4 a6 H0 c9 m+ ?8 f6 ^6 c% F
4. #include   "upnp.h "
   
5. 
     V, e# R" B. S8 b
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ! s& P( U# y5 i/ a- ~
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   . ~% X; n$ f, f. G& c) n
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    2 Q" C- b2 b4 |
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    # m1 p6 R4 u' K4 F! H
16. const   CString   getString(int   i)
    / X) b# g/ Y5 E( C- j/ f; j
17. {
    
18. CString   s;
    ! c% @! ?$ c0 N: V; e/ N0 o- _
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    $ w# y! h( S8 L2 Z
22. return   s;
    % a: k' R1 c& |- W% f. u7 z  t
23. }
    4 Z( K7 X4 P# {3 M+ ?
24. 
    ' F8 E% V; z, v, g5 H
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    - @* }' \& s8 @( @4 a
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; D! e6 O! ?$ K+ A8 L" o
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    + B8 e0 Q! W; n/ @
33. }
    4 U9 h  {) T4 Y$ W- z+ F8 m* B
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    - z, ~: H( X- |( _9 _! c
36. {
    
37. char   buffer[10240];
    
38. 
    # M- c# d8 m) J# n- G8 v
39. const   CStringA   sa(request);
    ; J' [: P5 {/ [* s* M2 \# i, w: c* t
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ; t9 r5 i. g# _" j$ R7 D
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    - |$ b3 }  t$ {  @2 Y
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    . k4 T( T# I2 g+ N  r
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ' a1 A, M$ W: R! E9 g7 p
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    * d' ?7 K2 D, T4 W; w" J5 y. e
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    % ~9 t9 ?/ q" T: |0 {+ X
55. Sleep(100);
    * t0 R2 W; r( x
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    ( x) ]6 h( c8 Z8 Y2 n3 O- [  Z
62. 
    3 r6 I; D$ }0 X: j: i( a
63. return   true;
    ' V! X& S( k5 j- }5 K
64. }
    ' L2 J( A3 a) X2 j* ?
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    ; l% n* X  ~% W& W# {5 c& }
69. 
    
70. const   CStringA   sa(request);
    # z1 E( T7 }. b2 `" b
71. int   length   =   sa.GetLength();
    5 A( M, M& n( O; P
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    2 T9 j6 S6 w' d: p' f
74. struct   sockaddr_in   sockaddr;
    4 a7 X/ c  h- \  k  ?
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    / O# u2 i& r+ w+ t  N
77. sockaddr.sin_port   =   htons(port);
    $ t3 v8 s9 y  u$ c  p
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    " N) u& c  w! A3 e
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    9 d' b# F/ k2 _: B. D
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    : d& ?: x/ [" C4 U
89. result   =   response;
    6 r( T' I- `$ d; }& b; I9 h
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    ) x6 q* W# C3 N. I
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    5 O! r* y# P% S4 u+ x1 X1 |
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    , g" J5 ]8 t3 R7 o( P) I! I: ^
96. return   true;
    4 I1 D$ k! B* M
97. }
    $ v8 d+ h, A( ?- m
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     $ J) h& Y% N4 {3 f4 ]
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     7 c& D+ s1 ]! M$ O9 t( M( W2 ~
103. CString   property;
     
104. 
     0 i  O5 Z( q' b" s( i! z) r' q6 A
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     - T# E7 |! |8 [: z) a  J$ |0 o
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     5 C7 U; e" z$ p! y2 p
109. if   (posStart> =posEnd)   return   CString();
     : }  P# c+ |) A1 B1 X  m  n
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     1 F* P+ ^/ P3 `  d1 L
114. MyUPnP::MyUPnP()
     / K% e. C1 B6 h0 r+ o0 E8 o4 Y
115. :   m_version(1)
     7 V/ E4 y( R8 W2 \* G) E
116. {
     : _7 y; A# x4 F8 M7 j
117. m_uLocalIP   =   0;
     # }. X, J" H0 Y
118. isSearched   =   false;
     # M2 q' F1 m- y
119. }
     
120. 
     & h& R) Q; ~/ g/ l' ^
121. MyUPnP::~MyUPnP()
     
122. {
       B+ j; Y1 l) }  h& j9 b; h
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
       k  Q" B0 L* f3 T% N6 }
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     2 \; D8 ~8 S3 i# V) H2 V, d" r
129. 
     
130. m_Mappings.RemoveAll();
       v9 d/ B- e/ V% u
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     - v5 _9 H5 v& c: a
135. {
     
136. if(version <=0)version   =   1;
       R8 ]6 L, _( E; l7 i  r) f- d
137. m_version   =   version;
     * J" X9 a2 a1 ]3 X
138. 
     
139. #define   NUMBEROFDEVICES 2
     * C* J+ H$ x8 r
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     ! f# @6 o0 P1 c, Y
142. {UPNPPORTMAP0,   _T( "service ")},
     ! @1 l9 d! j, V
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     + ?6 @# [( I* ~. u  b
144. };
     
145. 
     : X! g  [/ B6 a
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     2 y  ]# r0 u) i
148. ioctlsocket(s,   FIONBIO,   &lv);
     1 |, P/ q" s. [" U
149. 
     9 j- @" I9 m; {
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     7 S9 m0 d# V# x- \9 \" J
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ( [) t2 i) d' y0 ], U
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     ' P# P% k& z! l% X% E  z
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     + E  g" E8 M5 G: s/ j
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     ) b8 w: o( M1 F' p5 v9 I
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     " |* X" H: o; R! u" V7 U
166. if   (rlen   <=   0)   continue;
       p$ m5 `2 L* C2 W
167. closesocket(s);
     ! G" ~& @9 }1 l
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     ! v/ w* h7 U* o, j; O) _# ]
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ! c0 Z) _6 o  k
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     + v# s$ j) J9 u, p% H- N7 ^/ `( M0 _
179. CString   name   =   line.Mid(0,   9);
     ) A  G/ M+ f2 _$ V; C* d
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     $ C3 z& i" I- w0 m' k/ c& E( [: H
183. m_description   =   line;
     4 M9 i8 M4 }2 Y8 V" l6 s3 |
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     # Y' {. |% c6 m. ~
187. }
     & w& g3 X/ Q$ |# A/ v
188. }
     
189. }
     3 ~" A2 Q7 O* m, d( k2 F5 r
190. }
     
191. closesocket(s);
     , c* c+ w' A" Q: Z, i2 H
192. 
     
193. return   false;
       V3 R1 [& U& X& l% o
194. }
     

复制代码

天云上人

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

  i  e+ c! f0 U0 B; ~
///////////////////////////////////////////
* w; J, r% M( _/ c. x% z//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
. o# r' J' o* O3 u
9 N5 B* Q6 A" F( N! V' Y5 W
4 b, G0 o7 a; P6 D8 v& E#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
+ |( r) A! @- Z' j2 @& z4 c5 h        TRIS_UNKNOWN,
        TRIS_TRUE
3 S( V$ }" Z# o* O/ H};
& S% x/ g7 b& z) W
+ C' T; j" U- \
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
3 Q  H* L& [) W, F% {$ v$ s        UPNP_IMPL_MINIUPNPLIB,
# F8 j- y  b: g. g        UPNP_IMPL_NONE /*last*/
  M7 a  e- G& G& N0 Q; v+ Q};
0 F7 _9 ~7 l5 A7 b. j; K

# N( _, B2 W8 \) u& Y6 R( {* {
! Y$ K$ s* u2 P% F
+ ?* d; e, o. B4 Dclass CUPnPImpl
- U9 {4 Q4 c& y8 A( R+ V( B: E0 ^{
public:
        CUPnPImpl();
2 a  o( D; n, r5 c- Z/ B        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
$ |0 ]# a! h) ?9 U* g! e/ T                UPNP_FAILED,
6 E1 l# }6 E+ s: ^0 B                UPNP_TIMEOUT
        };
! {# z* B) }, [& D! }& O( {( _0 x
- j# @' s: ?! _. R9 I) F
  f5 [7 Z' O( S% s/ u" k7 Q        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
0 _* w) f1 L; P, S0 q3 c, r        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
/ @4 o+ G7 B: O+ C2 F        virtual void        DeletePorts() = 0;
4 b4 q4 g( Y2 y$ H  g) d$ g% B        virtual bool        IsReady() = 0;
7 U* E" [, t0 y5 q        virtual int                GetImplementationID() = 0;
7 A- q6 P" j. u5 ~' _+ Y# {6 W       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
$ X; D0 e. H% O9 R& s

/ e9 L( L5 A2 m6 n# P' C% P6 Y        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 T8 U; |7 W2 Q: }" W5 q! e        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
+ W/ [+ K# X$ @        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
! J/ |' x+ k, [5 i- x2 @4 J% j5 wprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
- p. {1 _- j6 g. k7 p/ A- O        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
5 ^, P5 R$ s' i; `- E        uint16                                m_nTCPWebPort;
& q8 p) r3 q# [1 W) I: i8 ]        bool                                m_bCheckAndRefresh;

9 P, U6 m" |$ q% Z# x' R  f! U. C
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
% z  N! T( ^3 c6 F1 b
+ K. I# w! l9 a! p3 n+ P1 G
& j# v, k+ Y. ~/ Y. [};
! Q! ~" Y7 X! Q" Y$ n9 K
8 l  q- X2 |7 ~% ~$ H" C' }1 o% Z
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
6 l! a5 m; u$ l/ h        virtual bool        CheckAndRefresh()                                                                                { return false; }
  w( C" s9 n& O  {$ n3 F        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
. V- b! d9 ?" D5 a/ U3 }        virtual bool        IsReady()                                                                                                { return false; }
2 ]3 b6 g& y7 ?0 [        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
- m" K5 |. G! I$ s( O1 |};

' Z2 V. o, ]4 ~( Y8 w& v( A
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
2 t6 H( i* i$ }5 u; A6 |; S, r6 m5 d

. p7 Z: E+ a3 H! e7 R3 A#pragma once
8 @! ?5 \5 _6 v3 l4 P2 R! K1 }; h& n#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
. Z5 M  e0 _/ u% [#include <upnp.h>
#include <iphlpapi.h>
0 o/ @9 h+ X; k#include <comdef.h>
7 u! J* I! O8 c3 b  K' T3 H: J" a6 B" ^#include <winsvc.h>


( D0 Q7 ^; s% n6 H& h& U( K: @#include <vector>
8 e8 y0 `5 S8 {$ |9 g; {; E#include <exception>
#include <functional>
& y; D+ b* _+ }" \8 h  C+ }& N9 c" c
2 X) v  m  u% w% W; w


8 ^$ o5 [/ |, W) L9 G; {- k! h3 ]9 Xtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
! x( b9 F+ a) W1 [" Mtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
* w# t6 B! `9 Q7 ctypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
0 G. {+ [5 A; o0 y6 U$ X! `typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
" o9 a. p0 j9 n7 n4 F& P

! W0 b1 n. @% Etypedef DWORD (WINAPI* TGetBestInterface) (
5 k* P8 f, [1 _3 e  IPAddr dwDestAddr,
; e; K2 c  r; f. [  PDWORD pdwBestIfIndex
) A2 R6 M2 M. q9 n( }5 a);


1 C( o5 a& U$ w- S. Gtypedef DWORD (WINAPI* TGetIpAddrTable) (
0 o  ~, b# m0 w7 R1 ^. q  PMIB_IPADDRTABLE pIpAddrTable,
5 P) F3 {( G) y) v! |  PULONG pdwSize,
4 R# v$ p+ D2 B  ?, v, E3 m7 K  BOOL bOrder
& L4 \" \) K: n4 d7 _/ t);


9 w2 u0 X" P. }8 a# ftypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

  l+ y. S& }' A1 S
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
, `' y* F( z) P$ y5 s3 h

class CUPnPImplWinServ: public CUPnPImpl
9 h* _2 X6 a4 l, ^2 w6 H{
6 S) ~  g1 e4 L; z! Y8 H0 ]        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
( B" c* h1 ~6 W4 x% j// Construction
public:
# ^2 V! ^' z( j1 w/ B# D        virtual ~CUPnPImplWinServ();
* ?4 r" {6 m1 a* K+ _/ W( ~: O        CUPnPImplWinServ();
: ~/ [8 e  M, X$ N( M1 Q
: n# V4 B- m' A  D. H
& k) S) Y! e$ E% h% p# P& M) v        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
4 q5 S( ^( {# u( v* Z/ R4 Q        virtual void        StopAsyncFind();
) \. B1 v$ @* }0 X8 X4 F. Y        virtual void        DeletePorts();
5 t0 z) L: ~3 P        virtual bool        IsReady();
' D* `8 P' E  @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


9 w9 B: v% z: m0 _: C$ m5 T        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
+ ]) j+ ~9 P5 J. z. f2 e, q        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
: @" \3 H$ d7 h2 _  u9 I        virtual bool        CheckAndRefresh()                                                                                { return false; };

) L3 @- s; |3 N1 |# ^1 `
5 r3 _& U% G1 q0 m- j( mprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
, v  k' n1 n7 q3 o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
$ K' L1 N' Q3 y" i' E# y% o        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
4 m+ G" Y& `. q# v# l. o- N& z+ P+ c        void        Init();

- @. N3 z. f5 x( ~6 R
        inline bool IsAsyncFindRunning()
4 Y  Z* x" y; k+ w        {
( f( [: P2 M3 x' s                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
7 Y% s- ^6 {  L! r9 J$ T$ A' `                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
! d8 C+ R+ w& ?: ?# `9 K& g- a                }
                MSG msg;
! ^' ^5 O3 f; e4 ^                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
3 \7 m, |0 X9 _+ [4 F& i                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


# J: f& }& b% _        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
3 h- m1 ?$ V7 j  Y        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
+ V9 J/ x. m+ ~7 Q' J( [2 j        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
2 ~$ A5 L( c! r        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

8 L1 n1 k* W' V# W6 K$ o
        TGetBestInterface                m_pfGetBestInterface;
; P- q( }7 W5 X' a& A! w' p        TGetIpAddrTable                        m_pfGetIpAddrTable;
, z* |* s4 ?' K0 J        TGetIfEntry                                m_pfGetIfEntry;
( H2 o0 S5 h9 w( ?- Z$ Q

* I6 @: @7 M' P        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
! y! i2 ?8 r. \% H9 R2 p        {
# h' _+ _9 P* I9 ]: z; c0 e                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
7 V/ ~4 [/ l# u. A0 z( u: a" y                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
; a2 T: t1 p! e+ @9 G% j) ]                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

0 l: N+ }6 _3 W) |) L
. q$ ]/ I5 M8 R                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
1 w( U& s/ X0 Q1 \. ?

+ ^8 J0 R6 m6 e# W4 S& q                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
+ m& Z0 G& B+ e2 p                CComBSTR m_udn;
% k6 i8 z6 E; D. I        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
. E6 E. a1 f9 n& F- ]4 M5 n        HRESULT        GetDeviceServices(DevicePointer pDevice);
" U) e3 ?9 o! X% p4 A, ?: X5 x! T        void        StartPortMapping();
9 B' d, k* u" D# O        HRESULT        MapPort(const ServicePointer& service);
" ]1 ]/ D* J. L+ {  \' g9 s4 ]        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
- q1 x0 h& _0 m        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
5 @) B! G2 m9 ~7 K/ x                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
3 n  ~5 Y, i- B/ Q7 J' p+ N. _
" W( h1 Y' l4 c  }! c
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
7 x6 z  h8 ~0 B+ s8 N1 z% }" l1 t        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
$ d- q" U/ w2 U& x! L/ W        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

: g& W; Y7 x1 Y& Y+ |3 O# J: \" E
// Private members
" p, q+ u- @) R2 n; eprivate:
& Y$ ]6 Q1 Z. @" H        DWORD        m_tLastEvent;        // When the last event was received?
3 k/ v. h- d" b* d4 L- x' q* E        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
, r4 L5 w# G/ s# {& D% b" h        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
1 r% T% x5 t- g5 G        ServiceCallback                        m_pServiceCallback;
7 h. x8 f; P: s

        LONG        m_nAsyncFindHandle;
6 O9 w0 p  h/ N6 ?. i1 R        bool        m_bCOM;
        bool        m_bPortIsFree;
) ?5 ?! f  u0 M* f2 @6 ~        CString m_sLocalIP;
        CString m_sExternalIP;
0 g7 m0 O; \: L        bool        m_bADSL;                // Is the device ADSL?
* p% l! _/ [8 a: u# E$ _; |* ~        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
" e; Z3 q% @, E. e1 G        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
! ^$ x" Q3 ?3 a+ g
2 P3 j$ z2 u% d. [( a! s- u. U
// DeviceFinder Callback
& z5 K1 l8 Q( \4 A* b4 d7 o' k" kclass CDeviceFinderCallback
& y# r* F& H7 S" I' N( f        : public IUPnPDeviceFinderCallback
# I  o: C6 ]7 Q! f0 q5 e, `, K{
8 r% a4 Y% j+ H2 S+ s. D  Rpublic:
8 |7 n0 u5 y7 h) ]$ x: w" a, e        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
& \; h& j6 v* l1 B9 {* c: k   STDMETHODIMP_(ULONG) Release();
, C" h+ c$ D: M. f

// implementation
private:
* [1 J0 ^) ^: f! u7 V# Y1 B! J/ [/ D        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
1 i0 t' G: r- ?3 o& A! y" W        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
" j9 p) A+ d* X( c  B        HRESULT __stdcall SearchComplete(LONG nFindData);
  ~+ j! N- M' n/ d

8 `! K; b/ ~$ L$ Q# r  _" Z! Wprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

% ]# ^" F% s1 P
* F8 l0 d7 p2 g// Service Callback
2 j; o! @4 r; F" D) \! l, Jclass CServiceCallback
, c* M( l* j% C; y; z- v$ F        : public IUPnPServiceCallback
{
" ~( W" M* I7 L: M$ epublic:
        CServiceCallback(CUPnPImplWinServ& instance)
- f8 q) v1 _* j                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

3 g: P& l6 g9 l& P
, L% t( d& p2 V// implementation
private:
4 b) `. V; f  v' t3 m        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
( J: O+ v% [9 A        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

( E# R; `: M2 \5 _$ Y+ p2 ]
private:
& K) U1 Z/ X" g( s8 x2 g        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
" b$ X+ J% Q3 h2 d% |. y, d};


1 P: ^' }" R+ [1 Q* p  N- B+ z7 c/////////////////////////////////////////////////
4 ?( i- Z& W7 `4 s8 t! }

使用时只需要使用抽象类的接口。
" @! ~9 s* e3 f% Z' e- M# J$ JCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
- f6 o: x" u) I2 ECUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.