UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   / @1 P3 D  q! I! F0 c
3. 
   
4. #pragma   once
   3 u! \) E2 @! D! W- f# O
5. 
   / K. H" n* C: I  B4 l. t. \) P
6. typedef   unsigned   long   ulong;
   * I8 H, Q6 R& l* h% Q" w6 ^4 V9 y% P9 K
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    / r3 S3 i$ G+ W6 b
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    : L6 X2 K4 y8 G5 b: g( C
18. 
    1 N/ O' M+ A: L- [3 ~" I( O; U
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    0 C" W) d0 ~) u8 M! t
21. UNAT_UDP //   UDP   Protocol
    ! l1 Y1 c* R0 _' @
22. }   UPNPNAT_PROTOCOL;
    % p( @) n: v# [3 I
23. 
    - p; H; L+ H* G, F. S
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    9 e; P5 ]/ `1 H) r0 J) q
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    8 Q# N# c$ W' L# o
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    4 g0 ~- G5 `$ x$ f4 `9 T
30. 
    9 Z8 B+ T/ n5 J* C4 Y# s
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    " O9 W! P. r, G+ [, J
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    ' ?0 R' Y  I/ e. P/ D# B
36. void   clearNATPortMapping();
    
37. 
    6 s9 z' r" u$ x7 E" q, Y
38. CString GetLastError();
    6 o+ S! I; A2 x. G7 r# l3 I& N
39. CString GetLocalIPStr();
    * D9 l5 B" F$ _  q3 V) A
40. WORD GetLocalIP();
    4 [9 h# _* O& s) Q* X
41. bool IsLANIP(WORD   nIP);
    $ s9 o% e7 }/ s* o: V
42. 
    / n9 A- B: z" R4 |; O4 L
43. protected:
    * A/ ?; C3 Z) d+ `8 E
