UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   ' p, Q* \0 b6 a6 |) J; U  W& `' _
5. 
   
6. typedef   unsigned   long   ulong;
     o9 n% h; x5 c
7. 
     j9 s5 \+ x% D* W4 h7 `4 H
8. class   MyUPnP
   1 `2 G  ~, l% v2 X: A6 x
9. {
   
10. public:
    + J9 e# q. \! d' D- V
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    % r* v7 A& B9 B. `
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
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    : R) R5 U1 v: ^% _8 X
17. }   UPNPNAT_RETURN;
    
18. 
    9 n) e# B/ A+ K" E. ]4 D3 t' t
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    ( j0 {! s+ b# G
22. }   UPNPNAT_PROTOCOL;
    
23. 
    + c: ^7 U- x& \/ q6 L" Q* \; E
24. typedef   struct{
    3 U& R% M! a; M: C; Q
25. WORD   internalPort; //   Port   mapping   internal   port
    # c& a: }, n$ H0 u* @
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    & B; x! ^- i. T% h% D5 i# C5 l* B
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    ; |1 g% H3 }) s8 u2 m) h: @
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    1 U* n8 d; ~5 N6 w7 ^
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    8 \* w( z6 m" D2 J1 O; y- {& e
36. void   clearNATPortMapping();
    + R9 p, J8 G; Q6 z# T, a
37. 
    $ H/ _" h# j0 F* W$ C
38. CString GetLastError();
    4 R6 f' \% w* k" u
39. CString GetLocalIPStr();
      Q/ ]0 u' w! O3 i( j! x
40. WORD GetLocalIP();
    0 K3 k4 Q1 J8 C2 M, j2 l) ^' L
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    - [" o  C8 P/ r4 P' v
46. 
    : S3 O& \0 G/ D. d* x0 T: O+ Z' D
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    % L, t5 h) @/ w  A5 [* z0 }. T& |; @0 O
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    / E  G( b4 Q  X
52. 
    + q5 r' T: J# D- Z) t$ Z
53. bool Search(int   version=1);
      @! a; W) E: A3 _3 E7 p
54. bool GetDescription();
    ! |; z; }) w/ L  B0 D( l9 M5 y$ m
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);
    
60. CString m_devicename;
    
61. CString m_name;
    # W* U: c' }3 F3 x  R
62. CString m_description;
    
63. CString m_baseurl;
    8 ^8 m7 G/ i" j% f5 l) I0 `7 G  k
64. CString m_controlurl;
    
65. CString m_friendlyname;
    4 D+ D; L% \# e6 V% D  l6 P
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    6 H% x3 k" v1 y% C1 _% u, Z
71. 
    
72. CString m_slocalIP;
    & }9 \& `% c, I: |1 C
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    2 U5 {: q; |2 o7 w+ M
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   $ p* O- m/ c2 r* A- ?$ a8 g
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   6 y4 t8 j/ ]1 j* t) M
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   8 l2 ]& h, u' H
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    8 R. j1 F/ t9 M" ~0 J3 L
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    6 \* \, p, U& U2 ^% R
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    . V0 s6 }/ ~# `' d
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    . g, w+ T  N. a$ N0 e, Y& b- H
22. return   s;
    5 r* D, ~9 ?! b# x  g
23. }
    : C4 v$ s6 Y. L' u+ f/ M7 s; u
24. 
    $ z8 l0 y6 h' }9 \) u
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    & \3 k4 s! z6 H
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    - D% D1 D$ y7 D, s
28. }
    
29. 
    8 M5 N8 S' Y% f( H4 o$ v% r! O+ ~5 ^: s
30. const   CString   GetArgString(const   CString&   name,   int   value)
    8 Y* i' e0 B) N. {3 N$ w5 U
31. {
    ) l+ d& }0 q0 b/ B: O. x8 I% ]; }
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    & t1 W) J/ L3 i; v
36. {
    3 c! J" Q2 X, u* p2 J' u
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    0 B1 R4 ~) X( o2 ]" R4 w7 T
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    2 Q5 P! r+ H: P/ X8 ]
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ' O/ {6 C' S' q$ G8 h; {
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    - k: {  W* O8 O: V4 J4 \
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    - S4 G9 U9 _8 w& j
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . c$ t- U% K2 w4 Q7 I+ {
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    7 Z* M. J. Q) S. I. t" U- b
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ) I1 u7 e5 @2 t$ L
59. if   (!rlen)   return   false;
    
