UPnP

天云上人

/*uPnP.h*/

1. 
   # b+ C1 l- O( V1 p7 y3 B2 A
2. #ifndef   MYUPNP_H_
   ' f1 T9 O* k$ W
3. 
   5 \; E- O( \, x$ I  r
4. #pragma   once
   
5. 
   % ]; }  z8 b$ q% Z/ o5 r4 \3 F
6. typedef   unsigned   long   ulong;
   
7. 
   ! U- V5 z9 p# y! U& L
8. class   MyUPnP
   
9. {
   ; V" o$ j' E( r7 D% p5 F& |
10. public:
    
11. typedef   enum{
    ) v, R6 L1 d; o$ ^- q
12. UNAT_OK, //   Successfull
    - L" \# I3 Q; J- d# u% f# I4 t
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ; @; L) \0 L1 N. f1 L* G
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ' I; ^) I8 q) w3 X7 r' y
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    4 i  N5 B! b$ ]4 ~
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    3 H4 ]. ~' G# w/ N
18. 
    6 ]! p) P2 z' ^- v# r8 H
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    1 H3 R: S% R- O: P" c
23. 
    . i& ^! h9 }# {3 M5 S- m
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    $ S0 V3 ?# v1 t' w! e
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ' v" r1 e& x3 l0 X
28. CString   description; //   Port   mapping   description
    1 J0 U, m; `& R& P+ h; o; t# |' g
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    / x! q4 I) s( \2 O/ B! u
36. void   clearNATPortMapping();
    7 A+ a4 _2 ^* V/ z4 S
37. 
    3 i3 s* M! ~2 t. K/ |  k6 r7 U6 A
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    ' p# x0 t' }% K3 l/ t; G3 T
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    2 |' B  Z- D0 q0 ?
42. 
    
43. protected:
    
44. void InitLocalIP();
    7 [( ^, o$ A  h2 ~5 H% h7 V, k9 M
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
      D: _2 u' A' e# \$ G
49. bool   deletePortmap(int   eport,   const   CString&   type);
    2 c4 a" Z6 \5 |
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    " s( o& I) \5 M+ N  p! Y
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    . g* p- _+ @0 [; c
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ) s; L4 o$ p( C  F: \  b5 e
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    ( {. M6 @0 g# ?9 a; Z0 u' ^# N! o
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    / n  S2 S7 E: R/ ^9 z
65. CString m_friendlyname;
    ! v( y5 h' k( n3 i- K0 v9 g$ `
66. CString m_modelname;
    
67. int m_version;
    0 F# V4 Z) h1 e: P
68. 
    * f7 }7 |9 g; q, B" }% u9 N# @5 u
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    8 L7 H( J; D2 Z5 D
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    * s* V* ]' L0 w7 z7 ~) d
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   5 v9 I/ ^# r+ q7 z7 {) q
2. #include   "stdafx.h "
   9 d, i7 H$ }, z$ k
3. 
   ! C2 @0 O& P7 k
4. #include   "upnp.h "
   / h  \: c1 b' R2 V1 z2 C/ d
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   4 s# w2 _4 Y7 g" E
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    / Z2 E' `. a# _
11. 
    $ r; M( v1 Z# Z+ y0 J. F7 _2 A7 y* C
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    / V$ k) B. D: c7 z8 b
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    - c2 [$ J' f( g4 u
17. {
    
18. CString   s;
    
19. 
    - X& d* O  O" R* [+ U. v
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    ; Y- F4 K- y4 @9 h; U9 [# m
23. }
    ( o1 t: ~' v$ V$ L
24. 
    ) E# x, w$ y( s  s& D  r
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ; n- f2 z. s3 H
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ; H- P% P8 c) y2 e! ~( Z0 R) c  O
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    3 _2 i4 ]- l8 v; J: e. I: D5 p2 e
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ' w6 J7 p  W2 @! h5 c
33. }
    . z5 i8 j7 H2 C: C( A
34. 
    - ]. a3 U5 T4 B
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    9 O2 }/ z3 G% G
38. 
    & w* x- q7 j. L' r1 h3 U
39. const   CStringA   sa(request);
    - i: T' `+ q! j
40. int   length   =   sa.GetLength();
    1 u# f7 N  `, r) K- ?/ n: f4 p: ]
