UPnP

天云上人

/*uPnP.h*/

1. 
     E* C1 z- r  V
2. #ifndef   MYUPNP_H_
   
3. 
   5 a) h4 I( W; A( ]& U  {. }
4. #pragma   once
   / x& j3 Z! F" `$ h* U1 R
5. 
   
6. typedef   unsigned   long   ulong;
   $ k3 q2 z; o4 P6 D; u4 L0 |4 e
7. 
   
8. class   MyUPnP
   ; P- [* H- N- S) N1 e. P2 k
9. {
   # l; w$ T# [! u
10. public:
    ! Z8 i" R- Q& J6 O6 Q( \  Y" Q
11. typedef   enum{
    - V/ q" S; v: u
12. UNAT_OK, //   Successfull
    ; I. i! ]5 e# w0 |
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
    , f. J3 V6 Y8 k7 B9 h  Y2 C- u  l+ [
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    - a8 P% o1 O, r$ \
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    " E& j1 O( N  v6 J. L$ B
18. 
    , j1 x6 ?1 \0 I' J: B3 N
19. typedef   enum{
    ! d' ]( u' O3 S: r; x
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    6 b  z% v9 k' i* k, r, n4 b' ^( Q7 r
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
      O7 S$ w( t! Z/ E6 |( z
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    & h3 p8 _1 G; N$ Y  a2 k$ D- ]
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    # g0 _( }+ U( T
30. 
    3 V, C! f" A+ s5 ~' t3 i2 v
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    6 Q' V/ H1 L3 g$ p: `6 Q
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    % X3 k* @0 ]) O( J7 F
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 {( U$ V8 Q9 _; D+ @
36. void   clearNATPortMapping();
    
37. 
    8 d3 z4 P; z6 ?  ~- B: s
38. CString GetLastError();
    - o; h2 u4 ]% [. N2 x( z+ {
39. CString GetLocalIPStr();
    4 `, s( m; ~0 u6 e9 K* k
40. WORD GetLocalIP();
    # s3 A/ ~+ D4 C) R* G
41. bool IsLANIP(WORD   nIP);
    
42. 
    # M" X6 h& G3 T5 u% I
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    ( S3 X+ L7 E: I, e- U! g2 e
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    . X$ x+ @6 c8 f8 J" u4 r+ b- h
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    - i; W2 t* }& Y3 Q" r$ w
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ) ?  L: |! J: f: P9 Z" [6 z
53. bool Search(int   version=1);
    2 \6 L0 w' z- R  R# U3 P
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    # [: }! _$ y! E, S( p1 W
57. 
    * Z* }- q" z$ U. b% w) |0 s8 x/ Q
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    : Y+ i6 Z' h- w) Z4 S
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    2 n7 `0 c1 G) K% R+ J/ u
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    2 O- h* v/ h% b! e9 D1 a
67. int m_version;
    
68. 
    
69. private:
    * ?4 W2 x* |' y4 i3 L1 v! ~1 @
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    4 M) }0 D& u! Y; Y8 L/ W
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   : p" F% f' E- a: a* \8 J- k( c
2. #include   "stdafx.h "
   
3. 
   4 V. M. b* N( V+ F9 g+ e& a  z* T
4. #include   "upnp.h "
   
5. 
   7 x. M3 D; i, l: N. T
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   4 k9 I# a2 i, @) e1 g2 ^
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   + Y# z4 ]: H: N' J' F2 n4 v
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    / j. i7 j$ q+ c) c0 O3 G- F4 I
11. 
    0 s( |" c' N  q- w- V7 K2 w8 f
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    " t' w9 p- k) [2 t5 p3 t
13. static   const   int UPNPPORT   =   1900;
    / a/ v! Y* M$ r4 ?& S5 |: \
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    % j& E' F' d3 h; S9 |
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    . G! m" m- O8 n4 H
19. 
    
20. s.Format(_T( "%d "),   i);
    $ t5 r" ^% M! e7 D/ D! m
21. 
    7 o: w" W2 h& r7 z% s* x& g/ o# @1 H
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    0 W4 ?0 A: @% _3 ^2 }9 a
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    4 @: B: C" O% i, q# E# k
29. 
    - i% a. @2 K) E( h7 R
30. const   CString   GetArgString(const   CString&   name,   int   value)
    7 T" s! ?# S& V
31. {
      g0 U6 u& W: W8 X7 s' Q3 O
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    9 C# D% p: r4 ~% ~: X$ B
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    : z1 \! W9 ^3 A% p
37. char   buffer[10240];
    
38. 
    5 w4 z+ q& H5 W- {
39. const   CStringA   sa(request);
      ]6 j- {/ c2 T- e# }' c$ D! J
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    , E  g4 F! D5 n2 D+ x6 ]$ w# J
44. struct   sockaddr_in   sockaddr;
    : G% W7 n/ B. k) F; b  Y  |- ]; d
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ' J$ s+ S% `9 T  h
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    : j# B/ V! K1 y4 q2 j8 q
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 D6 [: Q8 L$ a  U5 b* |' H
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ; e7 |2 e) c) y7 B( C( j
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    . H- V7 y: Q. ]
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % R# f3 g' X/ g) V/ F1 Y1 r: L
53. Sleep(20);
    . b! u- I5 y2 y& D
54. int   n   =   send(s,   buffer,   length,   0);
    $ R+ [! j4 }3 ^; T
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ; k( \7 l# J1 q3 D
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    4 R: E- f3 n  K7 e. T9 j4 w1 s
59. if   (!rlen)   return   false;
    2 M5 ^: C. U0 J
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    " I, c6 I: S1 F. E* v
62. 
    1 ?9 c0 C7 X" ~
63. return   true;
    " T; H3 i! e6 S+ o- N9 H$ }
64. }
    
65. 
    1 }& g, l0 [; O" N" d- a
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ) J9 i) ?! @1 L/ x
68. char   buffer[10240];
    $ r8 I4 D! m2 l4 V. w
69. 
    / A4 j" `" s9 q; h5 \+ d
70. const   CStringA   sa(request);
    7 Y: |- ~- d2 G' t
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    " B: d: B( D5 l. Y0 e  x- L
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    . u1 S* N$ y8 ^5 q
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 D* R& z: y8 {/ r3 R3 p
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    7 r1 m* Q  v+ m' L' W4 p
81. }
    
82. 
    ! f& `- N+ A# {! N7 g4 F
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    - Z7 s3 F9 J/ N9 c3 }6 i+ n
84. {
    ; {6 ]' B( f7 @# D) V) I
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    . \) ]& G  e8 |7 A, [* J, _
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    $ W  a1 A/ ?# C6 e
91. 
    ; u1 u0 s- R( M
92. pos   =   0;
      j* B" K9 G5 h. {# f
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    + ^4 ^" W, b  u1 k
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      @1 T9 M2 ?# ]' b. E
96. return   true;
    
97. }
    
98. 
    0 c  p' d2 c# T1 L% M" l7 U
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ! S! \5 @: [6 Y3 V
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     2 f: Y5 `; y. [/ @. ^
104. 
     0 q! x* q" w8 G* C# ?$ ~" [
105. int   posStart   =   all.Find(startTag);
     ! B+ c' i2 U: C1 v- O# M. f
106. if   (posStart <0)   return   CString();
     
107. 
     6 |& ]/ Q6 H3 P$ D: s( N% M
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ) h5 T* F9 T1 A, w9 R
112. }
     + \) k3 X# J( v8 w4 [9 \0 `
113. 
     
114. MyUPnP::MyUPnP()
     8 u8 ?2 q3 ]' y3 k" ?8 O
115. :   m_version(1)
     ! B- w1 O5 I' s
116. {
     
117. m_uLocalIP   =   0;
     ' s3 G' f4 v3 U' [  T
118. isSearched   =   false;
     
119. }
     : h7 d$ I3 j! D- y
120. 
     + I( o3 s6 }3 j8 v: n2 U, D
121. MyUPnP::~MyUPnP()
     
122. {
     9 p  ^" x# y: T- ?! m" }" \2 b" U
123. UPNPNAT_MAPPING   search;
     # W5 e' U/ I2 x. U: z" h8 R' G
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ! ^" Y/ ?" X5 I+ l) e, \! m6 v
126. search   =   m_Mappings.GetNext(pos);
     2 v0 |$ T* K% U: T# o
