UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   5 c' l3 u8 m& v/ z
3. 
   
4. #pragma   once
   % _. r7 \" a: A# ^8 b3 Y5 ?# s. J
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   7 E% Y6 G) Y. x  v9 l: m8 ?# q
9. {
   % s. p. _3 Q; S
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    " Q4 }: d1 p7 X3 ]
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    1 F. n5 G: [# S/ ]& x+ D( k- n
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    " a  K! H. |! Q5 L6 d! O' {8 u! S1 C
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    - G# Y: X( z1 A7 h$ W
19. typedef   enum{
    * r) a8 S$ Q- Z" }
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    ! H. u* |, v6 M2 r/ ~
25. WORD   internalPort; //   Port   mapping   internal   port
    & ~5 H9 L% t* t7 S/ V
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ' V9 B& o) q6 r0 j( M& T- S' F
28. CString   description; //   Port   mapping   description
    , t6 Z8 ]  L) Q! p! u
29. }   UPNPNAT_MAPPING;
    ! j% i- B; v: b# z/ [7 g
30. 
    $ _5 N( P3 F1 U/ T
31. MyUPnP();
    ! _: W" r; j2 g. s0 ~8 m
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    : K, Y1 J& \1 T5 Z: m. m4 v. R
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    ' k& ?% v: V- O8 f7 J
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    0 t. k: p1 {9 ^  p
41. bool IsLANIP(WORD   nIP);
    . R+ w1 G  s& H) E$ s
42. 
    ! Q1 I1 c0 F1 R$ D& g* P+ h0 L/ @
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    % l/ u/ j0 T! N. W. f# `3 q+ ?! _
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ' s% b) x" |; r' q/ P6 ~+ `0 E: o
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    & A7 w* g7 g7 z8 j' v9 i/ a( v7 Q
55. CString GetProperty(const   CString&   name,   CString&   response);
    : G, x8 }1 S% X- n$ z* v7 G0 h) v
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
      j6 e! m! ~# G! r# }# W5 O
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    7 ^5 A5 C* O8 _2 f: J# ]
60. CString m_devicename;
    
61. CString m_name;
    ! `2 {( Q+ e8 `) {3 }
62. CString m_description;
    2 p3 x3 D. {8 E) z/ ?
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    " X3 l' ?3 |# U  `4 i
67. int m_version;
    
68. 
    + F4 m1 Z7 M7 N& ?# \& K
69. private:
    5 ^# z7 d) A0 ?& O
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    $ V8 ]5 \/ \7 m$ M9 ^7 o  w
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    # V& o$ W  I8 |% `
75. 
    
76. bool isSearched;
      {+ e$ K" k6 D6 n
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   8 {& N: V# H9 O3 g4 Q
2. #include   "stdafx.h "
   ' V0 S4 Y1 G, H: z5 ^0 O
3. 
   7 r! a- c: N% g
4. #include   "upnp.h "
   / o3 K- A, x6 P! Q0 H/ p& u9 A
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ( y& ^& Q9 @1 P' O) p2 K2 |8 I
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    9 Y/ r- W, U3 c! l
11. 
    # E, i/ U1 [5 H. U' I
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ) [$ u3 O' g5 ^: E
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ) g$ m" _4 F3 ?9 S; G# ?; U& y) p
16. const   CString   getString(int   i)
    & \" w: N* u! ~$ n$ t
17. {
    
18. CString   s;
    
19. 
    " }& R6 K: w3 v( N5 G
20. s.Format(_T( "%d "),   i);
    , L# K' H. P+ f' }+ f4 U
21. 
    + n& \( B! E9 F5 z. G
22. return   s;
    
23. }
    / n4 R( ]7 J5 G4 ~6 ?
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    & d( O9 l, \/ ?8 T7 Z& W8 s
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    1 ~1 u) C; p2 u. e6 A' U* Q/ g  n7 @) X
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    * }/ C  {0 h+ A
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    " m6 Q/ g- o* l+ G$ x, W
36. {
    ( n9 l; y* o; @( E9 v7 G
37. char   buffer[10240];
    
38. 
    9 |& }: o* g3 c5 A6 s
39. const   CStringA   sa(request);
    - o9 j; U2 i& M; b! P
40. int   length   =   sa.GetLength();
    / ~+ ^  j% j2 ?. j
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    $ o& O; E& E0 ^7 x4 p3 p
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    5 m% v1 c: b( W9 b
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    9 `# G3 I' {5 x2 E  t
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    9 b8 a& d  a) c" x( c; ^8 h
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
      a" l: p8 b3 Y- e: g+ _
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * L3 y3 B9 ]; a, L, i4 Q( e+ r4 p. M
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    " I# c5 V+ K3 P# }8 T* U! |
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    0 L: _/ \8 ~( G  l
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    $ }: p# F* ?% d6 _' F6 A  W
67. {
    
68. char   buffer[10240];
    ; E& q' Q( `' e! R3 W6 U2 Q& R4 j
69. 
    " @+ Z5 _, [$ h
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    7 [) S, |; l5 K+ ]7 r
74. struct   sockaddr_in   sockaddr;
    : b5 @# ^  _: m8 u, y  C# ~
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ S! ?8 b! \/ Q
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    8 ~% S* i+ W8 q- M* R
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    5 n  _4 X2 |" ~, b* Q
79. 
    3 o3 t$ V- d6 Q- o/ f7 Z" q- C! |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    / [, k; Z; P# B* S- Y: w
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / b2 j  K# j$ X; Z6 v! c
84. {
    6 S0 N! U5 `; U1 K9 q" v7 l# L+ [
85. int   pos   =   0;
    : l% ?  s  l4 E
86. 
    . D( I8 N& |8 a  _" {0 x# L$ V
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    " Z) p. @  |: r; g% a0 T) ^- P) S5 G' q
89. result   =   response;
    ; Z6 X* l; ^* o$ ]* ~: X# F
90. result.Delete(0,   pos);
    + q' ^) K8 P6 m$ s) y
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    8 o8 \" f% d7 ?- V
94. status   =   status.Tokenize(_T( "   "),   pos);
    # v* k5 i# `8 a3 z7 Z5 Q0 ?
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    4 ^# k& M+ G0 u) o$ ^. D1 ?( c+ e
96. return   true;
    
97. }
    # c: F3 L/ L  Q2 i8 a# \% V( {
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ) Y& F( X/ K/ M( D
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
       b8 |& i! Q+ ]4 P. O6 U
103. CString   property;
     1 x! b" ~+ e) ^, U
104. 
     ! Q. V! {7 S  F. J% A4 e2 I  t
105. int   posStart   =   all.Find(startTag);
     ' W3 d4 Z% W7 w( G
106. if   (posStart <0)   return   CString();
     
107. 
     " Z* S5 V2 U7 Z
108. int   posEnd   =   all.Find(endTag,   posStart);
     7 V' S7 K# Z/ f7 B9 r$ v) B7 f. ^; O2 f
109. if   (posStart> =posEnd)   return   CString();
     & j9 o7 v/ h4 ~% _4 W
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     $ v$ m4 \7 E! X1 p
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     , R6 y  B4 V9 h2 S, q6 j2 T# g
116. {
     6 U* E/ N7 K; y& R- ~& K: _! b
117. m_uLocalIP   =   0;
     7 b9 r4 @( P$ W% m6 X9 d1 R. L
118. isSearched   =   false;
     
119. }
     1 G+ G9 s6 c2 Y) q6 u: }& Y$ F0 T+ |
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     ' k& k9 a5 w. j: v
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     8 p! `  c! S, U( V8 ^: h
127. RemoveNATPortMapping(search,   false);
     
128. }
     - J6 `8 x+ C- Z& d, [, W
129. 
     . N" O" E  N% z
130. m_Mappings.RemoveAll();
     9 m! ?% z3 A) I/ w
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     7 A$ N) Z7 S# k/ `+ `; K
135. {
     , r/ a+ Z6 U3 v" ^8 u% T" y
136. if(version <=0)version   =   1;
     : t& s" H" @* p1 e8 \
137. m_version   =   version;
     
138. 
       D. v5 }0 X9 C2 w5 }' z& |- d/ w3 f; e
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     * V, m# ~1 E+ ~% k1 n
145. 
     2 |7 ^" y& }$ q
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     . Q% M- w' V1 B# F4 C
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     7 U+ R: ]  ^! r& u
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; Z& m3 d) u: H; o( A7 y
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     8 a- m, N; i( h, ^" T6 O
155. CString   request;
     ! U) `) a5 u7 U5 V8 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 "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     4 g. t' y% Q& C& r
159. }
     ! C: U, {6 @+ z  \
160. }
     
161. 
     7 B* p/ @0 t' w4 ~
162. Sleep(10);
     
163. 
     $ D3 E4 A1 z1 c1 Y" n; b2 n! h% k5 K
164. char   buffer[10240];
     # A1 s6 i- s* L& q. p
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     - t. [1 ]0 f0 P; e* W$ R) X5 I
166. if   (rlen   <=   0)   continue;
     ' ~8 n; L+ l9 [; f0 l8 o
167. closesocket(s);
     
168. 
     # ?  n  u9 X" c# k+ {6 m: L
169. CString   response   =   CString(CStringA(buffer,   rlen));
     % }8 y9 B8 l# c
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     # Z6 C2 @' ?6 p* k* Y/ V
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);
     4 v6 O# T- C# e( q. p2 G
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     5 G" ]6 Z2 I7 y3 m% X$ l
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: "))   {
     7 H& {. \# ^! \
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     3 N# w- v3 t' P8 f4 X2 B
185. return   GetDescription();
     # o# C3 B% w) e7 K/ H
186. }
     " `5 \- v7 ^8 U6 u% g8 V9 G2 o
187. }
     6 i/ s7 t$ }9 b
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     7 i4 o- Y9 h. Q9 p1 e$ \
192. 
     
193. return   false;
     
194. }
     ' D  c% i8 S6 ?: F$ F5 T) e5 C+ Z  l

复制代码

天云上人

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


" e9 E* D% \( k; \///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

- u1 }( u8 q! B/ D% u* z' s3 i+ l
#pragma once
! V5 `5 u0 r" v#include <exception>

. a- D8 X0 Z- Y2 [6 N% F. O$ H4 c
  enum TRISTATE{
5 v8 }# }- p# M5 Q, `        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
' C& ~# T  f2 `' z) M  Q; X' S0 u. \5 ~

