UPnP

天云上人

/*uPnP.h*/

1. 
   9 }5 x. R% b7 ]. L
2. #ifndef   MYUPNP_H_
   
3. 
   # F; {2 B; u" h6 @) I; e3 _
4. #pragma   once
   
5. 
   3 }* C0 Z- ?/ b1 W
6. typedef   unsigned   long   ulong;
   ! @% W, L# k- L7 c+ l8 T0 _& ~  r
7. 
   
8. class   MyUPnP
   / v" P3 y: {  \
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    . @4 p: }3 z( S# h) b* c
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    3 A& @" ~+ d$ x7 }6 ]
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ( V6 Y9 v( I) I  [7 k7 g6 F6 d
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ j& _$ [9 j/ S
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    6 n& |6 ^( X* P$ T
17. }   UPNPNAT_RETURN;
    
18. 
    5 t% h4 Y$ B" @; ^3 t% W6 b( g
19. typedef   enum{
    # @5 L" u/ B! W7 H6 ?" d/ f3 m
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    2 P. ?1 M- w8 M: l% N  Z& m, U
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ! i' x1 k" d( L! `) h
24. typedef   struct{
    9 Q# D# A/ `1 p0 e
25. WORD   internalPort; //   Port   mapping   internal   port
    . [: }7 o5 ^/ }0 [! [$ H
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    $ S0 l/ l1 d7 \# ^) |# F3 c
32. ~MyUPnP();
    6 L. S3 W* p) y: H: q! @
33. 
    / M7 g# U& t  B& \- v
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
      T- L8 I- r. |& g! A% c
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    2 a& j. s9 i8 ?' i; l5 @
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    % c' {' }/ z; E% m8 V
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    " h$ z4 x, v) `( a1 q2 p$ Z, R
42. 
    ; |& r5 O7 t. D+ o
43. protected:
    1 D, ]5 o" M/ W- q
44. void InitLocalIP();
    6 d0 ?& o4 X& x$ d' h# d4 G, e  l
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 w* ^6 }9 K( t0 u) ]
48.       const   CString&   descri,   const   CString&   type);
    + f7 K7 H, ]+ Y- Q7 A) U5 ?
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ' ]! [8 a( K' ^- }: G: v+ I" x+ t
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    1 C: }+ {7 ~+ A' f& x
55. CString GetProperty(const   CString&   name,   CString&   response);
      ~3 B) o3 ?: L1 k6 J( I
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    9 w1 ~; T2 h6 p- j# W
62. CString m_description;
    $ L' L2 S- R3 U/ X$ S9 e6 ~) t* w
63. CString m_baseurl;
    % B. u9 W$ O, F3 V/ Z
64. CString m_controlurl;
      c, x. P5 b# A9 B) l; V& [: S
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    - Q9 Q: @, T) u
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ( h" m  d. K3 s' m5 U
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    ) Q+ r+ r( e: A" L2 j
76. bool isSearched;
    
77. };
    : G3 E7 F; Z- p/ k
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   + `3 x; H5 u: g5 p
2. #include   "stdafx.h "
   9 T3 t% t) _2 e6 E( F5 H
3. 
   
4. #include   "upnp.h "
   5 s' {) \0 `: @; D
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - C& D, n) T6 n' h1 I
11. 
    6 p8 k2 r, r5 i6 p/ C
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    4 c" M' Q' E" @, V, a
13. static   const   int UPNPPORT   =   1900;
    ; y9 H- k0 q: R
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    * D6 U9 R  e$ Y, [4 n
16. const   CString   getString(int   i)
    . [' p6 n. `8 E/ D5 t( ^$ n
17. {
    
18. CString   s;
    
19. 
    ! P0 H. M* k2 }4 d' |
20. s.Format(_T( "%d "),   i);
    ( K  D1 C3 X. l. m; o7 E
21. 
    " @* p2 Y$ ^6 \" x3 _- M
22. return   s;
    & ^. t2 s4 E4 V1 @  p0 v8 j
23. }
    
24. 
    , v( K, a* c0 p: ]( }
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    / u5 U. `" k6 `8 m1 o
26. {
    2 s% Z4 `) h" o; G& T* ?, _
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    . l! A+ B, f: V) l1 N
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ( S# F) ^7 e! v3 H
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ; W9 L& {6 ?  I) {: O9 F+ }
33. }
    3 v9 t3 J7 W. _2 D: V
34. 
    6 |+ i% ?: r# f3 ~4 K
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    % e: F, f1 x% n' g7 y
37. char   buffer[10240];
    % l2 }4 M7 x, o& O+ d# H& Z! ?
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    * L* H$ Q5 _8 |( s$ D3 C% q
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    1 J; i: G- Y) R7 F/ [0 S) t
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 u- O- k* U: \* N
46. sockaddr.sin_family   =   AF_INET;
      g3 o, ~. V5 K
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , M% X' E. z" U) z4 a) p
50. u_long   lv   =   1;
    6 x( S/ |; g% a
51. ioctlsocket(s,   FIONBIO,   &lv);
      H! S4 {1 y  p0 ]
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & n$ B/ J& _! O( t: V. v+ ^& Y- c0 b! E
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    " L' l( B" C" F
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * T, c" p# |! Q
57. closesocket(s);
    : K  R8 e+ q3 C7 \' \
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    , ~/ I1 s# @+ _
59. if   (!rlen)   return   false;
    * c9 J% U: t6 C; g
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    : ^2 J+ n' w; E
62. 
    
63. return   true;
    
64. }
    
65. 
    * z" u0 s8 L5 x# b( w
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    4 g7 ?) W4 `9 u1 Z6 n: F
68. char   buffer[10240];
    0 o- y$ A1 w* }- T4 G, B: [; \
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    1 W3 J+ G0 F  L6 D: Q
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
      `: ?4 w4 n' D! [  o; m# y& }
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    2 j3 c. g0 H: u7 [$ q8 \3 H( e: ]
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    . H7 T0 a( m6 c7 U
84. {
    . Q% s" G. h2 G% I  e) f8 G+ j
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    : |* O, Y" O3 ]0 U/ i" ^
88. 
    3 a( d2 A/ J! t, k( H
89. result   =   response;
    8 o. D! {. n  s0 J+ r: P% C
90. result.Delete(0,   pos);
    ( B5 N5 u$ M' {
91. 
    
92. pos   =   0;
      l# @8 r3 [% V2 [8 \
93. status.Tokenize(_T( "   "),   pos);
    " b8 p+ O4 a& K8 q6 e2 |; \: \
94. status   =   status.Tokenize(_T( "   "),   pos);
    . m8 f6 w4 o9 V0 W
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    $ U; g7 X: l0 D
96. return   true;
    / a; g8 H3 X. m/ N
97. }
    
98. 
    3 f. _* k1 G$ R) F
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    : u) t4 S7 T/ \
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     * l; r* T3 C: G! l; A& M5 E3 n
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     4 t7 T; t. ~7 y8 i% w1 g$ G
103. CString   property;
     
104. 
     + H* p) r; R* ^0 P) O3 j8 t& F
105. int   posStart   =   all.Find(startTag);
     4 q% g. V8 I4 G5 n
106. if   (posStart <0)   return   CString();
     6 q- r- T+ p8 T6 y: ~% _
107. 
     * F- o" H$ l7 R& u  N2 X9 Y2 e+ s0 U
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     1 l  j$ r. Z4 z) w
112. }
     / x/ Q' `1 D. c  e
113. 
     
114. MyUPnP::MyUPnP()
     8 Q4 t* e% Q! b; t
115. :   m_version(1)
     / z0 R. c) V8 f9 R6 K
116. {
     
117. m_uLocalIP   =   0;
     " e* d3 J0 v- s6 j5 n2 f/ q( g0 X
118. isSearched   =   false;
     ' L- L( f2 E' V% t
119. }
     ) g: j6 l: M3 k6 l7 |
120. 
     # a% W4 v  e, T+ K
121. MyUPnP::~MyUPnP()
     : j* h+ w6 R* ?) U) b9 N+ R2 I. w
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ; Y! D9 r; F6 e! Q
125. while(pos){
     , ~7 q* C3 i8 Y' E& d+ V' {; d
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     1 F0 J8 z: c' v9 s
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ! A3 Y2 O8 Y! y: i+ ^
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     4 I* a2 v. R# k' }# Z) t
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     . d$ H' S* i' |7 Q
144. };
     
145. 
     4 \) [+ X4 @- R: K. v/ J
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     + @* t/ J4 v# {/ e' t6 T
147. u_long   lv   =   1;
     0 j' ~; O5 o" b% `* Q4 y
148. ioctlsocket(s,   FIONBIO,   &lv);
     ( y5 O! M+ W$ b; n( u  c/ Z
149. 
     1 Y! @8 a- V7 Z3 b. G7 f0 m
150. int   rlen   =   0;
     4 G# I: J" P" E/ c, D
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     . _  y( Z7 N1 B# F
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ) {! r" Z0 N' m+ }
155. CString   request;
     ; A# T/ P; w2 F; O- o
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 "),
     / B- E$ J  I* y. z
157. 6,   m_name);
     0 s6 E+ H  Z3 ]
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     " Q$ {; w4 @$ g) b
160. }
     ( k; p! {0 f# f3 w
161. 
     
162. Sleep(10);
     : o6 h8 x) e3 O# O" m
163. 
     4 T- x! q  D8 K7 N0 f
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     . {; k7 m  ]  Q+ v1 U$ Z- ]: A
166. if   (rlen   <=   0)   continue;
     2 ^9 X* t( J. v, h
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     , q1 D; C# M. Q" U' `5 X- D, o
170. CString   result;
     8 T5 m# C: ^4 b1 o
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     : W. B" p- q1 }
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);
     , F8 D; [/ e- @$ Z% r# b# d
175. if   (result.Find(m_name)   > =   0)   {
     $ w+ K' P) K3 _* p# I
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ! z  L. u+ ^# Z
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     5 v* l4 F8 ]! {3 T* ^0 e
184. m_description.Trim();
     ) I3 l! a! R, b! t6 L5 i! X9 H
185. return   GetDescription();
     6 ]  U4 m+ i5 e  r
186. }
     ' ~9 [( i9 e, U7 |
