UPnP

天云上人

/*uPnP.h*/

1. 
   0 `( B) n6 M- ?8 K4 R$ x
2. #ifndef   MYUPNP_H_
   " u7 Y2 `+ y2 m
3. 
   
4. #pragma   once
   2 G- Y% x: n. O$ O( p) A7 f5 w
5. 
   ' q3 N/ l: n* N' m. K2 G
6. typedef   unsigned   long   ulong;
   , W( V4 u% W- A) {  d- F8 C
7. 
   5 B4 h; C/ i+ F3 C: n3 d. T
8. class   MyUPnP
   + V* e. b8 f# y1 H, G
9. {
   
10. public:
    
11. typedef   enum{
    0 d% D' N+ V2 E. C. |
12. UNAT_OK, //   Successfull
    9 y- t* a8 v1 F/ K$ v
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    : u4 v3 n) i2 E0 q
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
    
21. UNAT_UDP //   UDP   Protocol
    : C: T  d# i' }' E. a& J' o
22. }   UPNPNAT_PROTOCOL;
    : _* y5 r, |/ c' W# |) P
23. 
    
24. typedef   struct{
    6 I5 Y: F8 ]  w* F1 v2 W2 s
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    , U9 m+ v5 _7 a0 z/ O2 F* O
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    % |4 p9 [7 Z& ^2 y# _/ O; q
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    0 A  C2 m0 F: @. ^! G- Y; g- H
33. 
    0 P* a1 L4 T6 S' c
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    - o9 |+ T  v& F
40. WORD GetLocalIP();
    3 m) @% z0 E4 S: O4 @2 l5 ~/ A3 H
41. bool IsLANIP(WORD   nIP);
    $ u  `7 h+ O( R
42. 
    
43. protected:
    
44. void InitLocalIP();
    ( a1 ^: S9 Q3 g) c
45. void SetLastError(CString   error);
    7 \. [" Y, y' X
46. 
    ( w7 d/ ]0 X& }! V; \: f
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);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ! A, ~8 u1 G, O; y
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    ( _) r; P& x5 ~) E
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    / E' D6 \; n' q4 C# b! w9 F
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;
    ) Z, F- c" d, B; q/ E9 H; I
62. CString m_description;
    
63. CString m_baseurl;
    . I: T" m( `9 ]9 U. u6 [
64. CString m_controlurl;
    * l$ Z! z7 P1 I: F) C# t
65. CString m_friendlyname;
      S/ P$ B9 z) N/ \6 f2 B$ Z
66. CString m_modelname;
    9 l) K' ?3 M* G
67. int m_version;
    
68. 
    
69. private:
    : C: L: ~: ]; x* d& l* `0 }' [% U
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    * B0 g$ k5 U" I2 B/ L( Z
72. CString m_slocalIP;
    ) |: V: ^* s! f2 X  t
73. CString m_slastError;
    6 k, Z( g6 t$ d7 j" F
74. WORD m_uLocalIP;
    . }' {* F9 X  p6 }  b( W
75. 
    
76. bool isSearched;
    2 U& n6 [+ p& F
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   3 {' ^: A3 r3 S" C3 G3 U( G
3. 
   7 W* y& \3 y& [7 Q3 V$ A
4. #include   "upnp.h "
   1 }+ W8 N& i% F8 A+ [" B2 }: h/ X4 {
5. 
     J6 H. r4 y; t& p( V" b
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 ")
    
11. 
    8 [$ x5 H4 t" J2 L( ^2 G
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    0 v6 e6 Q8 [3 b8 M
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    " K0 c/ z) G" J8 ]+ D
15. 
    5 Z# B1 O& {/ C- k, ?2 O
16. const   CString   getString(int   i)
    8 J0 K- w9 ~4 b9 Q- `* _
17. {
    0 i3 l( g4 \  i# P3 u( n
18. CString   s;
    
19. 
    : {  B3 z2 C+ p! W) K9 v
20. s.Format(_T( "%d "),   i);
    
21. 
    1 @5 z6 }7 k7 O+ r8 X: T6 J
22. return   s;
    * I; O. O. \: m! ]
23. }
    
