UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   * x7 ^% [# A# Y
4. #pragma   once
   9 n( t$ F# f% U( h% Z3 [
5. 
   3 V. o! W7 |+ h6 w" M* l/ H+ e
6. typedef   unsigned   long   ulong;
   $ U. X5 [# C+ }# U
7. 
   % e8 X3 w1 y6 O/ T' `
8. class   MyUPnP
   
9. {
   
10. public:
    5 ?/ U' M/ F) X+ \# t5 _
11. typedef   enum{
    ) A6 C  G+ k. O. u( R: l# ?# ]
12. UNAT_OK, //   Successfull
    " f- B; k* F( x! T) h+ Q
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    , O: C$ v3 N# B/ O5 G& @% f4 s) e
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
    ; L5 i/ \/ |, X! W
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    2 X( C* I& h# P
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    6 u' Z6 B1 n0 u- C% R1 ], L4 d" g
20. UNAT_TCP, //   TCP   Protocol
    ) l  e7 W: A  \- t3 |
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    $ S; O8 C) \( d+ q
23. 
    2 p# w8 h: ^* o+ |# g; _* [
24. typedef   struct{
    0 g! {' J$ j; L/ f
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    $ n, A% _  \$ L0 q8 C, Y
30. 
    6 t/ F: ~+ _% g. L9 P
31. MyUPnP();
    * u4 s  S9 g6 l' ^$ h- s) Q
32. ~MyUPnP();
    
33. 
    : g9 F5 L$ h. n/ ^, |
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    % }* k; p5 c; j) Z( ~
38. CString GetLastError();
    6 X+ U% |7 J" u) x2 @9 r
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    9 @& @: q% m. M1 |
45. void SetLastError(CString   error);
    2 O( n& X( z5 h: r8 k
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    0 F+ t; _  j6 G; A6 |, L
48.       const   CString&   descri,   const   CString&   type);
    * S6 d5 |, _5 w$ K
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ) X% N: R5 a8 o- \
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ! Q6 ^4 g1 R% \) H
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    , a0 _, X2 q6 W2 y
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
      Q, }' R3 ]+ r
60. CString m_devicename;
    ; t& H  s! P7 s& _- Q* b1 y
61. CString m_name;
    
62. CString m_description;
    6 M. \% o+ R# k2 M+ A% M" F
63. CString m_baseurl;
    7 Q2 a+ Q' U4 d# J- r+ w: G; {* W
64. CString m_controlurl;
    % ^: z+ X+ ]# @( h2 |6 @
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    ; V% `/ {) o: Q, s: o# _. C/ U
68. 
    4 F8 I$ \! E7 F( d' L  q1 g) h
69. private:
    0 G' S+ \: ?! w/ a; n# F: X: a. ]
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    3 R+ X5 c$ y2 T
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    1 o. }* p& Q6 j; ?. t/ k9 f
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   " d: o/ q! d- ^. i2 q
3. 
   5 ]+ [6 b! [% Y/ M8 d9 l* m
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   9 {; E& H; b- K3 I% [( S2 t! C0 w
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   3 N1 r  ^3 S9 ~
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   + o9 d8 N, w* I$ N% x1 P7 t* y' i
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ) g6 B* B2 k3 z, y
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    * j" A' s8 A" Y) j/ Q8 C
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    % q% Q- J" q& Q
17. {
    
18. CString   s;
    
19. 
    8 y7 q; j6 f6 W9 k  P
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    + _5 a$ l; `7 d8 u  a
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    8 ]  E- Z. n" E: g4 n& d4 l% F* t
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    * \8 r6 o4 ^8 Q
28. }
    , v. z5 H, q1 P) P% `
29. 
    - x7 ~9 Y2 W( s  `