127. RemoveNATPortMapping(search,   false);
     2 }7 r  b* L1 c0 Q
128. }
     
129. 
     8 y" j2 E, \9 j: J: Z
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     ) j+ }2 e6 G( }+ U4 P
134. bool   MyUPnP::InternalSearch(int   version)
     $ L& U6 Q  X# L1 h8 r
135. {
     
136. if(version <=0)version   =   1;
     ) N6 \1 `4 g0 S
137. m_version   =   version;
     9 i' H- z! ]- p# R
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     $ H2 d% p$ ]* J/ i2 [( f$ K* J# a+ S
141. {UPNPPORTMAP1,   _T( "service ")},
     1 x0 m6 U8 B8 D* H+ `/ ~" [  ?3 \
142. {UPNPPORTMAP0,   _T( "service ")},
     , v0 O1 c% q$ F- I' `
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     0 R6 ]. K1 Q; ~9 ^: m
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     2 ^# Q& }) P0 ~# c. O0 u
149. 
     ; A! W* O  U! y5 ]9 f; u) k3 b" T
150. int   rlen   =   0;
     + B6 y0 \. ~1 p- r- q8 |
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     7 N) W- H. T7 B# k& M# n6 F
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     " z; ^' V! V# |5 ~% L7 G
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 "),
     