60. 
    7 @3 N+ _1 ^+ Q6 n6 E6 @$ c' P$ N
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ) W& L, T% G. n
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    - k1 p" U1 N$ K! S- }- Y1 V
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    , h' n( J8 {' o7 F3 ~. V! t
71. int   length   =   sa.GetLength();
    3 r" t9 T- N8 O
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ! T# I# H3 N+ S$ ?& b( q9 R
76. sockaddr.sin_family   =   AF_INET;
      G0 T# B6 L1 |. H0 F
77. sockaddr.sin_port   =   htons(port);
    # L: e8 I6 Y2 t
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    ' U( R1 y7 O0 K* j
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    - L  s) Z1 n! l! l7 m7 t
84. {
    : z9 C3 o5 [% X& S* {
85. int   pos   =   0;
    
86. 
    ! ]3 I) x8 M$ X# |; ]
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    * C- H% Q' a- m0 j- C& O
90. result.Delete(0,   pos);
    ! q; A2 @1 Q8 }1 R, p" k/ e* d# o- x  H
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    , \5 ?' Q2 q& [! \) n# l) k! @
94. status   =   status.Tokenize(_T( "   "),   pos);
    ) d8 x! i& x, T3 c8 _0 K/ D
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ) O& m& n  J5 r' C! {5 b4 I- e
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ) J1 ?$ T+ t: M. }$ q) Y- F
100. {
     4 l! ~! u* v, g- k- m) e
