UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   8 g0 a  q/ `1 X( E8 t, \5 r( j& l) _
4. #pragma   once
   # ~6 S; B# {8 U) a) |5 _
5. 
   
6. typedef   unsigned   long   ulong;
   3 [+ I3 t, |7 `
7. 
   9 P: {1 o& Y& J& H+ g5 w
8. class   MyUPnP
   
9. {
   ' r: ]& L* Q- d0 v+ P
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    - x0 {4 ]' N1 m) K
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( E6 n+ u9 u9 B' I3 h" W
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
    4 \  h9 h1 v9 s  {
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    4 ?' V$ b- M# y; ^
17. }   UPNPNAT_RETURN;
    0 a1 o) Z% z& O8 F% z# V/ v9 R
18. 
    5 h' e( O9 T; L. `0 V
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    7 c$ V; j8 w) q6 ?4 b
22. }   UPNPNAT_PROTOCOL;
    : E! V" n7 Z+ x6 c( o  K7 t# V2 w4 `
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    8 X& x  g8 J7 N% Z8 P  }
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    1 t# U8 t2 k/ c+ A' L' f& I
29. }   UPNPNAT_MAPPING;
    9 M3 J+ i& N: `, T$ V
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    3 D1 a+ D2 P2 s. O& w
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    5 d; W- C/ f- o, N  |  ]! i4 s
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    0 P6 G& X3 J* b0 \5 [- a
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    # ?5 [* b3 K1 V- s; S. E) v' V
41. bool IsLANIP(WORD   nIP);
    $ j3 ]% ]# h% K( E8 q) Q4 `5 K
42. 
    * F9 O: a8 u/ x9 p( \
43. protected:
    & }4 V7 X' P' z/ a
44. void InitLocalIP();
    & S* o1 u4 _5 q1 q5 q
45. void SetLastError(CString   error);
    
46. 
    ) F% s8 Z5 R8 c% r% d: G
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    - v  y0 d+ M* Q8 a7 ]; b$ W$ u$ F
50. 
    $ `- q5 J, I& S. Y) F+ J2 `
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    & e; l& n0 K& {  _
52. 
    % d  N+ _& B- {9 i) _- W+ o, Q
53. bool Search(int   version=1);
    / c- [: b  D. n9 n. Q5 e1 U) p
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    ( S" h; Y6 _5 m
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ( Q0 _8 @! n/ |3 b% L/ j6 u/ Q
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    1 I* o! K4 ]: M3 e
60. CString m_devicename;
    
61. CString m_name;
    ( ]6 F4 t! Q5 P, ~
62. CString m_description;
    0 |% W0 Y% o/ `, O8 I2 J8 S
63. CString m_baseurl;
    
64. CString m_controlurl;
    3 V9 }% _& K3 I0 I: }! P, a8 o- i( Z
65. CString m_friendlyname;
    # B$ W$ {# K/ x: Z8 J
66. CString m_modelname;
    
67. int m_version;
    1 C3 F+ k. e9 r# ]
68. 
    
69. private:
    7 e* N' r9 g% o; S, F  O# D
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ; m6 j! v4 V$ v$ D) y. J+ B
71. 
    9 f4 r5 k: Q! i# A/ |5 `6 E
72. CString m_slocalIP;
    ' q" r/ S4 R2 _8 ^5 B
73. CString m_slastError;
    ! P+ i! [4 O. G+ p. Z( ^
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    4 b" W  q$ ~( P! l  ^5 ]( a7 a% o
77. };
    8 m) P0 u; D* y5 v' J
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   # W- m7 s% K9 V4 H3 X: A
3. 
   
4. #include   "upnp.h "
   / @1 [: l/ p* X8 t' N0 h
5. 
   ! M4 [, O4 q& v; W- o
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 ")
    % _) A/ j! @. j
11. 
    0 b1 |3 N! o4 [) P
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ! U% X) |0 z2 U1 `  W
15. 
    
16. const   CString   getString(int   i)
    
17. {
    - N3 }4 V  b3 M5 O" A' ?
18. CString   s;
    ( X0 B( Q, Q% S: |+ ^# a
19. 
    " A2 z8 ?, o: H0 T
20. s.Format(_T( "%d "),   i);
    
21. 
    , f$ q0 o4 x$ o( U! u
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    / q9 C( S( R2 `" S/ J  l
26. {
    # |! b  F' P# w) P
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ! C/ g- a& _# a# \% F" N4 P
28. }
    : q: B; }) ^  Y$ F( g# @' L+ X
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    + T. P0 ^4 F7 ?4 T; n/ |* N
31. {
    4 A: Q. I* t& O+ C" D1 ]
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    5 H! e( e$ `' G" r( N
34. 
    6 ?# _* J" Q* Y: H/ ]8 Z6 m6 H
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    - x: e# c8 c- y  F: |0 d9 U% e
36. {
    
37. char   buffer[10240];
    
38. 
    8 _3 C) d5 O8 N0 h1 l
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    : {% o7 E1 h9 ?- P: \( d) x6 r
41. strcpy(buffer,   (const   char*)sa);
    6 |9 a+ g  n# e
42. 
    + M8 |5 X* j1 @; S( k
43. uint32   ip   =   inet_addr(CStringA(addr));
    6 f- L- h  R* `4 k) S
44. struct   sockaddr_in   sockaddr;
    & o# s) B6 Z( m1 ]8 y: x, Y
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    ; s8 T9 W+ N- j% _6 e# Z7 y: L
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 v1 P$ a1 g& v3 r
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
      e5 ^! z2 `/ D  N" ]8 B% `$ I
53. Sleep(20);
    3 F1 T" g' X! P  L) I
54. int   n   =   send(s,   buffer,   length,   0);
    0 C( b: Q0 r& N* D8 k; S, h8 u" a
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ) d1 ~+ S" J+ U
57. closesocket(s);
    ; j) M4 g, @5 `- G4 W. B/ \
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ! w7 W0 m+ u1 I6 n2 G  _5 X) N
60. 
    . d/ t' t( k! p" B3 c
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    7 {, M3 v, }* T
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ( x5 Z5 o+ {  W8 f
67. {
    
68. char   buffer[10240];
    , i. K0 V* T0 h
69. 
    % U/ S3 ^1 c" j; s( `( E
70. const   CStringA   sa(request);
    + e" o2 B9 [! t1 T9 {' w9 K
71. int   length   =   sa.GetLength();
    ; E3 c& a' C/ G2 w$ g
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    ) {! u, x3 m  H7 b4 m9 U7 u
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    * b7 t8 f: ?5 s: ]" ]' b
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    % h/ \6 r2 ~1 B0 Z( o' I' o4 G" @- X
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 Q) N4 x+ B0 P
81. }
    . d' O: ~4 G# G1 M( M! O6 @