41. strcpy(buffer,   (const   char*)sa);
    ! V8 J; K. B3 \' |3 Z
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ! K5 Y$ v7 J0 A& |
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
      p: G% r' c9 Y- g
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    + G1 |, l/ @2 A; A, X
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ ~/ D  @! X4 l
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    * j+ Z0 z' x( W7 Z
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    : p& A+ x& E* s' F1 R( E
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    * C+ C3 W7 c/ m3 t
60. 
    4 \& X' w% o/ T
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    1 Y/ L1 w3 Z' b4 S
63. return   true;
    
64. }
    0 S- J- U2 a3 A
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    2 I2 }( O' Q) m0 o/ d/ b) X. |: `
69. 
    / c# N! [" o0 K9 m1 {1 h5 g: t( C+ r
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    2 y' @$ b+ D9 B% j
72. strcpy(buffer,   (const   char*)sa);
    ) Z2 E! p: F: [  n, g1 |
73. 
    2 I. t7 I$ B" R, {' p+ G
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    9 t- P' E* S( P, R: T( i0 B( _$ f
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    ) }- ]3 a1 @3 C8 M( L5 {8 k
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    ! B2 P/ ]0 G" b& ^5 U
82. 
    / {. }1 P& R) P: r# i5 B
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    . P' [& p" o4 W4 H! M; C
85. int   pos   =   0;
    # [# H0 b) t. z  B! h) g' V
86. 
    " {# w  v0 X9 h1 Y+ Y6 ^' ]8 M
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    8 G8 n& _: H  S  ?: }$ n6 i5 H% C
89. result   =   response;
    . i1 X# n. v! X$ `2 A, R: e4 e; }. o
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    % t! B. ?+ h8 m& M6 K# ^, I; S6 l
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    " z" l6 i" H- D. E% g" r8 C! U( Z
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    2 A3 ~/ u1 c6 f, l( |) S; v4 S; o
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     3 t7 w; _# b  o, Q9 I  K2 w
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     0 {+ G9 t1 t& ^0 R2 C) X9 f
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     " s2 b& c& [* `+ B8 ~
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     & O' u  S2 i7 ?
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ' O: F0 u" }. G5 M( u
112. }
     
113. 
     ; |# c0 ]# E0 j8 K
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     * T5 o6 Y0 V7 k+ W
118. isSearched   =   false;
     ' w. A6 s0 ~& ^* G/ q  o! X; _8 w
119. }
     
120. 
     ' U* ?6 \2 ?1 G8 r