101. CString   startTag   =   ' < '   +   name   +   '> ';
     . r4 I4 }1 T" T( D3 P; W
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     0 ]3 t6 I7 E7 L% f  L$ l! F# @, n
103. CString   property;
     : G  d& T- D. X3 J: {
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     , ?/ B* n0 v2 v# k0 M1 _/ b& V
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     2 J1 p6 q- f/ V% O" w3 N/ b
109. if   (posStart> =posEnd)   return   CString();
     ' J9 V8 [0 l) ]# W$ z
110. 
     0 ?( S* g* }- z) [) D$ ]. _
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     6 b' O: l0 m8 F5 s* U: I, M3 }( P
115. :   m_version(1)
     7 V5 Y: c3 e! g7 m* j; l5 Q2 ?
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     % e* J, @% L( A+ m
119. }
     , M5 Z2 x; d0 V5 q* ^" D$ a/ y8 l' Z
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     7 d: l. k. a8 x- i+ E7 A
125. while(pos){
     3 Z7 \" R* o, m# K( t
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     ' {2 n: e! }  `( h
129. 
     * c4 r. r# m" S
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     + |7 L. H$ t) _, W9 u) r& _
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     2 N2 l7 I. {8 _6 k8 U5 J" j
135. {
     3 S; d) f6 ^% O
136. if(version <=0)version   =   1;
     1 K3 O2 K) G- _* S/ n- @
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     - H- g( u2 R6 j8 r6 N# x9 ]
142. {UPNPPORTMAP0,   _T( "service ")},
     # c; I# G1 c: u! Z
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ) ^; W4 G3 l9 H6 \+ C% t
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     , \1 s) M( s9 n& h
149. 
     
150. int   rlen   =   0;
     , s2 z, G; J! G4 ^$ x( ]
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     $ G9 O! B. Z9 o# F, X* _9 C
152. if   (!(i%100))   {
     # X: p- r  k) G# s; q
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     2 _; J! S" x+ u% R7 x: A8 A$ Y9 g
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     9 v/ L' z* G, K4 k# X
155. CString   request;
     ( }" p4 r- A) z3 Q) ?
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 "),
     ( W% I# E' P4 j/ x
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     4 x9 q$ `* w& T; o
161. 
     
162. Sleep(10);
     : e3 T6 w. O+ L7 F' \, w- u
163. 
     : F4 b" G& i2 z! U6 r, v( W
164. char   buffer[10240];
     ' }7 w! f; O8 j& o/ B. }, s0 S* J
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ( z- Q5 ]* {0 C* m
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     + n- Z( }: i0 e/ W$ G
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     8 T( D& ~/ M$ @* j. A
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     0 H, v8 I" p9 w* F
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ) d* l2 n% g! e
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     + }' T4 g3 k$ c/ r  b* ?7 Z) ?
183. m_description   =   line;
     
184. m_description.Trim();
     / g; Z/ j) f3 |. V
185. return   GetDescription();
     . ^% f) x$ _5 z& L: Q  v
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     ) W/ i% X$ s8 L2 f0 X
193. return   false;
     2 s1 l0 k6 w8 Y; S/ q" A0 F% s& ^
194. }
     

复制代码

天云上人

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


5 h2 `: {- r7 p2 W///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


+ P' k1 A! [2 d# }#pragma once
2 Q# f+ G& K2 ]- G#include <exception>
0 u2 l5 M' f& P$ g5 @9 D

, ~- A. P! I! Z  enum TRISTATE{
2 K8 Z( r4 d+ g! d, \3 Z! _7 J# ~        TRIS_FALSE,
        TRIS_UNKNOWN,
! _3 x  |# Q# u  _( Q1 h        TRIS_TRUE
5 M' [! A$ w" M' M, O5 ?};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
* D8 m1 a' ~, r& G9 X' \9 Z8 T8 P        UPNP_IMPL_MINIUPNPLIB,
! J2 m; q$ f* w  L* e1 j        UPNP_IMPL_NONE /*last*/
};



. l; G# C; @. _: c. e
7 l+ M2 U7 v, l3 I1 B! D( R/ Jclass CUPnPImpl
1 |; g8 r8 O) {$ I- U% n  M! m{
  S: h! w  q: h$ gpublic:
        CUPnPImpl();
* c" I6 k8 _: t' H' o9 h        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
& {9 w/ f7 C5 |0 h" n        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
7 T$ p: d% p* R* ^% }: O

2 F( L: S3 _( O$ \8 |; z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
6 f9 i. D+ m2 y# J        virtual bool        CheckAndRefresh() = 0;
$ I) Q% k: L7 ^8 u1 \        virtual void        StopAsyncFind() = 0;
( M* K' c7 Z; I* ^% m' p        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
  ~! I' b9 s1 I, E) B# D       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
& P$ A  w% T; h4 v" z" U. x& }

+ h# ]; Y- P0 g* D/ i5 A        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
: z% O9 i% V0 l( A" |- x+ {2 @  M        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
8 `) J& K2 G* H
# {* P$ a6 L6 k- v/ N
, K4 A$ c8 ^5 U* f3 ~+ l// Implementation
protected:
6 f! m& I% H& [' a9 l) V9 [        volatile TRISTATE        m_bUPnPPortsForwarded;
& q) s0 g! m" Y0 [6 r8 H) @: R        void                                SendResultMessage();
        uint16                                m_nUDPPort;
0 a  g* u, H% O- L        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
# s& t$ a3 d. o# ~6 K+ v
! a9 M7 T- F' [# y
private:
4 G+ i; W" T" n5 ~, @        HWND        m_hResultMessageWindow;
9 i$ O- K* Q* f/ ]6 }        UINT        m_nResultMessageID;

* y( ?, F  H, b; X( ^& a
5 ~- f& V# u8 c9 d/ [5 I  X% [};


1 x2 Z3 i% V/ b& C$ n7 ?- _8 @; n0 S. d// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
6 S6 r0 A' k) n" Z5 u9 npublic:
$ G, A$ W2 A0 @6 C2 }8 H7 G/ ~9 d* D3 O        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
  H9 s5 k1 K/ i& s        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 y/ w8 [' d2 q" B- O; U        virtual void        StopAsyncFind()                                                                                        { }
# x! L, M' Q3 K6 o        virtual void        DeletePorts()                                                                                        { }
. Y0 |: s8 R. x/ k4 ]0 g        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
/ V/ S/ W) ^1 z, G};


8 ?+ i. m. f! n1 W( l1 P/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

  Y& X7 h0 ?4 ~8 H% b6 ]" L  r
% O. X* H" G2 J; X#pragma once
: d0 e- K! c1 N#pragma warning( disable: 4355 )
3 ~  J1 Z4 I, [! u4 r! G

  A" t1 C9 g4 B6 W' ]  g& s#include "UPnPImpl.h"
" g5 K, G8 |: l. d#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
! P- T* A2 ~% o- i#include <winsvc.h>
2 F8 s4 z( f" G& y% K5 P
4 l- [8 i8 {. p" h0 b
#include <vector>
# \" _; |/ `; @; b#include <exception>
#include <functional>



! ]9 u9 @3 M$ e" h. `
5 h; i" E7 F8 p# ktypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
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;
9 j( ?6 ?1 M' A' ^7 }6 Qtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
" y/ O3 s+ E- d0 R3 W+ n
0 R- y* Z- l2 b: D1 J7 Z* W
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

! v; [1 G' ^5 c7 d4 |5 G8 Z) ~
; ]" G5 i; F  G3 J9 E5 f5 {6 E# ^* D. \CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


7 i  M' M: X4 u& D$ F7 M( S+ Cclass CUPnPImplWinServ: public CUPnPImpl
) j/ e' g' i6 Y" P6 K3 ?{
6 _  z5 g( Z; J- M7 K0 k        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
. _2 s) X0 m( ~. h" s# o" Y5 U// Construction
' }( B- ^' E9 ?8 l! apublic:
% R& C. @8 G2 b  R: R9 E" y        virtual ~CUPnPImplWinServ();
3 u7 b3 m, Q7 @, Q; b) l        CUPnPImplWinServ();

3 y2 d1 E0 W/ Y. w# U
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& ^  P# o% G1 B  b        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
- @. w5 n- r* j        virtual bool        IsReady();
        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)
5 J6 e3 e6 }  i5 U* A        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

/ i9 J, M6 K1 a7 Y& T
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
  {& ]2 e  @5 w$ B7 F4 I        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
3 K# b5 r9 [  M5 _9 @6 W8 \& \        void        RemoveDevice(CComBSTR bsUDN);
+ A0 C7 m& J1 N+ a5 X        bool        OnSearchComplete();
        void        Init();

( a9 E+ p; A% R" w: G$ @+ X$ S9 e- j
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
. ?( ]$ x  V6 G9 V1 f2 C) u                {
1 N- v7 D0 G, S% m                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
0 _9 N. x. h' F' y3 h8 z                        m_bAsyncFindRunning = false;
                }
* `) p. ^! S$ |- \9 @8 e' i                MSG msg;
8 ]  T) D0 z1 i* c* G0 v                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
: l! ^# p" ~0 w" @8 h3 E                        TranslateMessage( &msg );
7 C$ ]! f8 n7 k4 @                        DispatchMessage( &msg );
                }
1 o+ K5 t8 W8 c1 p1 g. J1 O( ~                return m_bAsyncFindRunning;
        }
  ?! _3 V0 A1 g. e! y4 B

4 }1 ^3 ?: P# b% G2 H6 m        TRISTATE                        m_bUPnPDeviceConnected;

3 X2 y( u, O: `2 _) Y- Y6 H1 k; `
// Implementation
        // API functions
( ^* \, p9 F8 k9 g; b' m+ e5 b) V0 d        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
9 |8 i( s! \% t9 \+ G- A        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
* S. K! f+ R; ^5 i- B: M% G        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
, @# V# y; @+ M: A! B) z, {        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
+ ?  x( I3 \/ y- n/ [, g0 V        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


) G8 E' A4 ~8 E        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
* z7 H7 T# ^# G2 [
, Z9 f( f$ x5 }- y$ |& _
        static FinderPointer CreateFinderInstance();
2 p. K& t* x6 X% Y! a# `        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
1 j1 T" I" R/ R9 b& G2 k/ X& E5 Q. k                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
" `+ P) k: \6 K1 f/ u: W$ Q                result_type operator()(argument_type device) const
8 ?& ^5 p' k% U8 @% e                {
3 g8 o3 J2 Q( f/ k7 i                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- f  s8 f  K# F' X& R) T
$ |$ R4 q3 s$ a. x. r+ m/ z
                        if ( FAILED( hr ) )
4 B8 l% l3 Q& p- x- J7 L8 U                                return UPnPMessage( hr ), false;

+ u2 E$ [  g1 q$ n. l& k" U
$ W3 h5 @* i4 o, ?4 a) |2 z3 y& h                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
3 [. H! @6 a( w6 p       
2 R! B# O/ U/ D9 S        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
; C( @8 S, d0 \! d' e! v        HRESULT        MapPort(const ServicePointer& service);
8 a/ J  h' M$ A# y# G- F        void        DeleteExistingPortMappings(ServicePointer pService);
. e; a. U; s7 z9 {/ k        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
9 X, B4 H5 M. M3 y        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
9 T# d. h, F' y- A) ?& Y( d/ f                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
& F8 Q0 f% _. n4 x" Z) U
: X7 l' o7 ?0 I
7 P0 Q; J# V* ^! p/ U3 A        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
- l* ?% j& ]& n        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
8 G* [" y6 [# N. U1 j$ z- j        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
. r6 y: `& q8 R        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
4 M, b2 @! j7 w8 I7 t        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
/ o1 z' o6 |! h' C4 h! w* Z4 U1 x
3 \" _) J# X  t0 I% y* D5 X& \+ C
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
* I* F, @3 z( M* S4 M        std::vector< DevicePointer >  m_pDevices;
; }! m$ O+ H( w! y        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
, ]; U, d: b2 @9 \% n        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
8 }: ]2 U1 F# h' ?% I+ R$ @
0 R/ S8 b- b) F% H" A! C0 y
        LONG        m_nAsyncFindHandle;
) V+ f. I/ V8 Y9 G9 u        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
" A6 V* y; C/ }2 q% W+ d0 b* d        bool        m_bADSL;                // Is the device ADSL?
# i/ H& w; G5 i3 }0 V! m5 p        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
; F" ^& ?& D% p5 r9 ~        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

, F: {3 E, O5 D( {
};
2 F! t+ ?& Q" e+ E4 c# V( ]

// DeviceFinder Callback
class CDeviceFinderCallback
3 A4 Q9 N, j" w/ w        : public IUPnPDeviceFinderCallback
! x. j0 `9 Z$ ]! R% |$ j1 f{
public:
/ \7 A, R) k5 x0 X        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
+ Z' K2 I0 V2 Y- A! O7 \9 R        { m_lRefCount = 0; }


8 n$ j8 o4 ^8 w; O% t& o, {   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  i& J9 l% h! R8 x; {5 M   STDMETHODIMP_(ULONG) AddRef();
$ R9 L/ n$ z; D   STDMETHODIMP_(ULONG) Release();

' E/ r$ u& D; W. v( J7 {
+ d7 x2 z, ]4 o& w# r// implementation
) v* F0 M  g8 v3 V' \( Iprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
/ W% I# K+ x' N0 q! F        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

- g, g; a' t) c9 q, i& F& f
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
; l* d2 s% e2 c( k2 O) O( M};
* V; ^9 y5 R# f' l( V4 d3 [
) T9 o) r; k% J9 y- w
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
1 G: F: ]% v* |$ t) \2 g{
6 m" T' U& q% |% L7 ^( `public:
1 E  k( W1 C. ]( a6 r( V        CServiceCallback(CUPnPImplWinServ& instance)
, }4 J* R! x' W& \) E  ]                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- p  @2 k; Z: U, l   STDMETHODIMP_(ULONG) AddRef();
9 K' ^& N# `( k! Y' a   STDMETHODIMP_(ULONG) Release();


// implementation
6 ~" x% B% [+ a- ]" Rprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
" A/ l: x9 a9 n7 x, L' Z+ d        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
9 {3 e* n" C  T4 I
$ C) w  e2 o: u, Z% Y
private:
# Y" y+ C% \( S! o/ T        CUPnPImplWinServ& m_instance;
. j2 V; I: @, U$ Y9 j        LONG m_lRefCount;
5 \( U- D, P0 Y+ H+ S0 O};
9 K/ X6 y0 v: N& x4 O' B

, k- U/ ^, }, q% P/////////////////////////////////////////////////

+ ^( \4 r; z( V' \8 J) [
, D, ]# K0 f% A0 l3 H使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.