UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   ' f- J, L0 C  S( E
3. 
   
4. #pragma   once
   
5. 
   - d. B4 \. i+ U  [6 X) V/ n
6. typedef   unsigned   long   ulong;
   0 ?* p; Z- O2 p' f4 R
7. 
   / l& {9 {& A- v7 v. ?
8. class   MyUPnP
   " x2 W9 O$ U- K+ r8 ?, |8 q
9. {
   7 j/ c/ ?4 W0 z( k, ]
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    1 O3 _( w( T, ^" z/ F
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
    ' g8 P. A$ o0 E- ^8 d( w+ L4 t
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : B1 B4 z+ Q" _( u
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    , c, ^, u% ~1 B9 x8 ^9 a, K- l2 @
23. 
    
24. typedef   struct{
    : c) W6 ?( r: W, u6 ]
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    7 z% z7 m: x& w) f6 v) L$ A' u
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    * f7 }$ q. c4 B: Q. H8 t
28. CString   description; //   Port   mapping   description
    4 [$ U( _1 d% H7 U1 y( f$ }! L" {
29. }   UPNPNAT_MAPPING;
    : W( M9 v" T3 _4 M! W
30. 
    0 g0 F9 m7 o. A& d8 f4 g
31. MyUPnP();
    
32. ~MyUPnP();
    1 e' T% ?8 C( R* s' I
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! _0 Z3 _1 I" t7 l$ Z' Z% g
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    * x8 a- R) A9 J& ]
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    . v$ ]# L- @% S' g1 W0 l1 E
45. void SetLastError(CString   error);
    9 G+ {% \/ S/ J0 d) n! E
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    0 H1 e  h! \3 ?. e
49. bool   deletePortmap(int   eport,   const   CString&   type);
    8 J" b, ?9 M# `7 \) y0 \
50. 
    0 e, n" b* {# S$ @$ Y
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ) {6 o! g9 t8 v- K- Y2 N% t
53. bool Search(int   version=1);
    $ v: b, C: n5 K/ w3 U5 ^9 {
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
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);
    6 x, T$ v! c: i! ^- M8 K0 W
60. CString m_devicename;
    6 h2 @" r3 p6 Z. l- h" C
61. CString m_name;
    # F8 R: [: P: k7 K
62. CString m_description;
    
63. CString m_baseurl;
    3 @. U: P* o, k6 V
64. CString m_controlurl;
    " p) o! A* S# B
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    % \  S( o7 j3 |) j1 k: o) G: T& v0 x8 f
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    1 e9 E& z0 W. ?' e6 X
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    # L- j: G8 H$ V( J, s4 K/ _/ R
76. bool isSearched;
    
77. };
    5 x6 T! O/ ?! D$ l  E5 G4 b
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   8 }" r- K5 i7 g
4. #include   "upnp.h "
   
5. 
   $ @- ^& F- T5 z  [- g- R
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   2 _( ^' |$ e3 }: Z( n% G/ \2 z
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ' W' f2 x+ [/ I( H
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ' E' _/ f$ g/ ^5 Q. Z
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    + l  N* R- H/ |6 S: Z( v, b
11. 
    4 z) ^$ q5 {! ^& T6 }# n+ i
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ! _  f+ e) e* H+ x- P+ t" m0 ^
13. static   const   int UPNPPORT   =   1900;
    # \2 k  E! G2 g& V7 `3 O! ^
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    7 T% x+ K& Y. Z9 _% E+ M$ G' ]
15. 
    + ^) c$ z( {# Y0 G, }3 ?6 @, q5 e8 s
16. const   CString   getString(int   i)
    7 ^- d* `  @7 l4 S3 e
17. {
    
18. CString   s;
    
19. 
    . e$ @) Z5 `. B% g7 X* O
20. s.Format(_T( "%d "),   i);
    3 M* @( A- w, ]' W9 V! K# s5 f
21. 
    
22. return   s;
    4 ^  H9 d2 q& X2 c
23. }
    4 R6 E- R0 Y0 f2 n# z, E
24. 
    & P% k8 ^" R- s+ b' U4 \* q6 T
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    1 h8 z7 O+ X8 h. ]  j0 E$ U4 L; a+ B
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    $ C! q1 c( g6 i
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    7 {7 b) z# J- ?% _. v4 C' Y* k
33. }
    
34. 
    3 I; B. s6 D+ w; @! |' M
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ' m0 p# U5 `; _% Y7 n
36. {
    % I# f% X3 y$ w5 V, r, h5 t& [7 G; U  W
37. char   buffer[10240];
    
38. 
    , p8 s) ~9 ]5 ]- T* z0 _8 h