24. 
    ; x% C5 F0 o% Z8 A
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    : E% U! T; n) `3 f
26. {
    4 W. p. M$ c4 `2 `5 N3 h
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
      Z8 f$ ]$ E/ c, d( f- l  O/ G4 }
28. }
    ) n% P9 J6 I8 v+ h% G4 P" j* ^( @
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    $ n9 Y0 o8 d4 k
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    . x) f) }: N7 |6 ?
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    ' Q+ P2 E' S5 ~: H2 s4 k7 g) w
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    + e) O! n, `. y: c4 W
41. strcpy(buffer,   (const   char*)sa);
    ' J( ], J6 K* W) C9 o
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    9 @1 C# @, s1 J5 B
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    1 e, ]) Y' d' l' v) F- @
46. sockaddr.sin_family   =   AF_INET;
    9 {% l/ C$ C" H" k# G. Z
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    " T# H7 e" ~. |. D. i$ X' c1 Y
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ( G, z6 h  h# m; b
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ! k' I; W, _' g% {
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    , T2 ^4 n3 v! l5 m6 w4 k2 U
57. closesocket(s);
      j0 f0 X) q: o1 x; ]
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    9 |7 ^; e2 H" a! y3 A
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    . X' Y& g8 j9 ^; f- r8 \+ Y- a: P# [
67. {
    
68. char   buffer[10240];
    3 x1 ^2 e1 \! f/ ?  {
69. 
    . f! P4 _+ m& u" }: U0 B
70. const   CStringA   sa(request);
    ; M- O* I$ ?  H+ x
71. int   length   =   sa.GetLength();
    + `. x/ C' G! Y/ L. s
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    ) H% H! E2 y1 o' {# o7 @
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    , M3 \. t2 t5 W$ v) Y6 G' I
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ) M6 X+ a5 a4 R3 r
79. 
    2 M3 p; O0 ^8 k
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    + b) i# a1 s2 G# y1 u  X. U
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ; w5 ^/ _- @$ S: q. K
84. {
    
85. int   pos   =   0;
    
86. 
    2 L/ f: j7 N" q( N& p' ~
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    4 Z3 k3 I% Z* X) `/ d* N$ Q
90. result.Delete(0,   pos);
    ! Y3 p" p; ?5 b7 }5 h* X! [# g
91. 
    + L4 q7 L$ E$ k: n, D/ D) [' `
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    : v# d) ^3 j  C3 W* R4 i
100. {
     / c- O. w3 G' c3 L9 f* {6 S
101. CString   startTag   =   ' < '   +   name   +   '> ';
     7 `% |# e8 @' K+ F
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     * d" a( z, ^3 e- I( d5 @$ Z1 }# E
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     + v; X/ r3 x! f( g. }& `' u
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     % c& q5 W) {7 K7 J
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     ' s0 e: C, d( S" f! ^
113. 
     
114. MyUPnP::MyUPnP()
     2 {6 T& S0 j3 e- o4 `  z9 Q( c7 j9 R
115. :   m_version(1)
     8 d' F! `/ x! S1 T& ~2 [* E
116. {
     
117. m_uLocalIP   =   0;
     / C  q) W" h: E" @
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     , y  w4 p1 c1 ~  ^9 D0 V/ l" q
122. {
     
123. UPNPNAT_MAPPING   search;
     5 W* t; x" G9 O0 F" x# L
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     * ]; x9 Y( d4 B- D! h
125. while(pos){
     1 o; l1 c3 Z' V7 g- j9 m- w
126. search   =   m_Mappings.GetNext(pos);
     9 S! |# c2 t# q8 q
127. RemoveNATPortMapping(search,   false);
     
128. }
     * s+ y, F# O  [1 M% _
129. 
     3 T0 R* B# @8 _
130. m_Mappings.RemoveAll();
     
131. }
     9 F; I& t0 Y/ @5 ]# v( C( x+ C! A% ]
132. 
       x$ `7 J; y) Y  c: C; s7 G5 \7 b
133. 
     1 @9 U& h) g8 x5 v. U- k: k
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ( V7 a! S- [  n) @5 l$ g, k$ @
141. {UPNPPORTMAP1,   _T( "service ")},
     2 g, |1 ]0 x$ C" |
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     6 J: F: F: x8 f- B' \5 {
149. 
     ; z! n4 n3 c( `9 Y4 @: {# m
150. int   rlen   =   0;
     8 f+ r, P/ `/ w% Q
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " L* M: R. c/ V5 _8 Q$ V) i. q
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , x9 z2 r9 r4 R& m1 k- @$ }/ m
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 "),
     0 x* p( }' @3 l5 i
157. 6,   m_name);
     : w1 ^9 I1 @" a" w- r- I
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     # U) P0 ?$ Y7 u
160. }
     ' |! [4 S. s. f, t
161. 
       U4 O: `; y2 z; M: i. m1 q
162. Sleep(10);
     , ?* Q: @+ D4 N- G0 f& T* `
163. 
     
164. char   buffer[10240];
     : {& E5 p  s/ l8 {! R# c! q
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     9 V/ j$ r3 C4 U0 T
167. closesocket(s);
     
168. 
     / `& J9 d! j: L+ C8 u' f
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . b' V, v; t0 z# c3 l* |& l4 @0 _+ y
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 p8 M2 u; X' O# C
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     8 g2 U1 ?- n( h5 o
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     : x2 T6 ^8 l. X# A9 Y0 B
188. }
     
189. }
     " l8 [+ ?. J  N. H# M' b" z
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     . Z. t( h0 L% Y( w  l5 B

复制代码

天云上人

以下有关upnp的接口来自emule，
) G. a- W9 b# z- o9 y7 Y0 e
+ W, P; H$ `: c) w
$ l1 e$ B2 M& h8 u3 }% ~* f///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

6 Z0 O8 x- Q9 g5 X/ E7 ^# T
#pragma once
0 {# h* c- ~" x5 u% r: n$ t+ S4 M#include <exception>
) e6 P+ L* n' d9 b& u" O

/ H' |6 E( ?! b6 o, C  enum TRISTATE{
8 W& E0 Z0 e, a1 g1 h! ^! ?% @+ ~        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
3 n' s6 Q. T6 t& N) H        UPNP_IMPL_MINIUPNPLIB,
4 n! R4 V2 B9 N        UPNP_IMPL_NONE /*last*/
};

  ]( l1 i# L8 l  Q$ Y5 z2 j

- y9 _& T3 m, e6 k. {$ F8 p; P* x
7 _% o4 ?! Q& `7 pclass CUPnPImpl
{
) J8 f: M& A; N2 t$ fpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
" C6 m$ O! T% A% j0 w7 w3 d        struct UPnPError : std::exception {};
, Z9 l6 Z+ G" n2 u" ~        enum {
2 e# Y) W+ [& T6 t  w- H+ W# A                UPNP_OK,
                UPNP_FAILED,
. [# J' ?" u# Q. _3 M                UPNP_TIMEOUT
# A4 }$ b, b/ B        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
1 |  f% u7 G( L/ [        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
/ D4 Z7 q9 z& G: ^1 N8 J1 f* ~, ]        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

8 w5 N4 K8 o3 W0 J: K; V
& E( w  J' l7 G0 a6 [        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
# X* e- ]7 w# F! I# s3 z6 J        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


4 H1 X( z9 g. ^4 x$ N// Implementation
protected:
6 Z) M: K4 a' Q5 y/ k        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
# K0 \. v. |' ]" m" a/ r4 H3 Q( ~( v        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
& F3 A, n' e& g+ O$ W! A# O

; a+ n& n8 g, R  z& uprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
# [& O8 z3 I- L/ S
4 g3 k! o& Q0 V* c$ U
3 T' m/ p4 e$ P4 Q' `) R};
! e8 K7 N% Q7 \7 @
! V( l) T  i6 B0 b7 s
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
) B( O- t& x7 z' w{
public:
* t: Q# o1 f7 k: z9 P$ z        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; u# {0 \3 [3 f# G0 M# z5 s7 G- o8 C        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
" P, d; k* D: X        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
; k8 z$ t$ @8 G. \};

/ U! X  B0 ?% Q
( B/ i+ i& H& U$ ^/////////////////////////////////////
3 t4 s" M2 l' K/ y5 E4 S8 a5 G5 X8 s//下面是使用windows操作系统自带的UPNP功能的子类
/ X- L6 {- }3 k# m  y& I
* x3 z# i- @" J* r: d
#pragma once
! ], W4 Q- |/ E5 h1 E; v#pragma warning( disable: 4355 )

, F$ `  J8 y) W' g
7 a% P+ |$ z! L" _#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
( ^" G; _! Z' L& q( s. [
# I* B" O. R& b7 S( A
#include <vector>
) ?4 [/ _* c1 Z% W0 s' }" j#include <exception>
#include <functional>

9 j# ]* A& }8 n6 s7 k


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
( x. a$ f: X+ y# Mtypedef _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;
: l* p+ P8 L- jtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
& y. {8 O  C; i+ ~- }: T
  d3 U! f8 S/ K" |8 b! o
8 L3 @+ }( _, P( t6 jtypedef DWORD (WINAPI* TGetBestInterface) (
8 N# u+ U( L1 j# y2 k/ s, G2 {  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
/ C* C5 t# s- e  M1 {);


typedef DWORD (WINAPI* TGetIpAddrTable) (
1 V" S+ P0 ~8 ]" Z0 o  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
" }- G7 D/ g4 q# u  BOOL bOrder
);
/ z& D8 U$ c0 e' p5 I3 C+ R- k- V3 Z( j9 w

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
1 {8 L  W! d  m. @* z: O);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
, q# ~: f$ |' T" n. W9 g; H$ T
% b+ F8 l& v9 C1 g) p6 h
class CUPnPImplWinServ: public CUPnPImpl
3 S/ @8 e2 Z' K% F" l/ o6 g{
        friend class CDeviceFinderCallback;
; d: m; K/ x- G( r' ^        friend class CServiceCallback;
5 _/ x2 x$ R. v2 y7 _% I8 e// Construction
public:
, {* J) ^* H- D& k& J        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
8 |: ]! z2 U8 n8 x7 D        virtual void        StopAsyncFind();
% G: c) p9 W: \" @+ \: _, }        virtual void        DeletePorts();
        virtual bool        IsReady();
( Q+ h) s7 h- v' r$ ?        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // 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; };
' [! u" G" u# F/ {2 i
7 j* c* _  A8 N7 _0 |5 Z
% v; d/ f  }  T$ m3 S5 Eprotected:
: h3 L2 J9 f6 ^- u; P. u        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
7 q4 K# |7 ~9 a* o! Y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
3 d/ X* I9 ?1 o1 L0 `6 \) S        void        Init();

4 G5 Y# O4 i& ?+ q
7 H( D* Q# X' \        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
6 }. a' s) _# O* v                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
8 f5 v0 x) A1 z8 V8 G                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
2 r* A0 v0 s1 R1 |4 L9 ~                {
: e! U6 v( h3 `: F( \: Y3 w                        TranslateMessage( &msg );
9 {9 N& N# z: K+ S6 n                        DispatchMessage( &msg );
0 O2 B/ b* A. z1 o! Z" U, H4 h                }
                return m_bAsyncFindRunning;
        }

2 n& H0 T/ n& g1 p3 o2 t
        TRISTATE                        m_bUPnPDeviceConnected;
3 A6 p, E) Z& B) \
. L& P! V8 w5 E8 M! R. [( D" M
// Implementation
7 r% ]- v! c) h0 @3 C, g5 T8 [: H        // API functions
9 c) N; z; S6 p/ X8 h        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
6 O* q7 v6 _8 B" [        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
9 D+ M2 {1 H. J# d9 R: ]        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
/ R* @# E6 ~% _* i$ \" j" L

        TGetBestInterface                m_pfGetBestInterface;
7 e1 K5 t: j/ m* S        TGetIpAddrTable                        m_pfGetIpAddrTable;
" l' k( l" w- {: m& A* r1 v, W) F  h        TGetIfEntry                                m_pfGetIfEntry;

) a% Q* F+ }$ o
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
5 M" ?5 T: N% @0 m3 c$ }9 h( ^        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


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


) r2 C/ D6 o4 {1 e3 j. _                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
: h* I9 V" \+ P) P: N                CComBSTR m_udn;
        };
       
, c; R& ?3 e9 m. u& n        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
/ }6 M) B" s7 I* v        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
; m6 `, w1 W' T- N6 H7 {$ ^        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
+ H  h' z: f% Y, Y( k' D$ H: |        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 V+ @7 H" o2 u: e9 e3 p                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


3 U+ ?& J, U9 \  ^2 r3 Z& Z: u  g' q: U        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
9 L  V; ^5 d! C1 k( K8 h8 P1 Q        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# r8 X- v% q' j# }, ?, G3 X        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
- H! p$ f( l1 B  O$ L( G% A        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
0 k( Y# k0 |, {7 [$ _% G3 D8 O8 l        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
+ j# W& K2 q2 L        CString        GetLocalRoutableIP(ServicePointer pService);
& z9 ~: o) x1 n$ v

// Private members
% T  J8 u. i* b! [+ d; f: f. Gprivate:
2 X: O3 m- }4 d" n$ Q  l# N        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
: g. @' J. [- @  K' G        DeviceFinderCallback        m_pDeviceFinderCallback;
& q. c7 M( V( g+ P3 F4 @- y        ServiceCallback                        m_pServiceCallback;
- ^! G) ?( O2 F0 H( r5 n! M; u+ F
7 `5 y0 A, a$ }% @5 \$ }  m/ Z$ G
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
' y* k3 d+ i6 q! e. E( w        bool        m_bPortIsFree;
7 B6 L& v, W) H8 D/ d        CString m_sLocalIP;
        CString m_sExternalIP;
$ F# M8 X# f8 S0 k. R3 Q: o7 r        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
& `6 ~' J  g8 A4 t# ?8 i        bool        m_bInited;
! I: `: @& @7 E& w        bool        m_bAsyncFindRunning;
' P# s' l6 g* m5 V# K& t        HMODULE m_hADVAPI32_DLL;
9 U( X( Z- [3 O: X  j: c6 Y        HMODULE        m_hIPHLPAPI_DLL;
- j/ z" A: N! I( Q6 x        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
/ C6 Q% a8 y* `  [) u        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
, \1 p% V" E* N) S
* a( k# X7 [; B( I& r
" s$ Z4 ~- s1 U# C};


// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
2 t' x* y$ p( g' W) ]8 |& k{
7 ]0 T* l1 d5 {public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
( P8 S+ q! x* w! y% p5 S                : m_instance( instance )
        { m_lRefCount = 0; }
, s, d$ s  {. S
; q4 ?& B; `4 Q" `' f- _3 q0 C& {
3 N0 I4 [( |" ]( y   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 u( I( d# r+ q# I$ B   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


2 \. Z: ^+ k8 H// implementation
private:
( A# d+ W& J4 `( h4 L$ T* A        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
6 g$ r' ~0 [$ {        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
; `* k+ j: a" J6 a9 M5 C. C$ f3 N

4 B  M) t* s% b% ]0 |/ ~; \private:
) s! t! m. Y, j6 I! o! a2 i6 t: S        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
6 D% H) N* \  c3 c& f2 t};
$ a/ u. X% k6 _, r  n6 B
4 Z& Y2 _' r& |% Q/ z8 l) [
7 V, }; p0 z! k, H* t4 q// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
3 h4 H5 c6 C! N1 g; T! ]{
public:
# S; V" J& z: h        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
/ s: ?1 D  ?/ o8 ~9 [        { m_lRefCount = 0; }
   
1 y) f6 |# C5 Q+ f   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 m1 S9 O9 ?4 U3 F/ _) |/ W* U   STDMETHODIMP_(ULONG) Release();
: [4 m, S$ P- }% t! l% L. A+ p. O
( g& R( L# S, y8 C6 `# _
2 }& S, m' y5 G: b* W// implementation
private:
! h; n9 A% Y" A* b4 f3 H- ^+ a        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
2 o# R. j# q5 d- K        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

2 p9 T* a6 c. d5 A$ v- T* ~
private:
* E1 Y) R, W) c* r  Y        CUPnPImplWinServ& m_instance;
6 \. U8 u1 n/ a' V* z        LONG m_lRefCount;
$ a( v: \5 w) U+ U. K};


/////////////////////////////////////////////////
1 K' W1 Q# @3 M- ]: A$ {
6 ?% |5 R! ~# d( c6 _+ [- p8 X
( C) n( w% c8 t' _使用时只需要使用抽象类的接口。
6 ~' W1 J8 ~2 r2 O9 ZCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.