UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   $ @" d& C4 a3 P  {' \3 v6 t* n
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   # H  m$ y1 c0 o/ A8 A
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ; a0 _+ N1 L( i  {. J
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
    1 |; t# t- g, I  f4 j' Y2 J
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    0 Y) N* g% E) ~: m$ k7 b. g: T
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    7 J0 T& D- Y8 h- v% T$ f3 W2 Y7 z9 `) e1 p
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    4 h# W: k5 K$ V
23. 
    % q& Y: m0 C6 V0 M  M
24. typedef   struct{
    # t+ J6 ~5 T; R  {* j
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;
    
30. 
    0 g5 e' J/ H/ ], e& S" w
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    * T1 x. A% q* c3 C0 E( L
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    9 X: a$ W3 M0 J2 ]! Y
36. void   clearNATPortMapping();
    # ]& r7 @$ I' P& p1 |8 }2 D6 U( {
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    , M: i. ?0 Z$ g
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    6 X9 N# v" \) g# s  E) p
44. void InitLocalIP();
    + M, K# G3 Z6 O( e' P9 r
45. void SetLastError(CString   error);
    
46. 
    9 F" j- @" i8 _: _! m: w% ]
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    % u& f+ j# E. l& I# P
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    , M3 G, z# q% f5 {0 u# ]
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    , r7 I3 r% J) O3 [* F
52. 
    
53. bool Search(int   version=1);
    - B! C: @( z3 y) y3 g; p
54. bool GetDescription();
    3 P% ~# E0 K) X% W) H( o
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    3 L; u2 v# O% v0 j- S5 X
57. 
    ' v% P+ B: u! V
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    6 d$ k) N- @8 }2 u% p5 s
59. bool InternalSearch(int   version);
    4 v3 k8 V( o4 U8 A6 z
60. CString m_devicename;
    $ B& x3 o5 U" O( J& O' u" H* ]
61. CString m_name;
    ! N3 Y; G# ?$ H( N" M* [6 B
62. CString m_description;
    
63. CString m_baseurl;
    # F" Q" ~1 Z9 J! v  k: M
64. CString m_controlurl;
    9 K& L, ^- w) W* f0 q, e
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    3 y8 Z1 J9 v0 K/ W3 m
73. CString m_slastError;
    " P! P! V# K$ c" Q; Z6 B$ F+ g
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    ' }! U2 p* Q7 |! a. Q0 s
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   6 ^) r2 b" k( J% q1 U1 K& K
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ h6 v' |4 ^- s1 X% Y; O
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   : e( ?/ H$ S) n9 F$ V9 [! y% N) l
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ; R: E1 F" t  F0 G% k) |1 G, Q
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ( m8 [$ I: ^, z3 ?, ]
15. 
    
16. const   CString   getString(int   i)
    
17. {
    5 \4 O: G) m5 t9 I4 ]; Y
18. CString   s;
    
19. 
    " V0 k, a0 @/ F0 ?
20. s.Format(_T( "%d "),   i);
    ; C3 }3 ]3 t- T0 O4 a
21. 
    
22. return   s;
      [8 v! p6 N" ?2 p) c, Q$ E
23. }
    3 `  h) ^3 P- y# d8 q0 p$ \
24. 
    ; S% O7 X  ?# f5 z: U! h, `) X) `
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    " m1 u& J' R0 X- }/ f( U1 \+ z
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % Y- s+ ]7 r- {' t. J3 ~2 T3 }5 a
28. }
    , q$ V5 m- [! T0 @+ \% O4 K
29. 
    1 Q9 R/ X6 ]6 ~; g
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 S$ H% ~: M, K4 H
31. {
    ! t' E1 F! y) ~$ k% o& c' B
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    * E+ i9 @# g+ |. g
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    0 ?: o' S" P* z/ `' A  }
36. {
    $ C/ |9 Q! H; ^/ V# j7 {: v2 O
37. char   buffer[10240];
    / d9 a2 g" Z$ L* Q3 \
38. 
    2 l& n+ e$ x4 t. L, |4 Z
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    : d* s/ `; X7 U( W7 P% [4 c1 `
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    % @- P# _1 k2 }: p% ^! Q! h) ~7 S
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    - |  j8 c+ a& I5 c7 @% E/ X
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + w) l$ D# m" U( ^8 n. F; q
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    0 B# ]: T* S% k. [% |( g9 |
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 z6 N5 d/ h, h5 E3 _3 m- ]4 T& }
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    3 {9 A' E3 \- N; @
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    0 \/ X' l" N/ Y, ^) G; m
57. closesocket(s);
      V! m3 ?- r+ m- L
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ( A. W$ A: F9 @. v0 q
60. 
    ! w' c( V$ y+ w5 M3 m: z7 X/ x3 }
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    ; y5 k' ?5 {! o: N1 ^
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    3 N; _4 Z  e- D) ~) X
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ! H! }: g8 B, Q% l0 m/ I5 }
72. strcpy(buffer,   (const   char*)sa);
    # g$ X- c9 P$ f' {+ z, F$ ]