30. const   CString   GetArgString(const   CString&   name,   int   value)
    9 y* V5 v0 Y* ^: E8 Q( w" ]8 F& x
31. {
    $ i4 z1 `$ U$ b4 G+ u
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    9 i2 d* t2 I7 y7 j  s0 v1 _
33. }
    % Q% N, M. U+ e7 q% l
34. 
      l/ r1 [2 f# b5 d& k: Q
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    * P' Y8 i% Y% m7 q
37. char   buffer[10240];
    
38. 
    9 m  ?0 i/ k" y
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    " j9 L+ n  A5 K: C& \& f! c$ p
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ' x0 B/ i* G& D% E7 I& n
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 }( R! }. ~% h$ [4 k
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    5 s# Q- l! _& @( N' a8 Y
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; [# K8 B; r  Q& W9 ^0 n7 r9 t
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    % ~) L; \% n3 {" a
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    % N4 w0 f: \+ @9 J! {  _
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    " U# I( p4 j; a- X, v/ o) w
69. 
    ) @; ]' P  Z. W
70. const   CStringA   sa(request);
    - T1 y' h- f8 ?: }
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    + k' {; O" ]8 B
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    9 c" L7 S2 `) j# d
76. sockaddr.sin_family   =   AF_INET;
    , u' y2 B7 M' I
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 n" A4 U) H  v8 `0 J2 T
79. 
    / e+ _* N# O% r& @
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 A9 R% E0 M8 z  Z' _
81. }
    2 ?8 r' d+ o6 @( Y: X$ R
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    # i% Q2 {: l& E4 i. e  ]" s6 }
84. {
    - [$ l1 o# Y7 s4 L
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    # d( u; h. g  B; Y: O
88. 
    # T' p2 [; [3 }2 n
89. result   =   response;
    . T+ `9 `4 p1 T1 a, u' D  F
90. result.Delete(0,   pos);
    , q- V1 t3 U5 X  i" g* v
91. 
    / `6 S1 o- @7 f0 k4 |
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    % P5 J2 I2 m3 x3 F9 c6 j4 t
94. status   =   status.Tokenize(_T( "   "),   pos);
    ; r6 k% L" y/ u+ E4 l$ H) _
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    8 o( q7 }/ _# J  l2 a
96. return   true;
    
97. }
    ) C8 ^) }/ W# D
98. 
    . H; @9 J. ?6 b6 {/ W' x0 x
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    4 v% N) \% V9 l! l
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( T; i7 I4 P. G$ M9 M
103. CString   property;
     . U6 c3 r( V" p' ~  X  b' h5 d
104. 
     / |% q# K  [1 F. S$ a5 m4 |
105. int   posStart   =   all.Find(startTag);
     1 z) ^$ B; J) j8 R! y/ s
106. if   (posStart <0)   return   CString();
       I) d6 _% d3 o' X& g) S
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     4 h2 F6 w/ |$ J9 z$ m, ^
109. if   (posStart> =posEnd)   return   CString();
     " b+ u3 }2 {. h/ z2 D( {1 z) r( \
110. 
     ( a/ ?& y9 S: E- W; r
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     5 R3 m; l! S2 \5 U6 F, L9 Q
112. }
     
113. 
     $ }% T0 X# R1 W( i+ t0 Q6 G: }$ l
114. MyUPnP::MyUPnP()
     - |: C, U9 a. y+ d. ]6 ?9 K$ X
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     / T' r* c) S2 E1 g
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     * e2 c* y" l4 c6 e, U+ q
122. {
     8 w$ I: R/ E+ @& b5 J; b
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ' e" Q# q& ]( z9 f
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     1 ^0 ?" G+ s: m% n) t) i
127. RemoveNATPortMapping(search,   false);
     
128. }
     # m0 S7 x2 D* n& u# L) b/ D
129. 
     . u% r( e1 A4 c9 X
130. m_Mappings.RemoveAll();
     
131. }
     " `+ H8 A! h( t/ B
132. 
     3 |9 N& s! d" V6 x/ @$ Q1 I
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     # M+ h. D& B$ G0 q# `. }: c. n
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     ) z& Q' q* F9 f; ?' ]! T0 L3 R* r
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     : [4 {1 R# w0 F  `
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     6 J  v  U& X, [( ^
147. u_long   lv   =   1;
     1 [; R$ `& Q2 o, ~% r
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     6 m1 n2 a3 F0 n! {
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     7 ]9 R! f& }) S
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ' n: @$ q( w  Q
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ' l8 u9 i, E1 y. b7 o
155. CString   request;
     6 Q/ s' |, O$ ^# H0 l7 I; X
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 "),
     8 ?' u4 Y# `" H9 H6 Y$ f. h
157. 6,   m_name);
     ) p: `5 J0 z) Z1 Y0 X2 V0 I6 j
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     $ M2 y! a0 _1 p" c
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     $ Q0 z) u/ C2 A7 [8 }: u7 S
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     0 Z+ W+ p/ M4 d$ T6 W
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++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ; l/ ?7 H9 z/ X. ^" O: @' [
176. for   (int   pos   =   0;;)   {
     1 F" V9 F: w9 M$ N! p6 M3 v5 w
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     % x7 u9 k& x6 c2 {' B
178. if   (line.IsEmpty())   return   false;
     : K; x1 U) P9 w
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ( r; u. x" r4 f- Z" ^) [
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     4 d" o# M2 I/ q
185. return   GetDescription();
     ( a, f% Z" l) w6 N1 o/ h9 b& H% X
186. }
     9 g# c& [9 r  P% J6 t9 \
187. }
     $ O$ j7 u4 }6 w6 L4 R
188. }
     
189. }
     
190. }
     & Q2 p! V2 I5 m; o* }( |/ v
191. closesocket(s);
     
192. 
     $ P6 I" t+ r5 v% j$ U2 s
193. return   false;
     2 D, b6 V. O3 Q
194. }
     5 u* Y; A* \% O, n# {! S0 O" C

复制代码

天云上人

以下有关upnp的接口来自emule，
( l4 w6 b) L, @8 C& P

+ r9 h2 p6 T3 d2 y4 l///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
# `  Y. [) w/ R  E4 B  y" Q

#pragma once
#include <exception>
5 e# q5 m3 }' T5 @$ ?
* I& j! e7 v' q8 B; ^
  enum TRISTATE{
" |, J3 R* O# |' l, l: V        TRIS_FALSE,
: w9 J% }7 U/ ~/ w$ ^        TRIS_UNKNOWN,
        TRIS_TRUE
};
, R1 a5 S5 g9 A( b# Y

& O' D0 W% D* C; Z0 oenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
& {' w' D; `+ ?6 s4 ^4 ?0 c) O0 |};

  J6 H* E5 A3 |: G2 x  P
; K+ W! V  t+ Q0 U+ U5 B# j) K& r

$ H6 @& C& ], a# X+ w; rclass CUPnPImpl
. T% _2 N+ e) [5 ]+ G3 A, J{
2 `! b" t/ }4 k( x/ ^) q) spublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
& ], {6 e: D; a1 ^        enum {
) O: u8 S" D9 B                UPNP_OK,
                UPNP_FAILED,
! r5 d3 n0 B+ h1 ~6 X% z$ U# z% l                UPNP_TIMEOUT
        };
5 F. l/ ^; ^- B8 H) I

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
: P1 }7 e7 ?" D& |0 }* I        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
0 t) r3 M5 p# J. z; e) {' [        virtual bool        IsReady() = 0;
8 z+ t1 N' n. V) I- l: W        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
2 C% M' h/ o4 b) K+ A
- n: W2 U1 ?  h: A# d9 C1 D
) h* n2 W" t" l! `( h/ c6 u        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
: u4 ^, W, a- w6 B- T        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
9 S3 h% k5 C" \$ u6 y1 _' b7 x3 M        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
1 l; l* Q: {9 R. N1 @6 \2 K  O
3 W3 \1 h3 W; G  C  r8 [
// Implementation
; k# Q9 n9 B/ t7 Yprotected:
6 B2 j+ R0 X% c) ?, R( r9 n        volatile TRISTATE        m_bUPnPPortsForwarded;
$ B  h: \: ]2 f! T5 I        void                                SendResultMessage();
7 `' k0 ^5 ~  b: }# `6 F; F        uint16                                m_nUDPPort;
/ y! ~# i7 D% G  \# C5 S: l4 o        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