82. 
    8 \% s& p% c$ @/ G- I4 Q$ i" p$ l1 P
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    # C- \  I" N% x
85. int   pos   =   0;
    
86. 
    2 x# ]) f0 h/ }0 Q8 m- j- ?, P
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ( D- S; N8 [0 f
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    " m4 k' {8 @3 k: S0 u2 z
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    2 T+ S& T3 L. ~2 j+ F, K: _8 E
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    . ?( B; y& Y8 _4 E
96. return   true;
    
97. }
    
98. 
    $ t  D# R4 a8 r$ D
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    5 |% x3 @0 l  V. ^) L
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     . w& a0 ?0 T! L" ~
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # x* w: A$ J5 M3 T6 j
103. CString   property;
     
104. 
     1 Q2 {* H; P4 j% Z1 |
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     8 K- B3 b- G8 L1 _
110. 
     7 S# \$ c6 R; ^0 t- a; z1 u
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     1 M& d. P7 v9 ]' I) F" [9 y' Y4 K! Y
116. {
     
117. m_uLocalIP   =   0;
     % J1 c& V" [" U( @
118. isSearched   =   false;
     
119. }
     + r  B: P4 O9 j! `- Q# n  \, m
120. 
     ' L0 }8 c( l* F' I8 f
121. MyUPnP::~MyUPnP()
     
122. {
     ( \6 a4 ]5 |2 ^. o( k! {
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     4 K# Z+ C0 ?3 e% g% M$ s' U$ G/ Z
128. }
     5 p4 @2 O  b# X: p
129. 
     3 [, ~4 _  B: I$ A! Y7 R( h
130. m_Mappings.RemoveAll();
     . k, P/ c2 `, T* m$ D, s
131. }
     
132. 
     
133. 
     0 H) T% T! z9 {: Q+ C0 U7 b
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     " @! V9 K8 p; J$ i% Q2 R2 c
136. if(version <=0)version   =   1;
     $ ], f  d( q2 S
137. m_version   =   version;
     4 m2 G( Z1 B: w! w9 a6 H
138. 
     # I  j3 g4 I# l% p0 e
139. #define   NUMBEROFDEVICES 2
     * p( F$ u8 C, I( n
140. CString   devices[][2]   =   {
     8 D9 }- r6 G3 W+ ?8 f
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     # J4 T+ e9 }5 y4 w  m% K
147. u_long   lv   =   1;
     . e5 Z3 M" x) a! \, Z) c1 H
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     3 R* p5 y) J) ?) h
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);
     
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 "),
     - `- m7 V2 i; j+ ^! S) ~- ]
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     ; @5 M4 t3 i! q) M" v) w
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     ; U0 s+ P8 C1 x0 D
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     5 p( X4 V! |* A0 ^2 r( \3 P0 V( \
167. closesocket(s);
     
168. 
     5 G* b, N! V+ ]
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     * f: \& q9 i  K2 l
171. if   (!parseHTTPResponse(response,   result))   return   false;
     # A8 w* D, t8 E9 R
172. 
     " ~/ `: M/ e+ ^& _) D3 T0 J
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     9 `! y# F' m7 K% J4 i
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     1 E2 ?" E4 R% V: [6 o
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);
     
180. name.MakeUpper();
     " L) f% [- j0 E1 h
181. if   (name   ==   _T( "LOCATION: "))   {
     + D% M0 J4 j1 R( f7 B+ O
182. line.Delete(0,   9);
     . M& ^: g: L, \8 [: l% o
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ! T, i2 t* B3 @- j6 ^6 o/ ]
186. }
     ; i: J2 M- U& c- A0 K  L
187. }
     . ^' b' x; G% i
188. }
     
189. }
     
190. }
       s1 }9 q( H2 M* m6 p8 x& }
191. closesocket(s);
     $ N7 d" [. P# @: |: `0 g