39. const   CStringA   sa(request);
    + Z0 {% v) H8 ~+ h3 |3 M6 q0 K
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    - Y2 i! Z7 {+ s, X& A& v" F
43. uint32   ip   =   inet_addr(CStringA(addr));
    : F7 Y5 h8 X1 O  a& U- M
44. struct   sockaddr_in   sockaddr;
    1 d; \* l! o) R- @( S* s% C/ ?7 m
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    1 ~& F, j# ^9 I) g: ]
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    " Q, a) q3 G3 P: F" J0 I
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    * ~, W" V; n. k1 J2 p' E+ I
53. Sleep(20);
    ( o9 x. R. s' j+ ?2 u
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    6 c: @) z- o, W. r+ C  [8 P, i7 b  _
59. if   (!rlen)   return   false;
    
60. 
    3 r' s( @6 N& n4 F: }2 X& N
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    $ h! s% e9 I" \% q! b
63. return   true;
    
64. }
    - c* t* N; m3 i7 t8 j3 t
65. 
    0 t# l9 K9 I) _' Z' V1 F  x8 G6 h
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    " R) X8 i. q9 n5 X* T: J0 U8 X
69. 
    
70. const   CStringA   sa(request);
      W9 Q! M1 c& X. k4 g& z; ^  _; D
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    # {$ ~, V- t2 V6 n( J
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    # B; w& F4 O, X* k
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    3 z/ d% _7 P: `+ y/ K  A7 _; j
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 x4 l" \/ g2 E# ^
81. }
    5 I: f' Q0 _6 @& S
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    2 |# n4 O) K: d
84. {
    ( q5 f7 H, M4 x8 T$ K2 t; @
85. int   pos   =   0;
    * K; Q- P9 b$ L
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    $ ?5 _2 q$ H: _- T4 u
88. 
    
89. result   =   response;
    - d3 x  q4 K0 }' B3 I3 ~
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    3 E1 c- y; b. K
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ! G1 Z6 R+ C' a
97. }
    $ }8 v$ e, ?& p- R7 }
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    1 ~$ O; Y( F; J! i2 s! P
100. {
     & k, j1 ~' m* t# U* o" j- p8 K
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
       T  J  L) y( Z3 g& O' D6 c
103. CString   property;
     
104. 
     1 L8 D' ]& b4 g' u
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     & @3 Q( t# G( ~" b: _
107. 
     ) a" R2 _) P; J9 E2 i. ^  |
108. int   posEnd   =   all.Find(endTag,   posStart);
     6 c# t4 Q/ L' Q* B7 B' o$ s) ~
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     / ~4 c* o9 f+ ^. x1 e
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ' c, H/ ?8 A2 @- j; Y
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     ( B* \* v) S/ B8 ^1 ^
119. }
     0 [$ w5 Q6 |+ [0 I4 S0 w) U9 G- P
120. 
     
121. MyUPnP::~MyUPnP()
     $ R/ d1 ~, c" J0 m
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     $ p- x/ U1 S5 Y2 ?& q/ L
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     . p, k# ]% \5 y
131. }
     ! G$ t6 P4 M. e" {3 I
132. 
     ( L3 f/ Q8 r. W: P
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     8 z( B: a( ^1 k. {% M
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     ( }0 |; j- ?0 o% j& M: i/ _
140. CString   devices[][2]   =   {
     . p* J1 W2 A7 l, Y- x/ G7 f2 ]
141. {UPNPPORTMAP1,   _T( "service ")},
     7 u7 R  u& n: }/ V* S" T1 D, }
142. {UPNPPORTMAP0,   _T( "service ")},
     $ F; i( q0 g/ _) ]; R
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     - u* d0 m1 ?) M8 @
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; K4 X( p% l2 A: b9 Y/ B3 i
147. u_long   lv   =   1;
     . E4 u9 x5 N6 d6 G& q, T1 T
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     " O/ q7 o3 V4 @+ h& K' F' f
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ' ~* z4 z( l) t7 y$ A2 U$ J
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);
     2 U( [2 J" u+ f) H4 b
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     3 [9 D4 Y) d4 N8 Y8 q5 V
160. }
     / h! t; t6 x, W
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     - Z" j. [: A  b- w/ {$ j
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     * e9 \6 K4 u$ {/ ^) K
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
       ]' D; L: d. D, B$ ^
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ( x0 {8 T! y! T5 ~& }( W) A; D
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ! j! P- q$ h" {# F! |0 }7 [  F
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     , u& I- `" D9 ^3 ]' I
178. if   (line.IsEmpty())   return   false;
     6 X6 m$ D/ ]. E) _& B4 ^, y) j7 q