187. }
     
188. }
     
189. }
     ( \; T7 }0 d4 w+ |# K. O
190. }
     : I7 d9 n, F! W' z
191. closesocket(s);
     , Y. S# P% h) Z( g1 m  `8 r
192. 
     
193. return   false;
     
194. }
     & s- x: b, w2 ~% l5 `. W

复制代码

天云上人

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


///////////////////////////////////////////
( G" R3 l3 M# l/ }& r& G//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
6 C- ?; ^. k( ?2 C' h& D

#pragma once
#include <exception>


  enum TRISTATE{
8 G2 R) A, ~. {) g        TRIS_FALSE,
4 m4 K% N6 O5 J, Y        TRIS_UNKNOWN,
8 ~# ^/ u  ?* Y: H0 \" S0 T        TRIS_TRUE
2 u3 o4 p5 r- `};

3 {5 w5 S( R4 U# @0 o' t  I3 s
enum UPNP_IMPLEMENTATION{
' m+ F, c$ ^. ~/ S9 I        UPNP_IMPL_WINDOWSERVICE = 0,
9 l* ?, u2 I5 S  X        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
; [/ F/ Y8 J9 i* w3 W% p};


4 U4 B2 e9 F7 p3 Y( V' F- k

class CUPnPImpl
{
$ g; W4 e2 J) N4 X) Npublic:
' D5 |7 [$ S  F1 e        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
+ F" x1 f& _& M3 ]/ U: M' n                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

. n5 f$ S. E) i$ Z8 I% [2 R4 Z& F1 }
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
6 K( O( D+ ^8 N! I        virtual void        StopAsyncFind() = 0;
- p- B# y1 l% w6 `: X        virtual void        DeletePorts() = 0;
* }  ^' E6 x2 M9 D' v. E        virtual bool        IsReady() = 0;
4 s( z0 l' H7 s& H3 b8 ~, ^2 b        virtual int                GetImplementationID() = 0;
; N8 o2 ?$ y) g4 f* v& D2 N& B5 b# v       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
3 ~- y7 C3 b! o! X9 B

( d, y' j1 \6 D! `6 L& F        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
' `$ L% H9 a0 m3 W$ }        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


* j9 L; C  |- ]// Implementation
7 C! a4 C! V& Q* g6 X+ eprotected:
6 X7 I! B1 b! c* ]        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
8 M% X# ?7 [3 l4 I( {: u) {# K        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
6 B. ^7 \7 L( ~        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


) _. h5 y! z  b% Uprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
1 }6 N% e! C/ J; c

* K, c" h  a2 G7 m$ W};
; A3 T/ v9 }1 }/ N+ V! A

) Q1 a& A" Z7 d// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
7 M0 k+ C  v7 `( b* A  Qpublic:
" W- p: a: ]. u: ^- u$ h        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
6 t" R) _# q" F8 R        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
5 ~3 F' ?& O- ~4 _" `# [( m
2 Z( ^9 e. G; P/ n" a" d
/////////////////////////////////////
* z9 q- Y( O$ Y$ O5 z$ H//下面是使用windows操作系统自带的UPNP功能的子类

" c! E) f' q) l9 {+ i0 ]
#pragma once
$ f" v* H( q! x# r; W& s3 Q9 y/ o#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
2 U+ W2 }7 b$ ^  i" Y. L#include <upnp.h>
# u: F1 t# R& U3 q, Z, c1 X1 ?+ k#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
/ Z8 O% _5 `4 U- ?0 z1 n#include <exception>
#include <functional>


# V$ ^: O1 R: f% L0 \
( j- c  o2 t8 r* v; R7 a
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
* B# T" e: R3 P" z. r1 u* ]# Rtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
. h: ~5 p) M5 k, c5 Etypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
* ]  R: k- J9 Ntypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
+ k' F' s. r* @! O! v  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
3 n- y2 M# n& K# \& L- n1 N1 _2 U2 n
' j: K; [7 [7 f% @' F
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
2 r, q7 J6 F2 v- ?: |  BOOL bOrder
9 W1 k7 C) a  v2 k- n4 c1 p. T, L, ?; ]);
, t  J7 F+ m3 ^& i9 \& J
7 H4 J0 z, X: n3 y$ b8 I
typedef DWORD (WINAPI* TGetIfEntry) (
" c# `( V, ~8 }( E! b  PMIB_IFROW pIfRow
);
1 H* C9 Y5 s' l9 Z; u3 Y* ?, ?  x* c

: [2 @9 T$ t/ J8 b6 aCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
' Z( b" n' F4 R4 y$ j5 v- I4 o
  U1 B: f  k4 J  b2 g
1 A% H3 ]) Y6 E% gclass CUPnPImplWinServ: public CUPnPImpl
{
9 K* w7 J* U( L7 \  c5 l% w        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
  u3 b& l) W6 d4 N2 F; E% `public:
% O5 G' F! [. P- f        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
4 d1 {- A' d: L8 B" _

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
* W+ ^2 ?0 q- p2 E- [% n& Q2 W        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


  P% `9 B* `( X" k3 k$ [9 n5 _        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
8 p4 P) f' B, s! a5 z: q( n        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
2 r# {0 s( w! ^; V* ~/ A" y; H        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
" M' O3 W0 H/ c) c        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
" Q! D3 K( _" `2 u; W  @7 W) r        bool        OnSearchComplete();
! V5 Y6 t3 [5 @, S$ U$ y" H        void        Init();

3 b  m$ {% y( ~, I7 ?
: I) k3 G) O# a6 w0 \$ S5 R        inline bool IsAsyncFindRunning()
" J- |  y" e$ t! o( m9 }        {
: T5 D/ m9 Q" n( h9 ?+ N$ Q                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
$ z! ]' Z7 n3 |5 l3 c                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
$ X1 f' h. l$ }  o                        m_bAsyncFindRunning = false;
8 u9 A) ]8 X6 o' B                }
, A+ d6 q' P* ^+ A9 E8 G' L                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
* d: L; \+ M  U                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
% x5 {& r. x* ?% d; R  S

        TRISTATE                        m_bUPnPDeviceConnected;

. l* Z5 B5 B( P1 @4 Y# O9 e/ e
3 a  K: e$ x5 I! K8 Z// Implementation
        // API functions
0 c9 k2 o; p* L9 a# x! ~5 z% M        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
) L/ T2 G8 P$ {6 p  W8 A- A        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
. K/ {2 B+ N, m1 M        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
9 m% Y' I: Q( C% j        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
! Z4 U3 w& A) Z( k0 Z4 [
3 v9 ~$ W! {+ H" g- c
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
" Z6 X; U) ^5 R% ?6 t0 z
3 S' s. k9 n( p1 n7 l/ J
, A$ K& R! q  [/ g+ Q: |        static FinderPointer CreateFinderInstance();
, Y9 N8 H! a; R' g; p. y        struct FindDevice : std::unary_function< DevicePointer, bool >
6 ^7 w0 S% N/ z; x2 L, D: M  X0 V. A        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
4 P" h7 h$ {9 Z1 U" R4 c3 h                result_type operator()(argument_type device) const
                {
1 J) J: G% ^/ g                        CComBSTR deviceName;
& x+ T7 i$ |$ ^% Z3 A- n                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
8 ?8 L( ]' N- m. c  A

' R  R5 }* H2 d+ Q                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

+ X% [. X9 U$ a; ~
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
0 @( u+ ?5 [1 U' e- N                }
) l, z+ E. S/ j* B* A9 v                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
( G$ q( W! a$ g8 _7 v7 C& i        HRESULT        GetDeviceServices(DevicePointer pDevice);
9 r! l  X9 q! C1 O6 A0 e: u        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
/ ?$ y! G( ^9 k- S9 U9 n4 E        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
$ p3 M1 y: f  u8 t0 L5 g( e        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
0 r* q8 I/ E2 C# u* @                LPCTSTR pszInArgString, CString& strResult);
5 B: x% T* k3 G# G" ?        void        StopUPnPService();


$ \& o( Z- R( z. N0 p/ c        // Utility functions
0 |' B+ U3 T+ y8 j1 R        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);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
  M' X% l- `$ t+ r        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
9 R" C  l9 M( H& V        CString        GetLocalRoutableIP(ServicePointer pService);

0 e4 l4 i% F8 N/ b! ^% X. ~
8 v* S7 X+ T8 p// Private members
6 o5 k% e# M0 Cprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
; o/ M6 q6 b  i1 e6 y, Q        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
0 `7 D6 @0 B1 g5 v; W        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

  H5 V. y, V2 K; w/ I6 x! y( f
        LONG        m_nAsyncFindHandle;
  B" ~7 O  ?+ a1 u, Z8 Q5 Y        bool        m_bCOM;
% S+ y1 @2 s, {! U: m; F        bool        m_bPortIsFree;
  u% S/ g& I8 {! L, I  l- e        CString m_sLocalIP;
1 Z" k, O0 e9 \        CString m_sExternalIP;
3 A4 D. U6 W; n7 n, V" x' r0 K( e        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
. R9 v  _% f/ n9 y3 b& |  [        bool        m_bAsyncFindRunning;
9 E% w. E8 c9 k5 J        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
8 e$ R2 N/ [: Y2 u* ]0 b        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
. Q  }! S, v) I) d
) v; I" m( M  P  L! E3 y3 v
& d: H4 I! }8 d% ~1 E% U};


// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
+ U2 F% \$ G) i$ }2 {! Bpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
9 ~& \6 I+ G+ g- k* D+ p2 `$ I                : m_instance( instance )
* P8 R7 ~5 ?5 ^7 o, I! `4 E4 |5 z        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  K' G: ^6 C( L# z7 i   STDMETHODIMP_(ULONG) AddRef();
0 W$ H; N! z% |& M3 G" O) [" a   STDMETHODIMP_(ULONG) Release();

  U& Y; W7 E  t2 U
* k; i$ v. t; m  W// implementation
private:
. j- V% V* T7 q& W2 m: w, P        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
, W' ?3 ~6 t4 G  [' t4 [        HRESULT __stdcall SearchComplete(LONG nFindData);

" @+ n2 w  |1 v2 c, b4 T( ^
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
/ X! o& {" P9 v};
4 ], C1 P. s8 ?8 j9 B% O$ l+ }* b

// Service Callback
7 O0 i2 L7 G0 A+ O+ Z" Y1 L# ?class CServiceCallback
+ r  a! D* P' j        : public IUPnPServiceCallback
( `! h1 _/ N; N% ^{
: k0 G' a4 l/ C$ q0 o, ^public:
% l1 J' x# A8 z) o3 M7 o- ~' I        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
' S+ Z- g$ X2 J  _   STDMETHODIMP_(ULONG) AddRef();
: y2 u: J) t' C& W   STDMETHODIMP_(ULONG) Release();

3 J3 S! k. k2 t
; g# H% P% U* K& {$ Q// implementation
private:
: N& L9 I  R* U+ c& n. [+ y! X& Z        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
7 n' b7 ~! O5 Z' C/ O& v        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


2 \9 X2 e1 N+ [" lprivate:
2 N. g5 o* a0 L: N" z  z        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
. S7 v" M9 t0 S! ^( y1 B- _% T
2 {/ F* f5 V' b" [9 j
/////////////////////////////////////////////////


6 O: X; g6 S* r2 w* h+ \# y7 {" n使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.