44. void InitLocalIP();
    2 I. O: h) m, U$ Q! R0 o
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    & N9 `& s. H& |' w* C0 U/ P( y
48.       const   CString&   descri,   const   CString&   type);
    : a' t! L- |# i2 m( w3 x* ]* P
49. bool   deletePortmap(int   eport,   const   CString&   type);
    , X& C# N/ v1 G' g' q5 V  `5 N
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    . }" D0 P6 }) f; s( _
52. 
    4 a6 D* N2 n; Q
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    6 [9 h* W- `8 |  V* D5 U
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    & ~9 s6 _, C7 L8 h
57. 
    1 Q" G+ c2 F) w7 c. p; ]
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    5 O6 K! j6 l; k2 z( V4 Z9 c# f
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    1 \/ s! ]% X  n7 @
64. CString m_controlurl;
    % @9 b+ z# d  j2 O
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    & W1 {" x9 E8 J0 z% Q( `
68. 
    
69. private:
    ( t  B% ~4 C+ `+ g4 b* c' a% S
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    0 A4 J% c- `, X  Y
71. 
    9 L6 D: Z1 Y, U4 f
72. CString m_slocalIP;
    8 G5 J" @7 f3 o: Z7 G; [3 U
73. CString m_slastError;
    0 t( X$ b, @8 x9 e- P
74. WORD m_uLocalIP;
    : i1 e+ D$ y, Z& R2 g$ e- ^* O5 i
75. 
    
76. bool isSearched;
    1 d, q' z! E1 ^7 ]
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ( }/ ~/ r4 l- t/ h6 V
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
     d7 ?0 l/ ?0 e6 j4 k
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
     q9 b1 Q1 o5 @9 J" \" o# G. o
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    # {* C" |! ^7 F0 C+ m
11. 
    2 I! U; y0 k1 K+ M6 l3 G! [: Z; e
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      g1 [( n8 k. x5 i; l
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    / E/ @9 q& x2 Z4 K
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    4 ?. N  k0 D* i+ t  O
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    ' D7 }) K9 D7 Z& s( b
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ( l' y7 Q% X4 C! S0 ^% B
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    : l* p. f4 e- \+ j# p- [
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    2 Y$ |7 B" J; b& y0 D
31. {
    9 ?! T5 R: W& q
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ( C; z  M) C1 O* ]
33. }
    3 Y5 R# ^+ B+ c1 N  {8 O% w
34. 
    & m2 I; X+ g& k. B7 P1 g) V
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    " p( P) V& f! b
36. {
    " W3 g  V  |% q
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    8 g8 C' s/ S9 @6 m) _( ^
40. int   length   =   sa.GetLength();
    - Z( T: Q' I. j1 ^1 l) S
41. strcpy(buffer,   (const   char*)sa);
    - W/ S* k0 A% k. l/ o/ N
42. 
    % s8 E6 ~9 D9 H: F) q* G9 v3 Z
43. uint32   ip   =   inet_addr(CStringA(addr));
    1 X6 o! [& \  u
44. struct   sockaddr_in   sockaddr;
    # R) V* g2 p) @; M1 S( |. d
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    , R, c5 d: W( P* U$ B
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ( h' z) o% ]8 K- I
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ( j3 L) F0 r, S$ q& B3 t, E: m! u
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ `1 F/ I8 ^* X) y
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    / \% o. G) ~) H/ V
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ( E) f8 s, h( i6 n6 J7 Z
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    # [# A; b/ x0 t
62. 
    ! A6 N' [& b( ^% V3 R; L. ^5 m
63. return   true;
    
64. }
    , c& i/ J+ o$ S9 |% G) j
65. 
    ; F( k& c# p0 a; W4 Y
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    1 u9 g7 V) c7 ?3 f2 W
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ) u' F- S# k+ U, [
71. int   length   =   sa.GetLength();
    , q  \# k0 ?/ x; |$ |* x
72. strcpy(buffer,   (const   char*)sa);
    3 i' t, E3 V7 ]" R- O
73. 
    9 ?( L$ m/ P: |3 W
74. struct   sockaddr_in   sockaddr;
    2 o: l& l3 q) I8 V5 ]% ]& z
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    $ v/ A" F8 T& I! w5 f; l
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 V- e  p' ~9 x2 t2 Y6 y
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    0 v: F/ O1 ~+ {
81. }
    4 V0 |/ Z! t+ r; W, D+ _, m. D
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    0 P7 [" d2 M; X, T4 B% I2 Q
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    7 u$ s. S% u+ g: D% A, |2 p  \
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    2 }: U# z& F, r* ^+ ]  B
94. status   =   status.Tokenize(_T( "   "),   pos);
    ( P5 `( n$ y$ X, }# w1 y
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    & `* |( j3 {  M! q4 g
96. return   true;
    2 j3 Q4 B) m, a7 K5 O) T' K3 N5 E
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     * {# v, c6 M8 T# s) p" x  ~
101. CString   startTag   =   ' < '   +   name   +   '> ';
     - Q( O8 i- m" F' O' x; ?( R3 B7 @( c
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     0 h8 ]4 q' M- q, G
104. 
     $ o* B& u+ ~  i1 x4 v$ ]9 b$ H/ L: [
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     7 a9 e$ r9 G( Q8 o0 D# q
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     + Z! H! V5 m& ]- Y0 b% e
112. }
     * z+ D3 q: P, k1 S
113. 
     
114. MyUPnP::MyUPnP()
     2 O4 O: {) k1 }, F+ Y
115. :   m_version(1)
     $ z& e& C  g" d! ~" [
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     $ i& y0 S9 H4 z  S- D
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     2 W; X" r" l4 u1 c5 q& J; @. M
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     , b  @: A' k- B5 D2 a4 n, n
127. RemoveNATPortMapping(search,   false);
     # @8 }- A* w: z6 R5 u2 w
128. }
     
129. 
       [7 @( |- l  W5 ?4 a
130. m_Mappings.RemoveAll();
     
131. }
     $ {. {2 `2 u6 q" t- ?* Q& T' w
132. 
     : e/ X1 d/ A/ u
133. 
     ! Z/ n- k( b2 G% J; Y6 e/ w  P
134. bool   MyUPnP::InternalSearch(int   version)
     ) |( G; f; y2 Y7 t* n1 l6 t
135. {
     7 r  C& o# l/ ?
136. if(version <=0)version   =   1;
     1 H3 s. l4 E( K5 u+ R4 j& t
137. m_version   =   version;
     
138. 
     2 q6 X3 q7 t+ Z& |* G7 P
139. #define   NUMBEROFDEVICES 2
     " \9 x' c, h' V' M
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     - ~. d# Y3 t5 ^) D; m
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     & n, T; _7 H3 B
148. ioctlsocket(s,   FIONBIO,   &lv);
     & v: d9 r4 t# y$ w, k% P" w
149. 
     - z3 k# @) N. Z' p! \) z7 e0 p' U
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     & c/ o6 v! ?; o9 w( j1 v
152. if   (!(i%100))   {
     ! i; `2 s1 m& w) q. g$ E. [4 V
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);
     6 G; C, k0 Y* Y* C) e0 B% n
155. CString   request;
     4 r8 ~/ S4 Q3 r1 b, m3 j
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 "),
     ' K1 z: |+ O  C2 W* n: O
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     " X' J$ N% P9 D4 E' g
160. }
     ( l( \' l& D# [
161. 
     
162. Sleep(10);
     , ?, _% U/ b( R% A" w) x
163. 
     
164. char   buffer[10240];
     4 w# ^: K" I& J" v; E; u
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     * g  k% X$ N% _& S
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ( k& C( P3 P; W/ q! W/ a
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     : P7 {' f2 `. E$ k; J$ H7 q# A( \
172. 
     ' y7 Y' w0 ^6 J* Q
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)   {
     # n! t" b( K1 E3 M
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     # i9 ~1 O! ^/ q) z" l
179. CString   name   =   line.Mid(0,   9);
     ; ^6 U" n+ A& I' T1 ?
180. name.MakeUpper();
     * T  g# ?: R# m& D; `# I3 U. k. i& X
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     1 V2 X" K, Z! X* N# ?- {
186. }
     , {- |( k1 _" |+ q3 Y" n* a: a
187. }
     
188. }
     # r& F' |0 O- s$ s$ @
189. }
     # u+ d% w! N" Z2 x4 v4 J# ]1 p! \
190. }
     0 `! p4 H3 p1 U  \
191. closesocket(s);
     
192. 
     5 ~6 G0 J9 X( d3 `& O
193. return   false;
     ; Z$ {7 A- ~$ Z* B/ u. n
194. }
     

复制代码

天云上人

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

9 o6 c8 \& Y. w8 x- E6 V1 O
2 ]; J# ~8 n3 g/ }- V) y///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

  Q- h5 y% M+ `- z+ r- a1 G
#pragma once
#include <exception>
) Z% N9 A/ r+ J
& U, O& G' k- F) X
  enum TRISTATE{
        TRIS_FALSE,
$ u0 f3 @) D* S9 L        TRIS_UNKNOWN,
        TRIS_TRUE
* E5 [* U; D' t6 x. C};

, _9 P9 A/ D1 R% i, m) B
( H7 N# J/ P) i! @! L7 l  c+ Eenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
" W) a7 a. O5 Y8 i6 i4 d( l- f. `! j/ i        UPNP_IMPL_MINIUPNPLIB,
; ], f# |8 a9 y+ `; A( X& h        UPNP_IMPL_NONE /*last*/
};
+ B! [. F" Q2 i4 C
' B& |5 O$ B* V& I1 E' F
8 {4 P1 |& x% D3 g

, `, O! w  G4 J3 k6 a' J' {5 J1 fclass CUPnPImpl
: f+ ^8 R& N, E  f( V$ J{
public:
        CUPnPImpl();
5 |+ [9 i* p1 ~6 Q- N        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
$ ?5 J. w0 C: B. A0 l        enum {
                UPNP_OK,
+ L( P% H6 k$ S$ m+ k6 r, g5 s                UPNP_FAILED,
; j7 O# r: E( t                UPNP_TIMEOUT
        };

, x9 K! X: _8 y
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
# O$ x. J. j3 I7 t% I1 `        virtual void        StopAsyncFind() = 0;
+ M4 e1 t9 D2 d% C        virtual void        DeletePorts() = 0;
. }0 ], `$ Y5 d" Y$ y1 O: H        virtual bool        IsReady() = 0;
" E# a$ A& _2 G9 r" v        virtual int                GetImplementationID() = 0;
% M% t( [, C. I- F( n! d$ n       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

" h* ~& D! `% b$ l* j
( Q% b' ^4 D0 u& ~        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
. ^4 O  G% l6 }( u        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
" y1 \1 [* X/ y* j9 W* E3 C, C        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
' p6 ?: h4 z5 m. S" i        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
+ F$ L; c9 Q0 u
- P7 _% w8 [5 Q1 a+ v
. n6 u: ~! w% L% d+ c  t// Implementation
4 f/ F6 J  v' f5 |' dprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
+ ?5 T2 l6 G7 u; m& _  m5 c        void                                SendResultMessage();
4 f# [( D9 N, Q% Z        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
; M3 q% S+ j# v        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
) Q( G" w& ~! o! s1 @! O
, z( i8 a% x5 E8 t6 W
- C" H4 Z: F) {, uprivate:
- t( [. {' T' v2 p) X& |* Q0 o        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
+ ^, t2 O+ m5 P
1 [3 C6 N, ]2 _% m7 T1 Q* i" E4 N
. w, e- F" T# X) O) y+ D! L// Dummy Implementation to be used when no other implementation is available
) P7 f( M( l" W# H$ e/ Qclass CUPnPImplNone: public CUPnPImpl
! @+ ]' Z0 H! M$ W. T{
public:
, m4 b! E" f0 d, @$ ]" t2 V        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
% j9 i# l2 _, @8 v4 J3 J        virtual bool        IsReady()                                                                                                { return false; }
, R6 T2 ]$ i: H* n( w        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
* h  e' {( f2 c- b* e6 h; \};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


1 D% d* o7 ~& u/ K  ^; Q0 |. S#pragma once
#pragma warning( disable: 4355 )

. i- b9 ]" t+ }; \, P% g
#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
/ H% q: U& S6 u6 {$ M) g+ ^#include <comdef.h>
; n5 ~+ k/ O& M5 `#include <winsvc.h>


#include <vector>
& }& v- Y  U$ }6 N/ E1 X  {9 `#include <exception>
4 N- I" k; ^' ]#include <functional>
) Q& G' j- G% a$ S
5 ~! r1 o& }5 m& ~9 i/ j( |6 r


; _/ a$ ]$ q" N9 i6 Otypedef _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;
+ q% \8 x" G- z! Itypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
; f/ z8 ], m; G- a3 A4 ktypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
1 t1 K, Z" S3 [  X* A, S- Ftypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
4 v, L, J( m: S. o$ d- Y0 l7 x( Itypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
$ ^% G) A; @) A1 F

3 Z/ D; v, I0 x; c$ C$ {( ?& Ttypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
8 B/ p2 N% h" V5 F; p  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
+ M- j% M4 ^7 S1 \9 F: q) X  BOOL bOrder
);
( Q2 n  D2 M: V. {# J

) f; \7 U3 N# ~) s) r% T/ N0 ttypedef DWORD (WINAPI* TGetIfEntry) (
( y8 [- J; j7 Z8 Z7 t& A  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
) B2 d& g. {6 r5 L; ^8 A* T9 }HRESULT UPnPMessage(HRESULT hr);


4 a7 s& O9 L9 b! F" y2 Hclass CUPnPImplWinServ: public CUPnPImpl
  U9 j" J3 s1 z$ ^& {( ?{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
) r% B% r  ~9 m7 X0 X* o1 h8 [public:
9 z5 Z! _. H' }* f; L* _        virtual ~CUPnPImplWinServ();
2 z6 [. _; ]! I# A5 o& A2 n        CUPnPImplWinServ();
# @" m' ~" `( w/ G% A! f
1 a8 X# D, ?) h1 v0 [  E. x
% a% Y1 C) l' s. b        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
% Y) h2 \( k' N3 j! A! _        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
3 C+ r0 e" L! O& E5 ]8 m        virtual bool        IsReady();
# f5 x# Y  }5 r, p5 U        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
$ p4 @3 t4 {# R5 e( I5 {

        // 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
$ _. p. ]& e) p3 u        virtual bool        CheckAndRefresh()                                                                                { return false; };

+ }4 ?' C" k  d) G
3 Z$ h0 s, s# mprotected:
$ W( C5 U, @1 L" J: M2 w: n, j        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
& Q4 y/ a% ^) h0 F2 L5 ^( o        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
  z: T) r: v1 C        void        RemoveDevice(CComBSTR bsUDN);
# n+ \% {- g! z; V7 i3 g        bool        OnSearchComplete();
        void        Init();

9 V( A; }0 m3 }- d! \, r8 N
        inline bool IsAsyncFindRunning()
; y8 l. t4 f  D, u1 W7 l$ c# R        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
' c' n. o. H8 x                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
; y( m) k/ Z- ?( L# a) u                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
2 M% H0 Q. z9 U/ k1 Q                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
1 K5 }" W+ p. ]                }
                return m_bAsyncFindRunning;
        }
3 d/ y3 }5 v9 J7 y( @8 x1 H
' V, {: d0 w) Z1 p9 B$ G% ?) S
        TRISTATE                        m_bUPnPDeviceConnected;

+ `& w) p" V/ G: L! I) ?
( ^/ p7 h2 \4 t8 H: L// Implementation
        // API functions
4 a5 O2 x/ V, w) D7 P        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
/ G2 {# w* z$ Q+ _( b        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
3 s4 J( z, d) d: V        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
8 H" k$ [! H$ f8 U% c) p        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
, G6 A. j- ]! Z( F; K% \

1 U( N! Z( N4 H5 o        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
  C& i+ C+ E5 B+ o( z3 a        TGetIfEntry                                m_pfGetIfEntry;

) N9 r- q$ {; S9 U
9 `3 j5 y# s/ g8 O  P7 o% f9 H        static FinderPointer CreateFinderInstance();
" @4 {+ F) u+ g3 I. l' ?- L( A        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
: ]& Z" g1 a/ h: b) F! L                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ w% t9 h  H9 G0 I                result_type operator()(argument_type device) const
5 r1 g' [# ^" T                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
) N8 ?$ s4 d: H* z

  q1 h8 W. S. |7 X: k                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

' m8 u: q& B- h
% N- ^$ f/ U+ _: ?$ d: A                        return wcscmp( deviceName.m_str, m_udn ) == 0;
5 U( d# s8 c6 ^                }
: e  h/ v: b% U% d, M                CComBSTR m_udn;
: _" B9 I( {4 k( ~5 ~7 {; w" z        };
       
. L" E* v7 T( d        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
: f' d- L7 f) `; w+ R+ V        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
7 W, `8 ^' V# g' r; [3 }        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
( `. ^* c) [3 F5 C. Y7 N
1 P  Y  R3 K0 N& i+ Q
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
2 }. K0 o$ [4 M2 I2 O, x& \        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
1 Y3 h. g0 V3 a  j5 E$ y, c        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
( a* K/ j% }4 W5 O        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
& }5 N5 r: X. A* Y( b; c% @' g/ h        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
# y5 @% `6 Q4 `: I        CString        GetLocalRoutableIP(ServicePointer pService);
2 O+ `9 Y- l9 n/ q- u

// Private members
* o$ ?% @$ @3 ^" k" M( H; V* l: P& n; jprivate:
) h) n" F+ q, R        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
* w. }5 `% {* u8 Z        DeviceFinderCallback        m_pDeviceFinderCallback;
7 h, ~' c/ f0 f, s$ D* A% D        ServiceCallback                        m_pServiceCallback;
# L5 e+ ^  U0 p

" f& B9 g/ M; }% W        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
) x4 {6 Y9 H  g        CString m_sLocalIP;
: K) Q  d: F& o; q4 p7 d! X* X        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
% C1 N& A1 P$ Z4 [' @, R8 B        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
+ t+ e$ Q& v7 @+ Q! P. o        bool        m_bInited;
* X( T9 l% t- A$ O. `' q        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
3 R: D$ S) {9 V& q

// DeviceFinder Callback
2 o$ i+ @. N3 C8 I% gclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
+ `6 D5 M$ O9 R6 _! Wpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
- f  j5 c3 g" p9 n        { m_lRefCount = 0; }


( k! {' w& H) H( a) A   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 ?1 v# k9 R/ ?8 o   STDMETHODIMP_(ULONG) AddRef();
; B4 _- I6 W, J+ @6 l7 o( k   STDMETHODIMP_(ULONG) Release();

- r, |; @5 t$ G( s, m
& L' t$ ^- ?1 p4 H// implementation
private:
# T/ ]* Y3 u8 B        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
4 H! X+ y7 p& F9 o        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
( t8 e2 i/ }3 u% v- ?        HRESULT __stdcall SearchComplete(LONG nFindData);
% f1 M2 g# x  q( _1 O7 X! {

private:
1 r8 ]4 z  ?: }7 H3 r$ t        CUPnPImplWinServ& m_instance;
4 f8 b  b+ X4 g& X        LONG m_lRefCount;
9 F  ^4 l2 u, L! j- t* B; z' m2 N. s};
, [; N' X! W0 w! u

7 S) g1 f. n$ e" w" M// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
2 N; @4 ~% T! d{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


; i- T! b2 l! X, Q, ]1 o% M// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
; p8 I  @5 T0 o7 M        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
* t4 B7 I1 L* z$ `( w' N* P" [$ u1 w

5 i/ t, V3 M/ _$ U' G) cprivate:
6 k% R, [% G% `0 m3 Y; l$ ?9 Z. l        CUPnPImplWinServ& m_instance;
( V: ?# R/ y' E2 }( J        LONG m_lRefCount;
};

+ s; |) g) s6 T+ B; b
, ~; V; E1 x; a7 `" ~2 v# H/////////////////////////////////////////////////
  g  U. ^, R. P" {
+ W  g3 K9 e+ e5 X% i
使用时只需要使用抽象类的接口。
; X( B# U1 ]0 t2 MCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
) y) R% L, _% v8 @CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
8 \# u+ C6 a# m0 BCUPnPImpl::DeletePorts删除端口映射.