192. 
     ( ?* t6 W- u- ?& q
193. return   false;
     / T5 ^; u% v. T9 e7 m
194. }
     2 J, D. O6 U1 S* F, W! h$ J

复制代码

天云上人

以下有关upnp的接口来自emule，
# I* I2 R9 O4 ^

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
6 o% B4 B3 D) x8 }6 u' [" `#include <exception>
  D$ ?3 B" I& T5 X
' b  O- j3 v4 Z# K" P3 X( W$ b1 q
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
' s) R# o2 e+ v. N6 l        TRIS_TRUE
};
& Y& M  n# f6 R) ?2 j( P
' W. h1 j1 q$ v: N* a- `  x* G
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
/ H% Q3 G: w% ]4 w' H        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
" b7 z, F1 Z0 E/ l; c9 ]
) C3 ?3 Q7 J& v8 I


1 V: E8 r4 e; Q1 p, ]class CUPnPImpl
{
3 g% @% s, N/ b0 H& [4 Tpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
% E+ M! O; t/ g0 p) i* I9 K% w# p        struct UPnPError : std::exception {};
        enum {
  G+ F4 o' \3 g  [                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
, {( ?( \7 \. e

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
9 I: O( r# W- H+ W7 Q        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
+ Q2 F2 E  K( A       
) c+ z0 G0 R1 T8 B( C        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


3 S9 A: \- ?. N, E; G        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
) F5 D0 Y/ x4 y" l3 H, \. D
/ k/ k  c" U, ^) W& T
  S) {  a3 C* G* Y// Implementation
: j' z) d  ]8 q% C( G; W0 W+ Uprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
8 K, N0 ?8 x$ _1 ]- E2 I+ H        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
0 i$ E# w$ w- m( f1 K$ M        bool                                m_bCheckAndRefresh;
; N' l  V3 [, u9 b5 w
$ l& V& B$ m: r# A8 q; N
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
; b' @1 B" e( M! b: ]
( Z* W4 U$ g& o3 }, {: Z
// Dummy Implementation to be used when no other implementation is available
& y" p# Q. C3 L7 Lclass CUPnPImplNone: public CUPnPImpl
% h6 d0 L  B5 J, H$ e8 F{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
3 L5 K1 P: O# O        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
  N9 H3 M+ s+ d" d6 K: k) ?% n        virtual bool        IsReady()                                                                                                { return false; }
; v# }9 Q+ \; V0 ]* Y6 c# G* u' x        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

0 D/ v* r, }1 ^, t, G2 J) r$ o) o
/////////////////////////////////////
9 t! c/ i0 J4 Z; X& l7 U3 _8 T//下面是使用windows操作系统自带的UPNP功能的子类

) `( a3 O2 @1 o1 ~
# Y1 i1 R# o$ A& |" Z% x" c#pragma once
#pragma warning( disable: 4355 )
) V* c" f; z( @- B( D  [

4 V$ H  W1 s4 ]3 @1 S' j. {5 t#include "UPnPImpl.h"
#include <upnp.h>
7 j7 {& d0 @0 J4 c9 J& g0 L#include <iphlpapi.h>
#include <comdef.h>
. o8 H. ?3 C! U7 D0 M' o, t#include <winsvc.h>
" k% y6 ^. ^7 S  Q1 i
! z. l" d1 \5 F8 ^  z
#include <vector>
: u5 U  X2 D* _5 B#include <exception>
#include <functional>
0 f8 i, P/ R4 l( i; h# L& M

( B% h. L+ ^6 e3 x* }( ]
# C4 G4 R* r6 ~9 I% R
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
7 R1 l$ ^2 v: T' [typedef _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;
$ z; r: `" |3 _$ Ltypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


4 p! j5 T7 o2 r, `# f. @typedef DWORD (WINAPI* TGetBestInterface) (
$ [% v" U7 y. l% `# a# `  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


1 O5 _" s9 T7 J' Etypedef DWORD (WINAPI* TGetIpAddrTable) (
5 y) `# ~! s. i. \% d  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
, B! Y% H9 o8 w1 d) Z2 e
9 k5 V2 [! s; d: g# R( k
! V  j5 Q( X$ a3 u3 B6 @  \6 x) Ctypedef DWORD (WINAPI* TGetIfEntry) (
" P- O, P/ |; H" F4 J/ v9 u  PMIB_IFROW pIfRow
);

& b; f) ]/ A9 ~; p3 u( _, ]
6 G/ r7 u" e2 A' A2 p: T0 C, d# JCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
3 P* E! Z6 U9 v* _& y+ [$ F        friend class CDeviceFinderCallback;
( J, x% U& u  U4 b0 J        friend class CServiceCallback;
7 D) ]0 o" B6 d5 R1 _) I// Construction
public:
) Z, x! g" q5 `5 R/ i+ {% V        virtual ~CUPnPImplWinServ();
( g3 |2 |" T  K        CUPnPImplWinServ();


; B+ K  T* U: u. W( Z2 Q. Y+ F, e        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
! I7 i# P, I& n- q5 o* j7 D        virtual void        DeletePorts();
        virtual bool        IsReady();
" A; k) V; W3 }2 K# [        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


  Q+ [- ?: U6 B        // 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; };
$ W& V" x" ^* C% B; b3 W7 p+ Y

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
; c6 I1 q7 ~9 D* o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
, a: `4 @( i& z        void        Init();

. \( F3 T& n; f& T* O4 e* M6 q
        inline bool IsAsyncFindRunning()
        {
. |  k/ U/ ]6 c- L                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
: E! v  E( z+ l                {
7 ]5 `; M5 v* q7 K$ N% t) _                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
+ N" a/ N+ e2 p$ M( n                MSG msg;
* Y, f: C  B6 ^' l/ L% K                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
8 J9 a* y* o" Y& ]0 W4 Z                {
& a2 X+ n$ }. {( V: y! Z2 k# g9 }                        TranslateMessage( &msg );
& \& O7 E) Y6 {; G$ r                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
  t$ B) ~+ J9 ?( J) J) n

& k# ]3 l) O4 K. D& @* T6 o        TRISTATE                        m_bUPnPDeviceConnected;


! j3 r6 m# T* p// Implementation
% q  k5 k& x( }- ^! F2 T        // API functions
. y: B6 n; Z3 [: u0 S- q; E        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
" u3 v8 z7 c; ?4 V# p" t9 n        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
% t7 D5 q; d$ k- a        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
# x. A% f% w  ?

        TGetBestInterface                m_pfGetBestInterface;
$ x6 m" o; R) q! W        TGetIpAddrTable                        m_pfGetIpAddrTable;
5 A- ]  u8 E! ]+ E: ^. Z' U        TGetIfEntry                                m_pfGetIfEntry;
/ K3 g; h2 P& r7 V4 w
% m- N. N) K! `. T
9 j, a! _) S+ c9 [- [: m        static FinderPointer CreateFinderInstance();
$ l; b7 Z) a) q: T+ l. v9 e6 L7 g8 E        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
9 I) |. M! K& q  W( g& [1 n  o: G                result_type operator()(argument_type device) const
                {
# a6 O( a2 H% o5 F% m. f                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


: S5 c& U& Q% B) Z* R                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
5 h6 i6 s! |9 x# J6 d$ m5 N
+ N, h7 {% K8 u& B9 X
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ k) k7 Y: Q  M" G' ?- ?& ]                }
: }! a. @/ S0 v8 G3 }' q  }  e                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
! {" F5 V8 W9 ]! b/ H        void        CreatePortMappings(ServicePointer pService);
$ R( d  J( I: a: {7 N        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
7 l: t+ t  c6 J        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ O. y: y0 _" j' {% e) s( F  c$ D        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);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
8 D! S6 Y( e; G- n8 `3 g* a        CString        GetLocalRoutableIP(ServicePointer pService);


& U% s0 J/ S. P% O. D+ A// Private members
0 r& I/ g8 L9 l0 T2 `private:
# n- k' p& F1 v: D& t$ f        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
1 I% R2 l' T5 ~! `- E- H% p        std::vector< ServicePointer > m_pServices;
: I3 @% @' h" @" w/ G6 s  b        FinderPointer                        m_pDeviceFinder;
0 i/ U4 }8 D1 z  g2 Q$ u1 [$ b        DeviceFinderCallback        m_pDeviceFinderCallback;
5 n4 h7 g8 G3 v8 V7 C8 x% |& {( J        ServiceCallback                        m_pServiceCallback;
$ D  e# {" N- J

( @4 h8 V/ Z% `1 P/ X% h# Q        LONG        m_nAsyncFindHandle;
% q5 `* [: r# V6 b# I        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
) K7 n# ^6 _3 J  n        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
4 x  Z9 ]/ ?4 K& {& H* r        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
" [6 m7 Z- Q. l        bool        m_bDisableWANPPPSetup;


};
( _6 m. J, u9 {8 c8 P

// DeviceFinder Callback
class CDeviceFinderCallback
5 I4 }% L/ i( [8 x9 ?, S0 I        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
5 f) t! M+ S+ J" q( G: w                : m_instance( instance )
        { m_lRefCount = 0; }
5 x7 e, z  K6 ?% R: s2 `( s
& T9 h+ b% Z6 j  X/ |
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
4 _: a: L+ U' U' k' h. c' B5 N   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

, p' e: Y6 l& @( m, Z, z* w
// implementation
private:
- `# k$ \' V' @0 j: W        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
+ A* t# J2 L) l5 P( a        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
/ ]9 m3 ]. k) L. `4 d
: w5 B3 p  {" Z( {. Z
9 r% i) F* t, d2 Iprivate:
& r% q9 }" S, U        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
2 \* O8 j; |4 a: g; f5 C
2 y  Z5 W6 x- C# N8 m# r& h/ R& a
8 y$ H5 S1 V8 \9 Z5 e1 z// Service Callback
class CServiceCallback
" S- a9 f) i: \1 L        : public IUPnPServiceCallback
{
public:
, F! A1 ~! {- Q        CServiceCallback(CUPnPImplWinServ& instance)
& C/ b: U* d' `# m. h: h# R                : m_instance( instance )
        { m_lRefCount = 0; }
1 t' N% [, h0 ~# o   
( m0 e0 o5 p7 q1 H$ l& U  J/ j   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! U& ~0 k8 k( R/ ^3 C$ @% z$ s   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

5 C6 Y$ D1 U! H# x
/ Y+ _  O4 b8 D% n// implementation
) D: X' k/ y; S- vprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
, w, \( i8 O2 r6 d. G* Z7 |        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

% k- k* d% s7 g7 Z9 H) _
. B) N3 T# Y. Fprivate:
8 A* v2 y, c2 @: l6 T/ k        CUPnPImplWinServ& m_instance;
& V1 E; k+ Q2 e        LONG m_lRefCount;
% W3 X9 ]# W$ J- \};


; n, |! M$ A* B2 G. g/////////////////////////////////////////////////

% v7 I5 m0 g1 n* a
4 ?+ @6 X# z; P* G使用时只需要使用抽象类的接口。
; E* q$ z; B% t& h/ V! pCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
, D& ~4 t6 V4 j7 x5 m& oCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.