73. 
    
74. struct   sockaddr_in   sockaddr;
    7 h+ \5 X) R  t. T: l
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / l9 E( [: G' e, z( x' E
84. {
    % ~# H& h' |5 j5 S5 X& A: \/ [8 }" P
85. int   pos   =   0;
    
86. 
    % q  b. o5 F8 ?9 X
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    + E2 q; Y2 `8 R$ S
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    7 ~. K  _2 D: B4 |: q( i0 m1 X& x: g
91. 
    * Y  w2 ^' A, q* \9 P' H
92. pos   =   0;
    - q, D5 [- }! }& I9 w
93. status.Tokenize(_T( "   "),   pos);
    4 X1 W* M7 D$ w+ l: i3 h; R8 s. N' K
94. status   =   status.Tokenize(_T( "   "),   pos);
    ; Z7 t) n) I& [# }8 h9 P% M
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    5 E1 v6 G  `, U! D5 p
100. {
     / O- V: W# m6 }. U5 n& p
101. CString   startTag   =   ' < '   +   name   +   '> ';
     " d  n/ {/ x0 T' o5 X7 R) W+ F
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     6 }2 C8 I( _8 j. P- L! p
103. CString   property;
     
104. 
     + i! \6 J5 l* g+ I  u1 Y2 W5 [4 S/ W
105. int   posStart   =   all.Find(startTag);
     2 o' T% j) P) E# X
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     6 Q8 v6 @& @: Y
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     % R  l9 u6 ^  t" v1 T( H: U  `/ a
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     % g$ n! j7 U; o
113. 
     ; B  z+ V% Y8 }# U
114. MyUPnP::MyUPnP()
     / A- Z5 |# c! E# J
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     1 [( A, o2 z) m1 J/ u
121. MyUPnP::~MyUPnP()
     * A6 o' g7 O/ L4 `$ q
122. {
     & J+ ?& Z1 |* B! m# @
123. UPNPNAT_MAPPING   search;
     0 k# i! ~5 w: H- N) D9 M
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . m) O6 `% y8 V6 l2 k
125. while(pos){
     + A- d# p( g: p7 b+ i, O
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     - C3 ?' z9 H& z( x3 L  }' u
128. }
     1 C- _% S) s) z! _$ O, y
129. 
     # P  R2 e0 R* L8 |# _
130. m_Mappings.RemoveAll();
     ! p5 |3 o) Z: S- Z) B+ L
131. }
     
132. 
     " c8 {9 w, p) \* U* g
133. 
     4 j! }! Y2 _6 v/ n! t& T
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     6 q8 `. s9 d4 L6 y9 _$ Z
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     9 J$ e4 g! h: Q0 J7 g% o
139. #define   NUMBEROFDEVICES 2
     ( f% w/ k9 I; B' K
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     : E  {6 U+ a, C
142. {UPNPPORTMAP0,   _T( "service ")},
     / P* h( i# ?" X2 x% j
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     & Y# N* g$ V5 G# c4 b
147. u_long   lv   =   1;
     6 P$ q9 F' f! k9 @9 B) I4 d
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     : o  n0 F+ N7 X1 ^1 m: A
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     - a1 r- i/ P* W3 G3 J! F! [
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     , |1 V" Z& H1 H0 b9 T+ F) i
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     2 L$ b. N. c5 A$ Y! m" S
159. }
     2 P# S, Q- K0 M4 }" [+ P+ a
160. }
     
161. 
     
162. Sleep(10);
     ! B/ J% P8 B2 T9 z1 @5 f5 p& L
163. 
     / x0 P% f3 U- [4 e+ R
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     , I* K) x: `2 j; O: ^, n
168. 
     7 d2 d1 @+ ?1 q' i) }
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( D3 n$ e; U# x3 y1 {# U, e
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     9 y. ]6 Q- K0 p
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     $ Y% Y! d0 E. g, j" y6 v3 O- f8 ~
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     6 b& v( q; V% v& k* s# N# z6 i( |
175. if   (result.Find(m_name)   > =   0)   {
     , ~* A" [* l$ Y6 T  v9 t
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     7 }. Q2 {8 O4 o4 H. {( |  E
178. if   (line.IsEmpty())   return   false;
     / k3 B: q& U# `. k. S( s/ P' }; b; g8 r
179. CString   name   =   line.Mid(0,   9);
     - H/ @+ y( B5 v( S
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     % z! I) o' s$ \6 s
186. }
     
187. }
     * ^. B% w3 l6 D" R! q
188. }
     8 Z$ f& s$ K* v4 k9 j& ?3 d, W
189. }
     
190. }
     2 G+ r" [3 r8 X
191. closesocket(s);
     % e( J3 A4 ^$ N
192. 
     ( i% M" p# \5 b7 W: Y- `1 Q
193. return   false;
     6 P" ?. S  Z, A3 r7 {  I
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
3 F' B3 }1 [$ i) v4 t

& a/ F. I9 w# K* a///////////////////////////////////////////
- x+ f! j; Z+ @/ p* A/ a# `, T//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

- s4 O: J/ u1 z
#pragma once
6 ^- b2 y- |  |' \0 A$ k#include <exception>
5 a3 ]7 {& W; ]. B. ?# [0 E
% E# m; L, G4 V0 V: ?0 T, F, v
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
) [! X. q/ r7 x$ a/ l        TRIS_TRUE
};

6 Y. i: d# \) `" g$ u: X9 t
enum UPNP_IMPLEMENTATION{
- w) D6 t: y* V# }8 w+ j        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
) b9 q; t* Z/ x* P- H9 q/ \) Y};
! C. M, {2 h, E) y1 X" @) O
, u5 {7 l) I3 Q6 b


class CUPnPImpl
4 q, d; P5 t' S5 y  |, n) {; x{
9 z2 t* C( S9 m1 R/ |3 npublic:
        CUPnPImpl();
3 I4 q! A/ Z$ ^: n) u  O        virtual ~CUPnPImpl();
- Z* P, f( e8 s        struct UPnPError : std::exception {};
        enum {
( Z6 i1 I# l4 [1 d$ @$ I                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
# F; N0 N6 b1 H& s% i# q9 L        };
/ Y4 [) ?- Q3 |% y% y9 D
2 r) s4 c6 s/ I* l1 y7 H! K
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
( c3 H+ b, `7 H5 t        virtual bool        CheckAndRefresh() = 0;
" {* d6 T9 ^& B* n5 E( g1 s9 }        virtual void        StopAsyncFind() = 0;
! s, t" }# x4 g, \        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
/ C+ P2 S" \0 T# D% T5 K        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

3 j9 e$ C# Z% o' m1 O6 ^" R
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
1 P( ~' V* n, ~6 D" ^        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
6 X& ]2 _$ f4 R% N        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


$ m9 J- T+ |" F// Implementation
protected:
+ H. x# U( I2 `5 e3 [4 _& O! a        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
9 [6 c' u7 Z8 Z* @4 G8 P        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
! V" W  Y  ]4 o1 ^( Y. I! _

7 @& l( e+ F0 j& G7 Bprivate:
* ]* o- D. C3 v/ G$ N3 Z        HWND        m_hResultMessageWindow;
- y) D( K0 y0 x: b/ b        UINT        m_nResultMessageID;
2 X8 p+ v# A' Z" X3 N

0 S$ G- }5 v/ P, u};

' a: K& D5 `6 f+ y& D' Y
// Dummy Implementation to be used when no other implementation is available
: K4 |  i& g" M3 Vclass CUPnPImplNone: public CUPnPImpl
7 c2 {; Y; j- ?; r/ y{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; x4 l# ?, i5 R        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
% d& O9 [! K3 i, h! {/ L. q        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
! P; P0 O+ X& }% p+ k, D9 u( H};
5 H; d% [1 ~" ]# [5 J

, i8 K) D, G4 t1 Y, N# ^/////////////////////////////////////
+ R( x1 u0 d3 U$ w//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
& L+ C- C. }% @- W#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
( g2 m! @. j/ J8 b( O#include <iphlpapi.h>
+ J# E3 b8 O5 T#include <comdef.h>
# P- a& U* b9 Y#include <winsvc.h>

7 P1 O2 ~% m% O2 Q
* ^5 |+ o% r3 ?#include <vector>
#include <exception>
: @7 A7 b4 \  L% u; E* u% N- d#include <functional>
0 K$ k& w& R: p* g* ~8 U9 {  P9 I
' |4 T/ r* j1 }; S- b


5 `. E/ o. g* A7 L8 Y7 {) ftypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
) b0 F5 E7 c  \. [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;
- m. {" b- m. P8 _; z( stypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 H6 g  j, ~. H9 S: ctypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
# ]- q% m# h2 j) c8 P$ t! gtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

8 W  f. S2 F& v3 p) d# K
typedef DWORD (WINAPI* TGetBestInterface) (
4 X) n) Y$ H5 u8 L( [3 i$ ~  IPAddr dwDestAddr,
9 R; u' c6 D& s# V8 X  PDWORD pdwBestIfIndex
);

. G6 i0 K' j% n6 N/ E6 g. t8 q( O2 n
# I8 E8 m4 q: L2 I! S1 ~0 u( Ntypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
& u) N. Y2 I9 ], T1 f, m) e  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
0 m1 S- n# k* Z% S);
' E# ^( U: V9 N) v

& Z$ j. y. W' S! y" m0 ECString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

$ g& I6 A7 f! I% J) U
+ I0 t/ |7 \% Gclass CUPnPImplWinServ: public CUPnPImpl
7 i- [# X# s  H9 n, ~4 ]{
2 ?1 v2 L+ d  e: X0 G        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
) l/ J8 L9 a2 R) i// Construction
# r" B- ~! E  E1 v8 R% Ypublic:
' b. j$ _* }# M$ _        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


2 }& d' I9 p9 T+ ?0 u) L        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
' U0 ^- g7 P3 O  J+ W        virtual bool        IsReady();
. N2 N8 l5 q% a* d5 i        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

3 L7 L8 z8 I( R, E: b+ z- Q7 r
        // 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
' l2 k" ]8 i2 E6 g' C        virtual bool        CheckAndRefresh()                                                                                { return false; };
+ Q# u8 O- }4 E8 E

1 O/ w  Z; d* ~& X$ q3 eprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
! T' N' |; D9 ?4 L/ U9 z, v        bool        OnSearchComplete();
        void        Init();
+ E* T; e' d& E) U+ [, L* u( n
* m% |3 ^: z; x5 m
        inline bool IsAsyncFindRunning()
. D- h' d: ~  K# U        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
# `4 @; Q; |- S5 G6 E" y                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
7 b1 L* D) D$ f; N, A                {
" q/ `+ Z6 i& s) q  D5 L) a8 a. U                        TranslateMessage( &msg );
6 b( }5 `- q4 B' I                        DispatchMessage( &msg );
" K% a5 f) @/ _                }
9 a4 G9 O: \8 n5 M6 B4 s' d) T: N5 {                return m_bAsyncFindRunning;
5 E9 ]  X. r! l1 K9 f4 V2 R) F2 }        }


        TRISTATE                        m_bUPnPDeviceConnected;

( z$ ?% H' X- u# d. r
// Implementation
/ k! k% K% t/ t( h8 V& b; ^9 n        // API functions
  T9 y/ d; Z1 N# y9 u, p/ U        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
1 ]& L! f/ J. o0 z' ]        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
. n7 k: S" F0 b  J9 s- R& i" q& }5 [  I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
( R+ W" G- @! x0 N0 g; u        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
' S5 J/ G' t* e: \. |. o! P

        static FinderPointer CreateFinderInstance();
5 L+ x8 e9 R* i+ a6 l& ~# w        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
! G. F4 a7 m' A/ J" s, c$ f                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
" v% u% Z& i6 o* A* c                result_type operator()(argument_type device) const
                {
( z/ @1 I+ U) T, I' ]- T/ B1 R                        CComBSTR deviceName;
* Q: v5 e8 q2 t8 e4 |                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

$ U1 y$ Y( t5 h  P: j# C
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
: e3 h; ?& ]' x9 i

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
4 t6 ?: m* j+ P/ v6 B; @                CComBSTR m_udn;
        };
' @* ~& P2 @9 q! v9 x; N6 y$ N# t* ?/ J       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
+ z, m' I5 t' `3 W/ x, o        HRESULT        GetDeviceServices(DevicePointer pDevice);
0 Q2 v0 [) O1 S) c% ?4 O5 E2 B        void        StartPortMapping();
, U& _, o' Y4 w- m* j        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
) M0 W0 m1 X6 d) s( z        void        CreatePortMappings(ServicePointer pService);
6 a9 b' {# J. S( x        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
+ B+ t; |- M7 I5 R9 A' q; d) l' S                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


- G1 ~( ~5 C4 j6 O8 o        // Utility functions
/ d! D6 j3 p; X0 ^# S" d  K6 `        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
2 h# B9 X8 z" J7 ~, b3 g0 E        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 d; A: ]+ E8 g/ v8 m) v0 X, v        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
0 }7 l) v5 W) z( U        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
8 l3 S. x6 i5 {1 D+ E7 v: ]' ]) J( H  k        CString        GetLocalRoutableIP(ServicePointer pService);
) n) b2 d* g% t% d% d2 I) w/ X+ X
2 H- m% ~- m% X0 k3 [
9 B1 D2 E! c, r4 O' e// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
. K! c% @1 Q/ L( n) m  k6 P        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
$ y9 h. u: Y& s* t5 W- R+ w# C' J0 x

8 U3 e7 p+ j8 X' M/ E        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
& k& d' J" `: g3 Y( H* f. ?        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
7 ]$ h( }; B% r4 ?5 \# _, B. F        HMODULE m_hADVAPI32_DLL;
# N/ W5 k, U. S+ T& ~4 v        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
0 H% x# }# E0 {" h
; [3 G1 w9 p9 W  f- e
// DeviceFinder Callback
9 @6 o$ D7 v0 E' }3 y8 Qclass CDeviceFinderCallback
7 f2 \* _, f2 y, B        : public IUPnPDeviceFinderCallback
7 Y* Y( D* J8 |{
5 \: f& r* o5 m# B1 S) Fpublic:
4 q0 s! o  @" H' r  B& v2 D        CDeviceFinderCallback(CUPnPImplWinServ& instance)
- F: P" _% k2 u$ B9 x1 K                : m_instance( instance )
        { m_lRefCount = 0; }

* M8 V( X5 a  T+ }
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 H# g$ w( d4 t   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


3 d2 r8 z/ i# y# Y// implementation
/ b+ b3 Y$ e. U) ^# V' `2 oprivate:
# |1 [% O, ^( X" a- W        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
/ x) O8 u2 `1 j        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  C( k- v: m! X& y1 j2 G        HRESULT __stdcall SearchComplete(LONG nFindData);
+ ^4 J1 `: Z0 e! t" u- C: }- f% V

  ~5 D7 p8 l6 i) kprivate:
7 J8 `) s* E# p$ O$ B. x. o6 l8 @2 n        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
' t5 ^/ a5 @/ j" }};

5 ?4 u' d) j6 h6 P5 R$ a
9 Z6 \3 [$ \. {: K' \  ?// Service Callback
$ G: |/ F8 s9 I! R2 A. zclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
1 h' V8 [1 _& K, E% `0 ^1 _                : m_instance( instance )
        { m_lRefCount = 0; }
7 y. ?2 C9 p. N( g  |- ?. |   
+ ~8 K1 |; b0 A; `0 `, _" t. W# C   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 m  [: T5 u0 `# ]- h; o$ v   STDMETHODIMP_(ULONG) AddRef();
$ |# l; S  o3 G2 b8 E   STDMETHODIMP_(ULONG) Release();

" z8 D5 D) R& A! S1 U5 _, W
// implementation
private:
% [" C- k9 v. F3 T. B5 ^        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
, w6 U4 B% d* ]5 Z' v2 m        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
4 s- s4 l2 b; U2 X& \% z6 K        LONG m_lRefCount;
; f+ m+ c( |3 A1 J- u- l- }};
- H5 Q0 R) X1 y3 m) s  h
9 W; n* n( y( w3 v/ F. e
/////////////////////////////////////////////////


8 K6 s- v2 [1 d' M1 U使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( t3 C+ w  k5 D7 U1 lCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
) e6 l* k( c: a. [$ h- FCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.