157. 6,   m_name);
     1 z1 f7 _6 V5 Q$ G5 D4 [
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % y( s! m( B/ B' k& c
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     " ^3 g9 [% u7 T$ w' [8 e0 m% L
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     . m' K) m1 b  J8 H' V. y
166. if   (rlen   <=   0)   continue;
     8 s+ p$ c7 J3 }0 s) [
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     0 s5 w( S" q! ?' w% @7 t! v
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)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ) K: `. h! X6 s: ?. r
178. if   (line.IsEmpty())   return   false;
     - i/ H: ^* G! s& n
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     % V' |) ~; D9 \; O& y
181. if   (name   ==   _T( "LOCATION: "))   {
     / m. n! Z/ N$ o) n0 m5 z
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     ; f8 |3 [+ F) d
185. return   GetDescription();
     
186. }
     5 m7 `5 `# [: g& n
187. }
     
188. }
     / M6 j  X0 v  ?7 X5 ?1 X/ l
189. }
     
190. }
     ; R* |: _2 V$ C$ J% Q
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     7 Q- \; S6 ]' g$ c( N

复制代码

天云上人

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

- f9 {: j3 H; t/ A
///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>

# G- n  C0 z/ h. p8 d" i# _
4 F! C2 ?1 i9 k/ W9 O  enum TRISTATE{
        TRIS_FALSE,
% t' I- f* F0 C3 t        TRIS_UNKNOWN,
        TRIS_TRUE
};


; K( c% I" S! ?( S  tenum UPNP_IMPLEMENTATION{
$ |; H; p7 d. m+ U4 H; k+ J  N        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
1 t3 p% a0 L* H        UPNP_IMPL_NONE /*last*/
3 r$ K# Y4 o2 Q" B0 E* n};
4 V8 L  y1 O! ?* @& s0 w

- \  E7 J$ |: u( c9 [; _* I

1 i6 i* C' h( U2 bclass CUPnPImpl
{
public:
        CUPnPImpl();
# y! [- V/ Z* b' [9 c        virtual ~CUPnPImpl();
: X3 V3 r8 q1 g( ^4 p        struct UPnPError : std::exception {};
' \" }/ o" X4 u" M        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
' O, x% \& M- f& v        };
7 i* O6 b1 g" n6 N8 [& I: D' [1 g

7 t, X: W7 K. M) @$ f' I        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
' b9 p/ U) ?+ {! \        virtual bool        CheckAndRefresh() = 0;
" F7 |, E4 m  v, a9 H1 h; E        virtual void        StopAsyncFind() = 0;
2 E1 d, j$ `7 Z5 u) s2 |) ?, W        virtual void        DeletePorts() = 0;
4 i6 o, D* s+ H5 i/ \: O* U        virtual bool        IsReady() = 0;
7 R) n2 B/ w8 |: Q: B8 W        virtual int                GetImplementationID() = 0;
* B3 K0 Z3 \, Q$ x       
$ M- {) g# w+ x        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
6 Y* [  k) f9 h$ C: M- ^  Z        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
0 q9 Z6 e# @$ {+ ?$ {" b0 n$ i        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
. o6 a- w% f8 A& X( m7 l/ C        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
2 e3 E0 [$ D* x! C# d* U
. d- |3 N! z7 K# }; f$ H
// Implementation
" t* C0 p% Z5 e" m/ S: n) xprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
, W3 U1 q/ \4 h, u, x        uint16                                m_nUDPPort;
4 x0 t+ D) V5 @# F' Y* L* p        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
) _0 X& n1 r. J3 o1 R- D        bool                                m_bCheckAndRefresh;
% z( h6 a$ Z- w
- l. ~4 g7 ~4 N. k! s1 L
( a0 O/ x" q3 O7 l5 tprivate:
/ [2 D! r" f% \7 x+ b; V  d        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
& ^; e' t4 R5 `

% R+ G' K6 B' z. d- z};
! |  _8 z7 [9 Y: O8 [5 O0 X
, F. U/ e- n- p+ z: R5 D
// Dummy Implementation to be used when no other implementation is available
% u1 X4 F. F* r7 o; P! Iclass CUPnPImplNone: public CUPnPImpl
$ K$ O7 c; X' T3 j' R' {  i0 {{
public:
3 n1 t4 w( F8 l& W! m  ^( y        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
% d' N/ l' N0 ?        virtual bool        IsReady()                                                                                                { return false; }
% i  f$ T$ S( e        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

" \! F7 C0 x( b$ @8 H
4 v' @* O& S! W. h: y; w/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
, g& ^; r  M) H; f7 x

#pragma once
#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
: f5 [: E/ T; q% s+ z#include <winsvc.h>


: X& n8 r# G2 |* ^# J8 }+ }0 Z#include <vector>
#include <exception>
#include <functional>


$ O& w- W+ K% r5 m9 }. F

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
' L2 f- m+ p5 C6 Etypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
7 B* i6 {6 P& r; xtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
2 C3 h( B+ x6 `# @5 B3 R) dtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

7 c7 G0 _+ W8 ~7 ?1 s
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
, [$ p( Y. p4 J

typedef DWORD (WINAPI* TGetIpAddrTable) (
+ l5 N: S: E  e" F0 E! G! ?3 k& A  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
, a# U# j* G5 S) e- q
* b7 n/ @( n3 {  V
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
1 e  S* g% ?' Q. Y. N);
9 n+ @# _8 ~! W" `8 c; v

1 V( N- \( t" m/ n3 aCString translateUPnPResult(HRESULT hr);
- Y& ^( N# U6 a) T' aHRESULT UPnPMessage(HRESULT hr);


# N  g6 k  W2 c% y) w7 P  o0 U3 Tclass CUPnPImplWinServ: public CUPnPImpl
{
* v( t+ m$ d! q- a; s        friend class CDeviceFinderCallback;
* u' |6 R: \7 d4 R! w6 j) V        friend class CServiceCallback;
// Construction
: I3 g6 B! Z. Hpublic:
* |" d& s& u- B" {7 ?1 \        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
! R* @' B! R, a
: |! w6 s8 V) s+ v) [3 ^* ?
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
+ ?! j$ F* o: Y) o
9 @2 u7 f& K7 K8 @$ b
6 L. p# _& @8 p1 X3 Q5 _7 e        // 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; };
1 r- n/ W' X7 g3 v8 I
9 r% D) l$ n: H8 @& m
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
: s  [2 ]) d& l2 H* e& M        bool        OnSearchComplete();
        void        Init();


: Y8 y, w: e9 i4 _, \. A        inline bool IsAsyncFindRunning()
/ y0 `6 p3 I" {- X) n1 w        {
8 Z& Z% P! ?1 s. R                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
+ ]5 S* B/ ^& v+ H; N                        m_bAsyncFindRunning = false;
                }
                MSG msg;
) C, c' \. l# X! a3 b8 Z( M- ?                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
+ B% G) ^3 O: G4 ^! E/ G                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


8 M% ?! h9 K$ Y! d+ \        TRISTATE                        m_bUPnPDeviceConnected;
! [% ~& [. c1 f4 k

. C7 k$ Z! ~9 d% j' F// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
7 Q0 {$ B+ t' t% v  d4 P        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
# {; D  P$ ]: J3 t/ N

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
' ]: g, x3 ]& q
. ~+ X: N& O4 y
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
% n5 ]. @; c' R$ w$ ]        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
2 S  y( s' P" D8 P                result_type operator()(argument_type device) const
+ ^: R  L) ^/ N  n                {
3 t  ]2 k; J5 g6 `" b                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
" E* L( Z2 i6 U8 h, R

1 r4 S. D; u- l: ~5 k" @                        if ( FAILED( hr ) )
5 y. }  M3 |8 V; u  M$ j. [                                return UPnPMessage( hr ), false;
# D) n. J$ L* B! |

/ ~$ s3 b  A- g# f  W                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
, l6 N3 ?0 h; C/ p! J" d% j                CComBSTR m_udn;
4 z  G3 O- J) T+ r3 v. F        };
       
# V: p1 ]/ @& W        void        ProcessAsyncFind(CComBSTR bsSearchType);
  Y+ H+ j+ v9 a( ]        HRESULT        GetDeviceServices(DevicePointer pDevice);
0 R5 @) `% d9 u/ i) M' Z- L* `        void        StartPortMapping();
! o' G8 @9 i8 M6 h2 x" W- w        HRESULT        MapPort(const ServicePointer& service);
# A8 c; [# A% L        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
  g7 B4 h) ?" i8 N        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 V( H1 \" {. @3 s% }4 i% s( D. F. U                LPCTSTR pszInArgString, CString& strResult);
( u3 F( G1 }; k& |7 d/ d        void        StopUPnPService();

; C5 a- V; [' y: y
+ i* w6 a; A8 B9 o        // Utility functions
3 h+ C! c$ W7 F. B        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
5 G- y  l3 j* \! B7 S+ |3 R# z* M        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
: f! w  M/ W4 [0 C  |6 U: M        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
5 x, h) Q3 o$ e3 f( }7 m        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
  b) ^3 u7 x1 O& Z        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
3 X8 B6 e9 b7 j5 t( @        DWORD        m_tLastEvent;        // When the last event was received?
! f) P$ N9 {2 q: j        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
8 g& l1 C( R% g        FinderPointer                        m_pDeviceFinder;
6 R& u; J" ~0 G4 C! H        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
$ I2 s8 q6 L3 X
% w5 I. ^. }5 p* h) j5 z- q- J
        LONG        m_nAsyncFindHandle;
9 J% j4 ~5 ?6 u, B& w        bool        m_bCOM;
        bool        m_bPortIsFree;
( M1 l& U+ f) d% O        CString m_sLocalIP;
        CString m_sExternalIP;
3 s3 w, K+ ^+ w$ d0 U/ U: j' F/ K        bool        m_bADSL;                // Is the device ADSL?
: h: j3 O( L+ U: Q1 U5 W9 [; m4 r        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
* c" A0 ]* h  k, o" U# H        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
) h3 D; `* E% G' P3 T        HMODULE        m_hIPHLPAPI_DLL;
$ j# M; K: }$ U/ F9 c( b4 h# }7 X* Q        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
  Y# x5 U8 q0 I& ?# _% z$ `) N        bool        m_bDisableWANIPSetup;
  W7 x  t* P7 A9 b% ]- F, y" N        bool        m_bDisableWANPPPSetup;


};

; H% u4 {- t: X: K5 d) A) d
4 E  X* S0 M7 g5 j# i7 I/ u; m' O// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
7 Q# z: H: X- k  d0 V; g/ b{
; O# |$ K: [- t: j5 i/ gpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
7 U% w- V; }0 z; A' [, o% z                : m_instance( instance )
        { m_lRefCount = 0; }

$ K9 m3 N  y* y) y; O
" r! H% W' d( F6 ?   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
/ a+ l' p) a. ^1 B: q   STDMETHODIMP_(ULONG) AddRef();
* b2 l$ ?5 z( y7 p( p' a2 ~   STDMETHODIMP_(ULONG) Release();
7 L. x4 g4 s; ~) m

  i" O0 B" E/ C0 J) p# G- H3 p; _// implementation
+ U! L) X8 r0 i9 C! `6 Y* k5 K8 U! o3 r% Aprivate:
  y/ |9 Z2 ~, O' ^        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
( B1 O3 Y$ k1 b" z! b* k: Z        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
' n) c5 C6 G3 c% f( i9 y# i+ J3 e

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
2 T# E& N$ c: u};
5 |) e3 T3 M  j
, J5 V; |6 H7 I
. G4 Z: B1 q; h" j+ c  n// Service Callback
% P( d* Z% @9 r' ]class CServiceCallback
8 |1 Q' }- h3 W  h        : public IUPnPServiceCallback
{
: T- Q& N1 \- S% B' opublic:
+ Z1 {+ Q2 S; j' p" s0 S( w        CServiceCallback(CUPnPImplWinServ& instance)
6 f3 A4 d. M' A% ~                : m_instance( instance )
( s9 F0 v4 B* U3 R        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
0 y+ B4 P+ ^" q* E0 d3 h* ~

" V  j( }* k  T. b9 s// implementation
2 R- g, u( \& Uprivate:
- ^) w0 k" ~7 s        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
& ?* u& ~: X  Y6 M* S0 M5 A        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
  T. H' {5 D4 V4 L- J/ |: w% F};
$ Z: _8 d$ n# m5 `# |
0 V7 u0 f" u% J4 q
% C1 T- c/ `8 D& `: |4 ~3 ^/////////////////////////////////////////////////
! s- {  x) i6 ?$ s+ u! y
: O+ }: x5 y( F. J
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.