7 L1 C/ H. J) I8 q" z. V) `5 k2 \) `enum UPNP_IMPLEMENTATION{
! |3 U4 m6 Y: L: ?# J2 O        UPNP_IMPL_WINDOWSERVICE = 0,
7 d7 @( @+ m5 G2 P        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
$ W  e3 \, _3 D8 q8 W" W};
8 P4 ^. P: _2 e  g
9 c1 q5 G( I& k& |7 L7 c1 y5 v* o

8 G1 Q4 F+ O/ J$ A8 A
& s; [. S, p* n7 wclass CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
. j3 e" P! U+ h: i% g  Q8 m# f1 X8 w        enum {
                UPNP_OK,
                UPNP_FAILED,
% ?4 o( a4 B% g1 S% V                UPNP_TIMEOUT
9 R& g8 g. d. D) O- C7 r        };
& D6 e, R. y  X% {! f$ ^, O* i

+ Q5 x8 {" S, K( W; l  b5 F: B        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
7 @$ n' c$ j/ U3 F. S" a2 P4 K  F        virtual int                GetImplementationID() = 0;
" J) i! H) q4 T% ^9 ?0 W       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
$ f, U3 D+ f1 ~2 b! k& z

/ R' G$ D& N' h; F0 l7 Q        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 K. @4 s8 z% ~& m
+ G3 `6 b- t2 ?, Q5 v
// Implementation
: v' C% M! k" S9 a: \" Q, kprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
7 V- n% T; q# J6 a& A% Q. I6 ]        void                                SendResultMessage();
! ^3 }* p+ t5 R8 @- g5 s        uint16                                m_nUDPPort;
1 ~3 r. k4 E$ P' P/ p' Z        uint16                                m_nTCPPort;
% [' ?3 K5 H. P) }2 ~        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
1 E9 N6 [8 s1 ]/ S/ @2 B+ @5 x
3 e+ W3 j1 g  v! i. u) z
( H8 |5 k+ e, H" H4 k$ \private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


# J+ I* H8 Y  ?};


9 D# @( f: p. Z8 J# u& V! u// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
  d* r8 C5 n, N- h% ?: m  Q- A5 ^{
# J7 o" u( r+ dpublic:
# [) u, |& o- S, [: A4 R4 T3 o/ u        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
6 N2 s% t: r7 H1 ^& s% P        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 w2 S, l) q% @/ s/ L6 [* `  [};
( _, c( O/ {4 h. `7 }
- `+ W1 R( W' b/ f5 h
/////////////////////////////////////
  J0 T2 t2 ]. k2 Y, U& ^4 S8 v//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )

( `; T# m1 W# d& M7 q: U
#include "UPnPImpl.h"
1 {+ E) W2 G/ H* t  T6 m#include <upnp.h>
- ~: H, \& |/ C/ F3 p3 W. ]8 v#include <iphlpapi.h>
#include <comdef.h>
: ^" e' }4 M" C$ z. V; U#include <winsvc.h>
# C, a& q. V2 o- N( g& O* U

#include <vector>
8 q& L) F( L; r: ?! d6 ?0 U" i#include <exception>
#include <functional>

- \( \% A! p3 H, P0 i" j. J


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
) M- C; u- C! `9 X- Etypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
+ {# J6 a! t+ X' Gtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
7 ~/ @: E: G( v# N3 ytypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
- X7 A4 m' o8 M. o* Qtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
/ V! E! f8 ?' X( d! `7 c( r% r

typedef DWORD (WINAPI* TGetBestInterface) (
  D# t0 N9 h* s9 L5 w' ]0 O2 t  IPAddr dwDestAddr,
0 q& ?2 q. k0 o+ _& ^/ D  PDWORD pdwBestIfIndex
! U, D0 A6 \+ w, W+ ?( P);
5 \$ A$ f3 F- a- L
1 w  N3 D! N! @. {6 V% M
typedef DWORD (WINAPI* TGetIpAddrTable) (
; D( |# z7 p% N7 _. ~4 s% Z2 f  PMIB_IPADDRTABLE pIpAddrTable,
$ v, ^  g+ b" a8 D4 _  PULONG pdwSize,
  BOOL bOrder
);
( a! y* Q2 a8 V  f: g2 S
5 R$ j- r9 O  z; A+ p
) n- x3 h4 c( W3 ^8 Atypedef DWORD (WINAPI* TGetIfEntry) (
( _1 i) m! B3 F2 d  PMIB_IFROW pIfRow
7 e9 I, F2 o7 P6 }4 |- M);
9 ?3 X0 @( Z( O8 `
- }2 m! R# ?" }2 ?$ I- N( p" K
! G$ H, V: N* |0 RCString translateUPnPResult(HRESULT hr);
* V1 P" z0 `& g% L, K5 D2 a4 a1 pHRESULT UPnPMessage(HRESULT hr);
7 L( |+ W# a, M" y

8 w7 X$ T$ G: k7 c+ l, R& U% _class CUPnPImplWinServ: public CUPnPImpl
{
" }% p" u# i2 C7 P1 J5 e. @        friend class CDeviceFinderCallback;
9 j5 n$ X" ?9 {9 a  ~+ p        friend class CServiceCallback;
% Y4 d8 K/ ~" g( o: `! f- J. l8 m7 A// Construction
public:
" r" e% N! l- n( _7 G$ j% n) B        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

( A' |6 K2 U/ n5 f- f. q1 ~, b$ n" G4 Z+ {
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
  l3 q- B! z& P& m. D        virtual void        DeletePorts();
( m: g" c# i' w& L2 k5 T- ]' V        virtual bool        IsReady();
+ P& S. E! I$ w) x1 X* l5 Y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

' q8 p0 J; i1 `! Q- I2 V' o
        // 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; };


* @- @! m8 l) L0 p7 {protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
, `2 v9 G0 T6 t( H7 x        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


2 j& M5 R4 }1 a" |. x: J+ |        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
  B  u# d2 d) A! a                {
) x: I9 z3 q0 b1 S# w8 X; C                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
- r' W& x, ^1 k  L4 K+ E                        m_bAsyncFindRunning = false;
9 E+ }$ u" x+ a* ~8 Q0 _0 D( X8 @                }
* {% D  W* H9 _4 e( C+ A4 x; N2 z                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
: v, ]3 d# T% P: `* i. _                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
" ^, z* `3 S+ a0 t1 _/ y! V                }
# t1 G' o, P& s, m, M- j8 |" i2 \                return m_bAsyncFindRunning;
        }


2 z3 O: C$ I; k3 I/ y        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
+ q+ ~# r8 z! f/ b/ L        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
7 Z  a# ^1 y5 ]% w0 |: N3 D9 G        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
& E& D- W+ b1 x; x" f& |  H5 k        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
. y0 [2 x& c! v4 b' Z        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
- s; h9 j) l6 d6 F% N# K
3 ^) `' r( k3 N! ]
        TGetBestInterface                m_pfGetBestInterface;
+ A$ l( T, U* A        TGetIpAddrTable                        m_pfGetIpAddrTable;
: p" `! }' ~8 ?( G        TGetIfEntry                                m_pfGetIfEntry;
) \2 J% M8 r: C$ X" e8 Z

        static FinderPointer CreateFinderInstance();
* f* b% ?2 w9 q- f1 Q        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
$ m, v, e) O1 M2 v6 h                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
  ?  S: p+ T. `+ H! l8 ]                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


" w) _4 S7 m" t2 T) R                        if ( FAILED( hr ) )
9 z" |3 G! N: T! M5 P2 G8 n6 y                                return UPnPMessage( hr ), false;


+ @/ \' j& P: Q1 W& P6 I                        return wcscmp( deviceName.m_str, m_udn ) == 0;
2 J9 Z, u1 t: L& ?7 c$ o                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
+ o4 m# T: v: N* `& F        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
. Q2 V; t! p5 `+ m! [        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ t. J5 u" x; ^5 d( l        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
, v3 r# _7 f, Z7 D8 a% R$ C/ d' D                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
! N* d/ \: ~; F7 j% r5 n4 C: i' G: H        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ b, G: e# i, B8 b# k7 X        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
( O7 h1 J  r/ @! m: T6 R, w        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
4 ?5 t3 h8 Z2 n; q2 R* d- z
! d% M( e3 v7 s# p
// Private members
# U. ]. `* c3 W  T4 v; _6 g" ~private:
        DWORD        m_tLastEvent;        // When the last event was received?
% g1 p# j1 ~. o5 ~2 m. T, h: m* r        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


+ p3 G$ u+ q1 T) m3 Z        LONG        m_nAsyncFindHandle;
9 g) b. y' `$ n( r+ P. Q% V6 p        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
7 Q6 T6 S2 ^% e, t! _        CString m_sExternalIP;
. t7 f% Q; f# v) J) I        bool        m_bADSL;                // Is the device ADSL?
        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;
7 _9 b6 C) p* s# L        bool        m_bSecondTry;
1 r! X% n1 ?! B- I5 j. Q7 U: `        bool        m_bServiceStartedByEmule;
' o& Y, `. r9 a2 ]        bool        m_bDisableWANIPSetup;
& h/ q6 _4 T3 Q; ]. ~        bool        m_bDisableWANPPPSetup;

& w; \& Q3 x1 G6 r' b% N
4 f4 H' r3 y6 a};


- }7 J" m2 l5 Q) Z: N5 g) d% U// DeviceFinder Callback
( M, _/ s$ |4 a" F# _8 qclass CDeviceFinderCallback
0 G( O3 l2 K4 Z( J. L        : public IUPnPDeviceFinderCallback
{
# j3 H: N. N: rpublic:
7 u" A5 V% `! b( ~0 P6 t* W/ X        CDeviceFinderCallback(CUPnPImplWinServ& instance)
9 R/ Z  w. H; h+ b- l8 J                : m_instance( instance )
        { m_lRefCount = 0; }
2 l  o/ D6 `! h9 I" {0 A! `

! T. ~, z2 T  U2 G   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 w  x/ V; N0 M7 y9 ?6 W8 O
7 b) n! h3 L$ r' q+ H1 U6 ^
  k' v: u: Q. [; u// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
1 L% J7 Y+ L1 D8 ]+ D" T
3 k% }5 o" \$ ^# l
" e% t6 Q- v- s+ D6 i/ m( rprivate:
- y$ a4 j& h- u1 n% z4 |- T        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
$ N& C" N* J) l6 Z

0 ]+ z3 v6 J9 [$ T. M6 T// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
1 c9 S/ T- |' j* {$ I9 p{
$ i/ D1 L( G) ipublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
! S- p9 m5 W4 q, Q6 ^( f3 @   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
4 f: m; C: M& @. [3 b6 h   STDMETHODIMP_(ULONG) AddRef();
8 B+ _0 u; a4 [2 }   STDMETHODIMP_(ULONG) Release();


1 k+ K& c) S8 z# k// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
3 ~* y( d( i* i3 a- V$ ~
2 v+ X* T& P$ Y6 [9 c! D* ]
1 l* V  m9 W. U9 S5 V: pprivate:
        CUPnPImplWinServ& m_instance;
) y! Q1 S. r! l8 w( @' ^% b        LONG m_lRefCount;
};


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

+ \4 U4 @+ ^+ ^/ r# o/ \
使用时只需要使用抽象类的接口。
; b/ i2 j) o$ V( d* e" FCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.