121. MyUPnP::~MyUPnP()
     ! S5 j. ^( s  V. ?: M$ u6 `& B
122. {
     ( M) w) I' }3 ?  j: Y! ^0 P
123. UPNPNAT_MAPPING   search;
     + _9 y/ n( R* }& F7 ^
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     8 r6 s9 Z, l- U" K5 t% P
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     0 ?# o. \- A- a; {' I5 _
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     ; a6 ?4 T2 S0 x
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     . W4 C0 a+ W& M5 i3 n
135. {
       ^( m$ B$ j0 r3 r/ @  u, ?  x
136. if(version <=0)version   =   1;
     8 D. f" J  `6 R' @. Q7 K; ?
137. m_version   =   version;
     
138. 
     0 M6 n$ X" E6 n' |, }
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ! `* f+ O5 N4 j* ?/ V& @
144. };
     
145. 
     - }: P8 ^, O) p9 k8 R5 C* V9 p
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     4 ]- y. I- v6 a
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     / \; X' U# H4 A$ n1 R
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);
     ! M# |7 A% H# J" @! Z2 U
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 "),
     + b4 d) d, Z2 E1 {7 k8 b
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     * \# p& |* C7 M* l: Y$ `9 v
162. Sleep(10);
     3 A! h  {& C& J- Y5 J9 u+ M! P
163. 
     2 m& \! H6 o3 g" @' X
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     5 u1 k6 u) _* G: N! M
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     2 n% r4 a% G% y1 `% I. A
168. 
     0 c) U: u, }7 ^/ R  u% k
169. CString   response   =   CString(CStringA(buffer,   rlen));
     9 \/ s9 l: E0 T5 `) `( ^* P1 Q
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     9 I9 ~9 q- M) b  M* l) j8 W  b& ]
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);
     & J% J$ ?8 M) N- Y! h
175. if   (result.Find(m_name)   > =   0)   {
     
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();
     $ P, @- @$ p/ ~2 I4 M# I/ p
181. if   (name   ==   _T( "LOCATION: "))   {
     / ^7 }7 S  H6 l" g. I
182. line.Delete(0,   9);
     # l  K% K+ W8 b) {! R& {- {3 E! L, V
183. m_description   =   line;
     , ?4 w" ~8 O. |1 Z4 o
184. m_description.Trim();
     
185. return   GetDescription();
     ! [  {; X  U( E/ V8 M2 D
186. }
     
187. }
     9 e/ P9 F+ C  u' E& N0 A
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     , K" h# G9 d( m: M/ e1 O
192. 
     
193. return   false;
       q  U. W. I& I! |; ^) g9 f
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
5 }6 w- W5 }/ p/ L

+ Q" C9 H, p( C8 Z///////////////////////////////////////////
/ f. ?+ w5 l, Z+ q% K$ }8 U+ `//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

) G6 {4 a" n7 ~# ]1 v
7 D7 Z1 C1 {% r#pragma once
#include <exception>


/ j! _( _2 w/ n; t' O* @  enum TRISTATE{
        TRIS_FALSE,
; V8 o. [/ N, c. Q7 T" e        TRIS_UNKNOWN,
        TRIS_TRUE
  C# R3 w/ w# c1 O1 ]) L& I' G};
0 H& l0 ^; h( w' K# w

6 w9 d- T7 L# `# v% q+ c8 Jenum UPNP_IMPLEMENTATION{
( {* S; {! C' d- C' }* L        UPNP_IMPL_WINDOWSERVICE = 0,
# Z0 C5 F5 [: t2 j+ j. k$ ^        UPNP_IMPL_MINIUPNPLIB,
/ W. k/ F2 w( h5 _$ z        UPNP_IMPL_NONE /*last*/
! l  R% @9 g0 K( w/ ?};

$ P* Y  D0 [; e
6 }! i& X6 W5 @- B- b: |
- C: n+ t% f. n, F4 ?' q. T
  `! o4 t* `* _7 H# oclass CUPnPImpl
: D, P1 z& {" ]{
public:
        CUPnPImpl();
# C6 Z$ B4 G7 z0 @        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
& [) V1 Q4 v: W1 o" j/ {' r1 a8 I                UPNP_FAILED,
                UPNP_TIMEOUT
        };
# ^" v5 }0 O! F! }( m: i2 u- D

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
- T. z- M" C4 v( T6 m) j( q        virtual bool        CheckAndRefresh() = 0;
- t1 [8 U3 e; U, U  J+ a0 ?        virtual void        StopAsyncFind() = 0;
1 p# _6 v0 K, `+ F% F        virtual void        DeletePorts() = 0;
$ f4 b# N4 [/ C/ G* S        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
' M5 R% L0 D4 }" R/ z% w/ ]       
, O; O/ S" Q' I) ]2 }        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
* ~1 D; b. z' Y: T! Q: f/ k
" h6 s3 ~' @4 Q( i9 u5 j( w
; b9 `2 u1 [5 x0 A: z        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
! @/ \- z( @% F" n% y6 d* R- H        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
' s" Y& ~4 t9 Q; L3 F2 N        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
( q6 Z- p. b- x4 N

// Implementation
+ g# R6 k. ^2 b5 Wprotected:
5 p3 }# f) J: T        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
: h* g4 f8 n& r( B5 v" N: j$ A        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

4 G( C% o2 A! r
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

, ?: ^. {' }8 f* _
, b) ?3 ^5 V; i9 L$ l0 q$ d};

! R2 L* `5 Z: Z: T& h7 s
$ s1 o  _% d3 U; e9 v// Dummy Implementation to be used when no other implementation is available
4 A" C! b' D% c) ~  Sclass CUPnPImplNone: public CUPnPImpl
. {' Y6 j, ~# }{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
9 N' [) N# M) e2 x+ `9 Q        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

, G8 p7 \6 O! u0 ~2 @
/////////////////////////////////////
8 i- X6 _/ F" V! x& ?//下面是使用windows操作系统自带的UPNP功能的子类

" e% e  J0 K( h; ?  Q( ?$ M
( K( U; w* ?: J0 o#pragma once
3 Z+ Q# D* k$ }4 t* p5 _' k#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
! \& f3 P, n* I' r5 Q#include <upnp.h>
/ O, i4 H4 `: S$ ^# r( ~# ^& n% g  b#include <iphlpapi.h>
) R/ |8 g$ `& w/ m3 n#include <comdef.h>
3 C. r6 A6 f2 `% |7 U#include <winsvc.h>


#include <vector>
! u0 h8 z) P8 N- P1 R0 S& T, ^4 f#include <exception>
4 G! A4 d3 Y, [5 Y2 D% N  }#include <functional>


# @: a5 o5 ~! I! p
7 t5 x0 \5 z: `7 I( U6 w$ _
6 D' a, B0 L1 N- [typedef _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;
' \' \" p  z# f6 btypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
/ {( i- ?; _# c9 B  `4 T" F# \9 {0 htypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
; K$ b# a! B/ b" w' O5 w  S& Ktypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

% X' |! \* b1 c% e) j$ s% E
1 d2 W/ E& E. B/ x, x6 F. ftypedef DWORD (WINAPI* TGetBestInterface) (
4 K: ]. _' d+ j% s6 ~  IPAddr dwDestAddr,
5 U) y  |' k. A* M4 R* V; i( G  PDWORD pdwBestIfIndex
);
% D1 }9 D: _  m, |7 o% z
( s- N* w1 J% k8 D) [6 @
typedef DWORD (WINAPI* TGetIpAddrTable) (
5 v8 c- I* O9 O/ ^4 v  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
% u. v; q9 \0 e0 y' }  BOOL bOrder
3 [. l% F/ E# h);

3 H7 w) r. r$ o6 Z  G" ]! g2 @
typedef DWORD (WINAPI* TGetIfEntry) (
: Z) P  [' D; z" L8 [( L  |. o  PMIB_IFROW pIfRow
);
. Q1 f3 i4 h" |' S$ k4 K) d% U! L
$ F/ K9 n# j! t+ ]
CString translateUPnPResult(HRESULT hr);
% R: n$ O# S2 n7 }HRESULT UPnPMessage(HRESULT hr);


7 @& _  C0 J* P8 bclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
2 \" `/ S: w$ n8 @5 Q' X        friend class CServiceCallback;
4 S' X' H9 h! R5 @) x+ [% D: [( l0 j// Construction
. B; }2 ~6 E/ q" q! K- e% {, M4 spublic:
# |$ U9 Q( r* P9 _* Y4 r. k9 x! {        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
0 y) ]& V" ^: K        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


* v  x  g! @! B3 z        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
# B- K1 B2 @* K8 U% H% S        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
5 o" I1 l% E3 n/ B

( ?* h5 N; ?4 Sprotected:
5 u, K2 L! c& `2 A        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
$ m9 U! W) u; Q  g: d        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
7 f7 a# P$ l* F# E  U        void        Init();
! {. d; e9 [9 G% Z4 c

        inline bool IsAsyncFindRunning()
  u2 V6 x' }* m5 _7 r+ r) B        {
2 e6 t/ M0 S$ c                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
. v( o+ i8 E$ V3 M                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
. f  ~) u0 f0 d                {
7 H8 A5 L" y0 }                        TranslateMessage( &msg );
# B  R% d/ Z& N9 S5 [' X                        DispatchMessage( &msg );
* `6 U8 o5 V7 T% i( U                }
                return m_bAsyncFindRunning;
4 N# Z$ i  a; a$ w( F* o1 Q        }

. {: f0 m/ ^/ X. N$ F% d
        TRISTATE                        m_bUPnPDeviceConnected;


  s/ g, C5 B  j/ ~, \5 }4 W// Implementation
8 a* i- c$ a% V# s        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
& h# n- c- M3 ^2 Z: z* o$ s* d        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
' }/ O% _/ U0 E7 C        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
+ P, _% g3 D# Z5 y/ z% M        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
* U# h( Q" H0 a- u+ x        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
8 ]+ T6 m. @: S

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
* x) V0 |* Z* s) W; h        TGetIfEntry                                m_pfGetIfEntry;
+ K4 h" N3 a5 g. O: ^

, s9 L0 D& U# [$ N4 `3 w8 J        static FinderPointer CreateFinderInstance();
/ l' E+ ~# B" q% H; {  I        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
( S6 T8 z+ c& M. n/ e$ W+ Q' U                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
, V/ w- s% V( }  Y; \  B1 v9 D4 C" R! M                {
# h, m) {( R. B+ r                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
& N5 a0 f6 P$ Y2 H! p

                        if ( FAILED( hr ) )
) R6 \5 h: z% l5 J1 S: i; s                                return UPnPMessage( hr ), false;

/ C! }+ V7 b0 l& G# o+ b
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
" V& f( x) H  @  {0 @, u       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
# u4 o) ~7 H& R1 i: r2 z! ~, Y  t        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
$ P# U% ~0 h0 G5 E                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
0 D6 g: k! U) G8 K4 x

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
6 r2 K) x# D; q" i, n" h! b, K  K" ~        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
  p6 B9 z# O  k2 p) `        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
2 r0 `& y  F/ S7 [0 L        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
! V7 w- |: Q0 f
* q, I0 N* z' V
// Private members
" w& y: J8 z+ @8 M+ T; f: Zprivate:
) n) ]' M. _4 H        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
9 L6 c; J: @# T- ~! H( ?) m
4 G/ I* q7 V  d: y# @- u3 g
        LONG        m_nAsyncFindHandle;
+ Y& H% C% X7 ?! A' o        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
# `! I2 S* n0 z* B        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
+ t7 D. B! h9 E        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
; u3 x- \% K0 W" j5 G8 t        bool        m_bDisableWANIPSetup;
8 D" c# Y3 L2 C, v6 X7 `" n        bool        m_bDisableWANPPPSetup;


" u* R+ H3 k5 A3 J};


- m7 M# t6 q8 R7 c, A6 A8 P9 t  _// DeviceFinder Callback
9 K4 Y8 D" T! t) F  m; Tclass CDeviceFinderCallback
! [* h/ e* a. ?        : public IUPnPDeviceFinderCallback
5 }0 Z0 F* ~3 g" q* R7 w) h{
public:
" P9 b. Q/ Q* `) V# ]8 Q- O1 a        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

+ S: d3 ]7 Q$ p
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
" x; ?7 o& m) o  c- {   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 l) [5 N9 e8 W0 X6 t( q, M

// implementation
private:
& A9 l1 c* [& u- V1 b        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
/ [0 @) S6 C% J

; w* e0 Q: O8 X5 ]3 A) r% Oprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
6 X  g  B0 z0 Q8 ^; x. d

/ z8 W1 h/ G  W& b  B* }) [// Service Callback
class CServiceCallback
8 v) @$ }) l  h: x6 Y6 C- e9 i! V7 \        : public IUPnPServiceCallback
1 N) M; K5 K$ ~. F! E{
* Q  H: E9 ], X7 o( a5 @public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
1 W/ l) ^9 }$ B8 o9 ~5 K   
# N; U. O% u1 u7 T: N! c2 Z   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ P  e6 C" S, q: @" x$ c   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
: }7 P; O2 _2 ~! c4 d# `

// implementation
) X: ^- M8 [2 F0 E$ p4 C0 M& S' x# Sprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
) K$ ~, N$ P4 V8 T( j: E' B        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

. b) I+ b* d5 k* B4 {- K
# V) r: f, M( g3 ^private:
        CUPnPImplWinServ& m_instance;
8 B& a- A. M$ y3 ?; e& J/ p9 r" {        LONG m_lRefCount;
% ]* N2 k' @# |2 E4 K& ~$ C; E! D};


8 J9 e7 P) Y+ N$ R& h& G/////////////////////////////////////////////////
5 m9 Z6 c; ~& K' E4 K) y& S

使用时只需要使用抽象类的接口。
* j) h* n* L3 w' h; b9 @, @' BCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
# y! w: X7 m* w6 o6 t& R- x3 U$ |CUPnPImpl::StopAsyncFind停止设备查找.
" V: w2 B' y* z6 S0 W+ m' fCUPnPImpl::DeletePorts删除端口映射.