: p: E6 ]& x( E
private:
        HWND        m_hResultMessageWindow;
/ u' v5 t; K' a( }/ u        UINT        m_nResultMessageID;
* k) A% @0 |: M8 _' d

};

8 M! `: l5 `( Y/ P; v; V& z9 A: d
# W" D5 G' E( Z7 X# r// Dummy Implementation to be used when no other implementation is available
  `% w* S- ]1 I' q' }% uclass CUPnPImplNone: public CUPnPImpl
; F% R; A/ ^: {4 o5 ?* J/ O( U{
7 A# ^) W+ U0 d, f' L; Ipublic:
2 p( Q! Z8 o6 ^        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


5 ^7 ]. l7 r+ b4 Q8 k' u8 S/////////////////////////////////////
  B2 G1 A; ~* m( e$ B4 j& V0 D//下面是使用windows操作系统自带的UPNP功能的子类

  X" Y+ x8 i/ F  b! y
#pragma once
#pragma warning( disable: 4355 )

# }8 v. q# ?2 S; u# E  J
! m. m" U1 i% }" u: k#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
* z# J7 W! F$ @) C, C$ \#include <comdef.h>
! }* i/ d  A$ @# x+ P6 w#include <winsvc.h>
4 X5 D' r" c( e) ]! y

#include <vector>
#include <exception>
#include <functional>



% W" Y5 X8 |' Q2 K  c6 X
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
6 u6 T! H0 S9 u3 d7 Xtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
9 v; G' F" d& |) X7 ntypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
  X# j0 r) x  A$ A6 m8 t+ atypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
8 T4 N1 U5 K' r  J  IPAddr dwDestAddr,
. f, o0 }, R; I: U8 A  PDWORD pdwBestIfIndex
3 b4 f2 h/ r2 r8 s);
: \7 X4 J) o& j- Z

5 ]7 R& N5 G. c4 s3 j) s2 ltypedef DWORD (WINAPI* TGetIpAddrTable) (
/ p1 f7 S# ?; k; F. i$ A/ A  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
# [: R! T! }- y  s( d* n8 F6 l8 b

typedef DWORD (WINAPI* TGetIfEntry) (
2 H! w/ w; ^% e1 B% \; Q6 q  PMIB_IFROW pIfRow
- j7 Z0 N2 N  q);
# J) g* ^( q5 u, I9 @  g7 a
# H  O& Q! B3 E5 {" k
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
+ D& t1 M! G: E7 C) ^2 ]% i! P2 W{
- ]2 ^8 U$ ]( ~% \: G8 H        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
) D% F8 O6 M- H& b

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
! q3 P" b, F+ ~5 G6 \8 ^        virtual void        StopAsyncFind();
" k! {  ^. K( ]5 `  m+ C6 h5 T5 K        virtual void        DeletePorts();
& b* J! D1 v. B+ i, V7 n        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 Q9 \: o- Y6 m7 C$ y        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
, Y* W# n" S$ l. Q9 S8 a6 S        virtual bool        CheckAndRefresh()                                                                                { return false; };


. P( Q# y: Y& l. h; qprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
+ M/ D$ F" r4 L" r* |) M        bool        OnSearchComplete();
        void        Init();
( j0 @( }) U/ z; @& m: C, g7 W! M$ h

5 N8 p- f+ O+ f) i        inline bool IsAsyncFindRunning()
6 ]/ |' S1 i6 o3 m        {
5 Q  H1 S, _; ~9 g: r( i% O                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
' Q0 ~. J3 _- @' q: u& g/ i/ e, y                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
, P, F5 y; C$ u% ~, c: q                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
) ?7 B" D& I% c3 `! q/ [% @        }


        TRISTATE                        m_bUPnPDeviceConnected;
) r9 T) _8 ~" q! }- o, k
7 |& h; P& h( Y8 y1 p' e: M
// Implementation
  `2 n4 _* _: _4 U( G2 ^* K- c1 t        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
. L6 N" h9 Z7 Y3 Y5 F6 _        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
. N- F; t) w' M9 E9 Y4 c) w' b1 ~' Y        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
  E. A! q) ^( V# n7 E6 g2 w        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
3 e6 O+ A: o; G1 f5 v, L; P+ g

8 w2 T0 X% N8 n. Y+ W4 U. }' Z/ ^        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
* j/ M& V7 `% O3 J2 G

        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 X& a  s# B5 I; B                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
  A8 X, g, c, D( H3 W2 _2 s                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
% P# k$ {0 o( W# n5 o+ @                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
/ ]: ^8 D. ~0 O7 ]9 G; G# j8 ?
& E! [. {3 E8 c# X$ s
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

+ K: m- }- f* U: ^8 H6 F) Q3 H
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
3 W9 m$ m5 {% ]6 I6 ?" J0 }/ r        };
       
) U$ R4 t9 b! ?' K8 V8 c# V9 R        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
$ p: m. Z* L8 X) Z0 _5 |: C        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
) j. E, F0 L' s8 V9 e        void        CreatePortMappings(ServicePointer pService);
2 e% U7 r* S; [% M2 w        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
( |. E/ n' j) p3 E4 H& ]. H# `- h& O        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
% e5 I/ S' V: h" z, R* N0 \8 a

        // Utility functions
$ o8 u9 Q' L4 y$ b5 o/ W0 Z        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
5 F. V6 P; {( R        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
" F( h* W% z: t& Q4 J        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
& t+ z2 q$ `0 S( [7 q$ z2 d        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
% ]# W8 H4 D" U8 ^- H8 T        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
9 m5 W- `/ z" N/ h        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
- i& u) F/ f6 P( _6 h        CString        GetLocalRoutableIP(ServicePointer pService);

6 k# {" V  g7 Q9 Y' m/ X& s
* u- p; [+ f( E// Private members
private:
/ o2 M( N' O8 r$ D0 N, D        DWORD        m_tLastEvent;        // When the last event was received?
$ [+ d5 ]8 H9 O* G        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
" D  b0 z! G/ ?& }1 h4 P        FinderPointer                        m_pDeviceFinder;
5 B5 Z# @# W$ C/ q5 \" l6 B! v        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


  ^; F/ p+ s, W        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
$ C( J3 J, G1 Y0 M        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
7 }1 W" S' f: E; s9 V        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
6 W7 l6 M& e9 Y1 Y! \( c        HMODULE m_hADVAPI32_DLL;
" t3 M+ p3 y+ y( @- V& X        HMODULE        m_hIPHLPAPI_DLL;
) b1 O! |" R% Z" D. `# h        bool        m_bSecondTry;
2 E; B' k+ [. w9 e1 Z6 s        bool        m_bServiceStartedByEmule;
$ q) W) Q# a1 H. \8 l4 r' ?7 }        bool        m_bDisableWANIPSetup;
' f* A- W2 l6 v- C; c) W6 e  g        bool        m_bDisableWANPPPSetup;
, A) Z8 T" Q2 w9 n+ }+ n! {6 ]

};

: e. n$ L# s9 r
// DeviceFinder Callback
/ y/ f/ z2 V! b* Uclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
, c2 ?: h) M0 a3 [4 Q  D* c- E3 q$ x4 ^public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
+ _# M* A$ l5 h/ q                : m_instance( instance )
$ h9 @; [6 J6 A% y8 n, \- A        { m_lRefCount = 0; }
8 f& D( I# s3 c( I* s. a& Y% d
2 u9 M  K1 V* O! Y0 G4 ~1 x% S
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ M5 G6 c2 `, Y   STDMETHODIMP_(ULONG) AddRef();
" S  _/ N) B' A4 Y; P4 k* J5 ?4 h9 Y   STDMETHODIMP_(ULONG) Release();

( M  V) P& r/ |. e4 ]) L- u0 B
: O3 z8 n3 j% C: a/ \9 _3 H// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
: f# ~- B6 k4 M4 R" {2 P        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
! T$ L2 ]8 v) p- A        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
( J8 G7 T- \* [% S: d2 p0 X        CUPnPImplWinServ& m_instance;
/ R/ ^' o. P- R7 d        LONG m_lRefCount;
};
9 f9 K! l4 p& a) {9 ^$ u/ b& O

8 D! H' Q  w, ~7 i, J* h// Service Callback
& `  C1 T- l* xclass CServiceCallback
1 D  ?( s3 `4 `& u        : public IUPnPServiceCallback
. ^( v3 u6 T0 C0 u+ z! o{
2 D! X! \8 r% ]public:
        CServiceCallback(CUPnPImplWinServ& instance)
6 O1 h" S# u9 E6 a                : m_instance( instance )
3 E6 }! ]) A* N        { m_lRefCount = 0; }
* O. ?7 n" `: ?   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
  C3 v! M& d3 @0 @; ?' v6 i6 e) r   STDMETHODIMP_(ULONG) Release();


  n; z' @1 R5 p- M# v1 F) l// implementation
private:
* r0 |) R* k! E        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

5 d, T/ g7 y! S& E7 s3 y2 M' o" y
private:
        CUPnPImplWinServ& m_instance;
' ^! n7 B- \' w4 _" @        LONG m_lRefCount;
};
& a# s/ h& M  V1 a: j5 w  Z0 E4 m' R
% C, t/ X5 W" ^1 F1 F: y( b
/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
" K# y, C+ f# p2 W# RCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
2 X& V0 P: \# G+ i' ?CUPnPImpl::StopAsyncFind停止设备查找.
7 p& z: _) Q9 A1 D6 |# cCUPnPImpl::DeletePorts删除端口映射.