179. CString   name   =   line.Mid(0,   9);
     & R+ S, `4 h) }$ p/ P, G0 q; {
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     5 y8 I! ^1 ~. ?" A1 J
183. m_description   =   line;
     3 M/ ^" c9 L0 `$ U3 N+ b/ J
184. m_description.Trim();
     " K9 C; P2 x1 {
185. return   GetDescription();
     1 q9 U* Y! b" {2 A; a5 D
186. }
     + W& M" g' E# o3 P1 `+ v: ^$ H
187. }
     
188. }
     
189. }
     + K  |& S7 s# y. o
190. }
     ' Q- g" ]) S+ S. C7 e
191. closesocket(s);
     " t4 @% Y9 L2 o  [7 D, A
192. 
     2 K6 E/ [+ f( ^% I; _6 X
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
4 a6 k$ W0 |* ]  q( I0 q0 t

5 O3 |. {% o$ @/ r0 Z///////////////////////////////////////////
5 f, u, o1 @( H//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
4 J6 e) w9 J8 l! [4 X7 B" N

#pragma once
#include <exception>

; H5 j8 o/ f' ?4 U
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
7 h3 U8 U8 Y% O) G( e# {

enum UPNP_IMPLEMENTATION{
# c2 l7 n4 _' u        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
* o  @! V6 M$ {  G2 @        UPNP_IMPL_NONE /*last*/
};
) A2 w- ?, b/ G: l0 V- D1 u



class CUPnPImpl
5 V) c4 P. {3 T  f$ y{
$ Z$ ]3 b2 K$ L, S1 Ppublic:
% H' Y* w! d1 Y6 N6 ?        CUPnPImpl();
5 B* u, B4 t: U+ i% |2 |8 O        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
, U) x* r8 B8 n' L; K: K  {' _                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ T0 i( \, y, d        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
$ d# y" o! N, Q- z! X' ?5 w        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

2 d8 Y% N. s! _2 F
7 P6 k* i" m0 p/ A        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
0 Q6 y' n  q1 H* j        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
- q2 O# q- z$ O+ P9 x5 A: [

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
4 P) A9 ~) C- z+ V
# @4 g( E. M6 n$ t3 y, c/ t
4 U: P5 W: F9 s% \. I# w' Qprivate:
. [3 n  @) U# I) K; P0 n: Z        HWND        m_hResultMessageWindow;
/ n! p( w# \2 y* _% V0 z/ T+ B        UINT        m_nResultMessageID;
8 J& u) W* @3 c: A
- h" ~6 e4 `" o7 h* c/ E
4 @4 G3 g" H) E  F};
: H) S5 n8 ]. g+ a& b/ h
3 [7 b/ n8 |7 t/ ]0 P0 v- W3 R
& g& q9 s3 C) y1 t4 ~# l) z$ `% _// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
3 V* ?" Q6 U3 @6 n! H0 G; rpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
9 C/ j& \" u' w, ^- L+ _        virtual bool        CheckAndRefresh()                                                                                { return false; }
+ N( l. X" }" \  O2 y2 ^        virtual void        StopAsyncFind()                                                                                        { }
* o$ e7 G  B4 @/ N, i        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
8 f: j: {% f. {

. c% P/ e0 S& R; Z! u) M% j$ p; [, L/////////////////////////////////////
  W  @1 |! k( A. h1 q- V//下面是使用windows操作系统自带的UPNP功能的子类

3 A% q& |4 H9 ~8 x: h2 y
9 U0 y/ @8 }0 Z% U9 s# d#pragma once
* s2 w8 ^# _7 v8 z# ]! m#pragma warning( disable: 4355 )


7 Y- i4 g9 ?1 F( F9 J$ L/ p+ J#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
/ r1 |1 c: S8 t3 I9 D2 }5 }
$ F2 ], u' ^% @
0 |$ p* o" j# c3 S0 A, J: R#include <vector>
#include <exception>
! D- Z4 ^: F$ x7 o# K#include <functional>

/ t8 w5 L" ^# ^% o: T6 S0 |. F


3 {5 j2 B! |6 A- J$ ntypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
, h2 g9 E: ^; s% s8 D; D% |6 qtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
  K; b" o/ H7 Y6 _typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
$ d% O0 T# ^. I5 n; h; H4 Dtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
' B- u. A: t  ~$ \

3 R1 N; q2 Y: Q$ R( {" ]( wtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
0 G  `# Z' V/ P% ~! i* Y5 m);
4 n  y$ u& E1 t9 M

4 W% U3 Q0 d" ~/ f2 W  _5 `: l2 t' {typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
4 E& s! K7 @# {) h  PULONG pdwSize,
  BOOL bOrder
);
# V% F+ ]' ?" s) \0 K& w3 W0 Y
, G0 Z* ^: G- A
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
' `1 E6 F& V% h/ ?2 Q2 N' @$ _
5 Y& x3 X; ]4 x& y. v
CString translateUPnPResult(HRESULT hr);
- z" Z5 X" @4 u1 M4 W/ G  p3 L2 XHRESULT UPnPMessage(HRESULT hr);
. M6 d, H" Z; _1 A1 n- s( U

class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
. b5 `3 B6 [+ x* b$ upublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
- M; `! o' m; y
' I) x$ i" m. Y( |
: n' V, q) O* o" j/ g0 U        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
  u$ M  K( L; J; W1 ?! Q        virtual void        StopAsyncFind();
7 p% D* a4 R( L1 v        virtual void        DeletePorts();
% j  i" _  L! q) m; _4 i: |) x        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
& n8 `9 g$ C$ u, a+ I
2 F0 Q& g- U, `( }( D7 U9 A) }, c
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
7 X+ y3 h( ^6 M1 x- q2 P        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
! m8 w' t% L0 Z& P) C  a: w: \        virtual bool        CheckAndRefresh()                                                                                { return false; };
' X. N- f8 i; b; H7 y. ]# i
& W4 C3 Y! [$ W  i% i5 L# ~9 O5 J" \
: T  H- R5 y  i7 n  @protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
# j: V2 M1 M" a' |4 d0 \        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
( T/ y# q% s3 R) a- g        void        Init();
1 k. V' B2 s# E6 m4 N3 i: |1 E
+ P; j4 S! ~9 M! F9 Q
        inline bool IsAsyncFindRunning()
        {
" c3 D/ @$ r: _: W/ g. ]* @; V                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
( V9 X: W! j# q* X4 {2 y                }
' R. _  Q- j! ]# v" s; p                MSG msg;
* i0 F: P/ @2 ]# q2 e  f                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
3 ^4 O1 d8 d) e5 \/ O! G                {
                        TranslateMessage( &msg );
. J+ B; m) m& L" n" U                        DispatchMessage( &msg );
/ w% k- q- h4 {$ C; _                }
                return m_bAsyncFindRunning;
6 Q+ X" b! h6 f" I3 y! ]! d6 [3 z" p        }
* [, M- N+ q$ t: R
" T) e" P. C4 O/ I) f. h# F
; H: ?/ B3 o$ _. s        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
        // API functions
; u: h  O1 Q* c' k3 Z0 s        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
* v' V* D# \+ Y4 q1 M6 B: q2 M0 g        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
  Z1 u/ ~" [5 b: Y0 A        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
: `4 }% O" F8 F; E- w& `9 p        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
, `4 q7 t& @1 {" S
: h8 I( c) g. X1 u8 M
3 C, C7 ^3 Y3 P  T        TGetBestInterface                m_pfGetBestInterface;
. a7 K7 x* L, A6 j        TGetIpAddrTable                        m_pfGetIpAddrTable;
; H. {% G% E7 D) o( F        TGetIfEntry                                m_pfGetIfEntry;
( Q. C5 @( u8 p" h1 g; {

+ A0 `. I2 o/ M; o* {9 }  j: I0 m        static FinderPointer CreateFinderInstance();
: ]5 K, \6 A$ E7 q        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
7 o0 d# ?2 g7 V! d5 B                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

) n$ {" q: j( E2 l& m  v$ H7 T
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
/ @- W* c& N" x5 Q! e6 W                CComBSTR m_udn;
        };
       
! v# ~/ D# h  o% _        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
6 Q- Q( ~7 S% |. M; n        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
- A4 i8 j, B9 S  Z; ]" Q& z  r        void        CreatePortMappings(ServicePointer pService);
( d4 c+ Y) E  O( E. f) v* v* q1 G        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
  ]- I! B1 A3 H6 D% t        void        StopUPnPService();
' X+ k( [8 H$ W4 D1 s" y

$ ~# t& Y$ o: ~' L1 e* h3 w/ W5 w% k        // Utility functions
( D# {( y0 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);
; X/ P! M* O8 `        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
3 c0 e/ c4 H' h7 r        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
5 C8 t2 Y" a* x! B5 L        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


. ]/ f2 n! l3 K// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
+ u- `: I+ O  k& `( {& j        FinderPointer                        m_pDeviceFinder;
. U/ v' H9 E+ T- E5 n" [  {! Z        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


% L5 ?8 a# r' i* g# k& ?0 \        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
, Z  e9 c% I* E. M+ `# F5 J/ v        bool        m_bPortIsFree;
7 ]  m0 s* {' ?  @        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
/ |2 |& Z3 j9 ]: e. u        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  K7 w4 x$ K; x* N0 F1 n        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
) V2 r  x; [  r+ [% d+ M+ Q% `        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
% s: X: O1 ]: v  M4 C0 c4 @8 s        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
6 K4 |+ K; l' z* u+ T
7 a+ J9 Q9 }; a# Z& t; B/ |% u) Q
};
- V0 s, H  v+ J" Q2 [9 s9 \+ C; \

2 o9 ?" c2 A; f. E7 L// DeviceFinder Callback
4 n) `2 ~- l$ w+ {class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
1 b+ r; Y5 H4 [( Wpublic:
# W9 B  g& }6 c$ [; @9 k        CDeviceFinderCallback(CUPnPImplWinServ& instance)
+ Z6 h, M8 ]+ p( D0 r                : m_instance( instance )
1 t8 ^4 U# |- E- P; F- n. `        { m_lRefCount = 0; }

8 L# f4 O5 Z: k- `  Z7 n' Z
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
2 W9 h$ R9 M; Z/ B  \: z   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
- `0 [; r/ c- T* M% R1 I
1 {( `9 K/ `" G1 q8 C
5 z. k- w$ e# j% M$ S7 t// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
5 X/ r* |: `% ]        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
# J8 o5 F, x! p; G. m  c& a        HRESULT __stdcall SearchComplete(LONG nFindData);

! P' f* d# n  k( O0 ?9 n
private:
. d. v9 F% J  Q, K# u& c" y        CUPnPImplWinServ& m_instance;
6 V: L" u; s$ h8 k        LONG m_lRefCount;
+ @" V6 }$ H# @3 r: a};


// Service Callback
class CServiceCallback
; a, P1 N( c; m) |        : public IUPnPServiceCallback
. X. M9 m+ S* _8 {7 a9 @% ^{
8 _) s' V* R4 D( C9 K3 h: }public:
- r! D5 h, Y+ C  |        CServiceCallback(CUPnPImplWinServ& instance)
# n6 ]3 v9 Q& Y6 N1 z                : m_instance( instance )
, s; k: \: f  s" f6 z! y1 N        { m_lRefCount = 0; }
   
( f8 C, x, G2 c9 {$ ^& k   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 j( o& u2 @  U! A   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
( @3 E) ^! V5 a/ h- I
* \7 X$ }* ?& t1 p" R
  x" D) H. A: m' O) r7 {4 X// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
. }2 ?( h$ N1 u( J% D2 u( l

2 V" G7 h( `  U0 @private:
        CUPnPImplWinServ& m_instance;
8 `" ^2 f% l) k3 V5 V. I+ ~( r! S        LONG m_lRefCount;
4 m5 v# `2 ]$ D7 r};
7 Z2 n- V5 k9 k
+ x& Q  [5 F' p) M; u$ h) b
8 `2 O: b% P5 }/////////////////////////////////////////////////

! M! s+ M1 J/ d2 e7 x* M& e: n. b
0 N; O# P( x$ |4 I5 j% G使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' ]: Q6 F# K! }' V' I2 _" t8 hCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
7 J5 _+ g; @9 k3 ?1 aCUPnPImpl::DeletePorts删除端口映射.