UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   7 h% A* y/ A1 o: j3 p( z/ V
3. 
   ( L4 b+ \- M; h4 ]& z( h( F
4. #pragma   once
   
5. 
   % i3 }: _8 w. B+ [& |
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   ( J! |0 X) e! Z: t5 `& w! }. x
10. public:
    
11. typedef   enum{
    7 I& T$ g! E% h5 G. H3 m; D) g
12. UNAT_OK, //   Successfull
    
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
    3 [, @. y* N4 E  o
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    ; x# h3 [, f4 }# C* F' R7 C
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    ( p) ^' O5 @* M5 k
17. }   UPNPNAT_RETURN;
    $ j& R  Y1 p/ J8 ]/ c) e& ^2 \- B2 l
18. 
    
19. typedef   enum{
    " J3 j$ u  t% i$ C: B
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    ; M1 Q9 o# `" c# b; ~
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    4 ]' ?5 C* ^7 }3 I$ `
26. WORD   externalPort; //   Port   mapping   external   port
    3 @* o0 u8 B0 U/ m
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ' f) i9 N/ t% b* Y
28. CString   description; //   Port   mapping   description
    # v: x1 z% u- e! |; z
29. }   UPNPNAT_MAPPING;
    
30. 
    ) y. W$ X$ O$ C8 P+ {; g
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    & S( k+ o. A( v# w/ D: u
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    ; M: q. R* _$ V$ v7 r6 g
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
      u! j2 v0 b! x+ P# N3 Q( c* h
43. protected:
    ; X: V; }; Y) E/ ~6 D
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    : f3 o# @: G% \* d9 s# E, e' A
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    4 J3 E/ L: K4 \  Q2 X3 {% [/ G
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    - T' ~- C, `2 M' M! _" s; b; V6 v, _( |
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ( c( |, _. Y2 c- ?/ O; O
53. bool Search(int   version=1);
      G( E5 k/ W) l
54. bool GetDescription();
    : i- n. G6 }8 b
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! ?3 I" m( v- ?  W7 O8 R# N
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    9 n- Z# ^2 [1 S/ G
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    1 K# Y$ E0 W: ?- s) h
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    4 N4 R8 l. n0 f- Y6 [
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    * S- K& G* J7 N$ I: E6 b. z
66. CString m_modelname;
    
67. int m_version;
    
68. 
    4 ~- x: K( s( ?
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    - a8 i/ W% d" r6 ]1 |* x+ Y( M
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    % z+ ~2 p# ^5 b2 b4 v
74. WORD m_uLocalIP;
    " y+ T0 ~$ c6 Z. B( _
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   7 c3 f- e5 Y1 ?' V
3. 
   
4. #include   "upnp.h "
   
5. 
   4 ^) Z7 i) Q) v3 G. L
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   1 g" [' @$ X1 l( _) l" ]
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
     ?7 t1 A: D) [( F$ v* s( b: s  Y$ d
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    * w( J) j- G: _! [0 J) B9 T
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    & A6 h' u# _& Z1 w" |3 o
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    $ y+ a4 k+ @; |2 @; T& G; q
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    8 l. q% S# \4 M2 [0 K
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    # g2 D1 @- F' [' [3 R5 M
23. }
    , u+ S- m& {/ E. W2 [
24. 
    6 s1 o8 ]$ d9 u( T4 T2 N0 x2 p
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ' x- C( x& i% P/ k3 C
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    $ z; u& X' B* z8 a5 Z* C/ R
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    5 F0 C  J6 y( q" H4 J  y
33. }
    
34. 
      c, r* y" r0 H+ B% x5 T8 J
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    / ^% d( Q: A5 F4 Q" O( R
38. 
    ) k& w1 u, T( c4 z8 [2 B- c
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    / A  G6 K& X  a" d% {% F& M
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    " a8 n! z! k, Z6 L. k% I! H0 E1 _: h
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    * H2 f9 G  q( X$ @
46. sockaddr.sin_family   =   AF_INET;
    , r! o. n- M% l% U  @3 Q$ M+ D( x
47. sockaddr.sin_port   =   htons(port);
    . k" v5 {5 u( ^8 b$ T9 O* g
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    : P% u- O  p/ y2 G9 f% p
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    - s9 b8 \% N' J
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    0 N4 v# p$ k5 z4 p
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * j5 P/ f( ~6 _: ]7 V- ?( X, e
57. closesocket(s);
    ! ?: x" ^% k) p; w/ t3 L
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    1 I# l, T( R  d) v0 L1 l1 |
59. if   (!rlen)   return   false;
    
60. 
    . w9 L0 r; b' Z- L9 n4 p  v
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    : I" P2 N' S  C: n+ N
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    " M& _* a* C3 r) |
67. {
    ) _" r5 [- E% T) g, ~0 K
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ' ^! U2 ~4 \8 l. a: y
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ' m) W" m9 P) G+ {+ v5 K( e
73. 
    
74. struct   sockaddr_in   sockaddr;
    & o4 ~& M2 G5 C  G: W8 o) @7 @) M
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ( n9 x9 j0 I. b- {& y0 A  B; @
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    , _: b; E0 |, s' {
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    5 Z* C2 C, j/ z2 A
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    4 v7 ^; }* C" {2 o
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    $ r. A) V/ s6 b7 a9 y: k
86. 
    8 D( Q0 A  T6 _# r6 ~& R, X
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    % |& B' b4 @3 N9 B. ]1 M+ w
88. 
    7 e& I6 U7 I7 C) q, w5 O
89. result   =   response;
    
90. result.Delete(0,   pos);
    2 c$ ]+ k3 q  [
91. 
    , C8 M7 F" [" C0 }' D2 h
92. pos   =   0;
    1 I5 v3 e' y$ K, x, \4 |
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    6 w) {" n5 C1 N. I
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ; u5 g/ V6 d& v+ R; i
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # D5 X: \* n9 f
103. CString   property;
     , g- T5 L  d+ X5 j  f
104. 
     , a4 p/ Q1 w$ ?* d" E* B7 e/ q6 |
105. int   posStart   =   all.Find(startTag);
     % U% b( C: G+ B0 {3 W5 ?3 Y1 }
106. if   (posStart <0)   return   CString();
     
107. 
     3 }2 `6 A7 ]* c8 O. }# u' d
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     6 U9 H; ?. d7 ^  }
110. 
     , v. _: |3 P2 I) J- z! H
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     7 ^+ |. `) ]3 M# T6 a
112. }
     3 z7 V9 l4 E" t% t" T, G" P
113. 
     
114. MyUPnP::MyUPnP()
     - K2 b. K( L5 a$ r' t: K6 _% A+ J+ r
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     / ]5 h; N4 C. p
118. isSearched   =   false;
     
119. }
     ( y# ~! ~, C- [7 X" ]
120. 
     6 G* H, p: @9 V
121. MyUPnP::~MyUPnP()
     0 u' {% g+ u* a$ u7 `# ^7 Y
122. {
     : D3 h5 A0 F2 l2 Z2 {/ u. d
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     9 [/ W# P2 D% }& L2 ^; I3 }' x
130. m_Mappings.RemoveAll();
     : ^6 s* K9 d& y5 f: V8 f
131. }
     
132. 
     
133. 
     ( D% R2 L; O* m8 U: T$ ^0 j6 s- p
134. bool   MyUPnP::InternalSearch(int   version)
     2 S  T% N; d  \+ @4 D
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     , b- w' p/ W* R$ N8 {
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     , y3 s9 u! u& t) E# a
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     1 V; S* R0 l7 f$ w; K& N
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     ( S- d. _. |3 o% N# V3 J( ?+ d
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     # }8 P$ k  D/ l: m! q4 b8 e
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     5 W- E( S) R* @2 n0 ?) b
155. CString   request;
       _) q5 l' @9 W1 u5 y$ u/ W: 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 "),
     - g5 g/ x4 {0 M) J
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     7 W9 W4 a& H" b' U# G
159. }
     
160. }
     ( D# v' r' j: F8 Z
161. 
       |- L! ?3 q% E( d7 g2 Y
162. Sleep(10);
     & A  B2 `' o  _0 C( A# K
163. 
     
164. char   buffer[10240];
     * \8 S) a  E: ]/ P% g' [" D
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ; ~! S; Y4 N! I2 Z: _
166. if   (rlen   <=   0)   continue;
     2 R: P9 p) k: C  x6 f% V( V8 r# v( A5 T  c
167. closesocket(s);
     1 C6 K+ R4 ^6 W* c* `
168. 
     ; ]! M1 q: b7 s
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     , P/ O  ^3 G( x$ V9 \4 _
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);
     9 L) b. E0 R$ Z, `1 P% Z! R1 W
175. if   (result.Find(m_name)   > =   0)   {
     3 [- H/ L' U4 U$ e/ A
176. for   (int   pos   =   0;;)   {
     1 U' ~" \0 B. a; M/ S
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     * v: ~2 W( g0 K! a6 q- v5 F
178. if   (line.IsEmpty())   return   false;
     6 c+ P/ q" i8 Z9 I
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     . Y7 `  d) [/ v2 h' o% ]
181. if   (name   ==   _T( "LOCATION: "))   {
       R2 n% {* |  F' F
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     : N# S/ ]! c8 {4 ~! m3 K0 Q
185. return   GetDescription();
     1 U* I' M' z6 @# m4 T4 ]. O( d" Q2 K
186. }
     
187. }
     
188. }
       ^, o4 B* T$ e; K" t
189. }
     # Z( A/ e( ]* h8 h) ~! B
190. }
     
191. closesocket(s);
     
192. 
       K6 `. h3 {2 M. x) U5 Y
193. return   false;
     
194. }
     

复制代码

天云上人

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

& A9 s4 T5 J$ p$ [$ }. q
///////////////////////////////////////////
$ C9 d2 W( k0 s6 U//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
8 J. r! _$ j: ^& f, n% @! I
. `7 C: J% |, \. _1 D
& f/ s  }* x" a5 O# g#pragma once
#include <exception>

2 G5 C# @. r2 l  n( e0 w9 O* i, _
0 o) m9 |0 T, I6 {* d( Q  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};

. H( d# W& b$ t, `8 u
/ y# Y: _6 E+ denum UPNP_IMPLEMENTATION{
' p) t( [" M7 l; P6 t9 C        UPNP_IMPL_WINDOWSERVICE = 0,
8 x& h" R; e/ y% T! G        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

  |0 |2 P* M9 s

& _4 }& y2 J/ i! m$ l, m
& F8 i8 [& ~& v4 f9 I. f( Wclass CUPnPImpl
0 C: w' Z4 T* ]" G1 J# [% @{
public:
5 z2 w5 S7 B& q: m1 [+ r- u        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };
$ B1 y1 [) G. S

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
2 f& F! w5 Z5 X+ z$ j. }: E        virtual bool        CheckAndRefresh() = 0;
& L( t, w# \6 p5 b) I        virtual void        StopAsyncFind() = 0;
& Z4 @8 [4 E9 M0 [% r" Y        virtual void        DeletePorts() = 0;
# [) }+ S9 d/ D1 r3 f% `        virtual bool        IsReady() = 0;
6 q$ [1 J" g' r        virtual int                GetImplementationID() = 0;
       
) e3 M( a1 w; Z4 u$ H/ }- }        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

8 D: w) K9 f; t
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
" k0 @8 s" n7 ]. \0 w& K        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
; d3 n  [, ?- D9 \
5 H* v8 J6 S) G( V) K
// Implementation
1 h# E3 K. G  G/ pprotected:
! ~: C3 o( {5 k        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
' K( v3 X3 J' B+ i8 T0 i6 X        uint16                                m_nTCPPort;
8 P  @) e6 `: o) ]3 k& b        uint16                                m_nTCPWebPort;
: s* s# L+ R4 e+ t0 B) V( ~        bool                                m_bCheckAndRefresh;

: m( H, P5 z/ c
; m, w3 ~& j$ K$ z; vprivate:
        HWND        m_hResultMessageWindow;
+ J7 Q/ @# J( l8 F4 ~6 I' @$ B        UINT        m_nResultMessageID;

$ V3 ~# o: h2 O. ?* R
2 b8 r9 j2 m: |. Z0 z- C7 b};


// Dummy Implementation to be used when no other implementation is available
0 J3 F/ w; I% b9 N/ vclass CUPnPImplNone: public CUPnPImpl
4 g- Q1 o3 U7 [' |{
public:
% Q. [  d$ d, M4 O7 @9 Z' G        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
& H% m$ J+ R+ K$ ^) S( |. Y        virtual void        StopAsyncFind()                                                                                        { }
* n" I2 h4 r+ v0 `        virtual void        DeletePorts()                                                                                        { }
3 j; |0 E( |+ _. o- t/ H; g* B        virtual bool        IsReady()                                                                                                { return false; }
6 M8 O9 ?" ^2 }. Y% C# A; A" h        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
1 M/ e/ j5 N0 N  u% z: w6 V& `% U};
- B; E- g9 G; w0 [. t

$ @) _8 ~$ k* w' X/////////////////////////////////////
6 e' U: ?1 R1 G3 q$ n! ]//下面是使用windows操作系统自带的UPNP功能的子类
9 _( }9 N1 w8 c; _
4 Z& m0 P* N* M* s* D3 S) y
4 z' S: d  W0 K# k#pragma once
2 \* u& j9 l6 k- y2 g: [9 t7 f( C#pragma warning( disable: 4355 )

. t; \, v& T9 B- v- w' n
6 q3 @% D' I1 I5 v8 q8 ?! `  h+ ~#include "UPnPImpl.h"
' E4 j% Q- Q4 _  l#include <upnp.h>
, A. s+ C* L7 A5 C) l+ O, R#include <iphlpapi.h>
# q" I9 F5 i1 K: i( f0 D6 G#include <comdef.h>
#include <winsvc.h>

9 ]% C5 I5 d5 H3 l3 H
8 b2 _7 A$ _' T+ [3 \# Y! O#include <vector>
#include <exception>
) d: H4 ]; ^+ ]- c$ l#include <functional>




9 R( z! c$ \% j4 N! s9 ztypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
/ R- D) |- S5 n- G/ Ttypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
$ I. B# ^- W8 {typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; u) m0 K7 t5 U( q

typedef DWORD (WINAPI* TGetBestInterface) (
9 N$ w2 \! f9 E# F  IPAddr dwDestAddr,
- B8 |, X) |  h' o( |  PDWORD pdwBestIfIndex
);
. h7 S2 Q# M9 s2 i3 u" |

+ e+ w7 ^! N3 E3 C. _typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

0 S/ l' w6 V6 Y! I" j. B/ Z
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


; h9 s5 K9 F) sCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
( W7 |$ L4 C3 w8 m, Z& g" d% h

1 e, K( i( u. a3 `5 I' ~class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
  a2 ^- Q1 T* H. [. Y5 O0 Ipublic:
        virtual ~CUPnPImplWinServ();
# A4 D, N! u1 @- W- ~        CUPnPImplWinServ();
# S  G" k$ q. o" O, T

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
2 g& p  }/ a' u        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

* F* b0 r% f0 B* ?/ j# F* \
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
. B& N/ o3 y  ?. i  U. m8 M        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
  l& F) N. ^  Y! Z# M; {1 A        virtual bool        CheckAndRefresh()                                                                                { return false; };
& m% x7 ]1 G- s0 s
' B. @/ r# @( D9 w
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
: f0 \7 g6 g, \' {        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
8 o- i9 N. u3 Q8 N! U5 \+ O% l        void        Init();


3 T' D. i+ V6 S        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
7 D+ d4 q" ?1 C+ I9 q- u                        m_bAsyncFindRunning = false;
/ f! O, Z) ?4 i5 K! h4 C                }
                MSG msg;
* b5 x( m6 Y) Z                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" Z0 F7 t- `1 W9 s4 e' [# p5 ?                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }

1 S" Q7 m9 Z& |( e: [4 l5 s- G
# x" }6 D( _2 O3 w        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
" H1 I2 ?" {3 e7 G        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
6 ?2 x) L; D, }0 C+ ~0 K& B- a7 F        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
: Q7 q1 f: h+ M8 T) y        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
0 {9 w" G; }- m/ h4 O: }0 t$ r* S
; G- @. t0 [5 j7 p1 S
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
9 x2 v$ Q9 D+ h- b/ Y- z" S
% w& G1 i) P: Q+ k
( E5 y+ z" G5 |4 g4 j, A# U  Y8 w        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
8 d9 G$ R2 [7 L. b1 l7 q9 ~        {
5 }8 J/ w. o/ x8 G& X% L                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
1 L' K+ V: Z6 u3 M- ^& k                        CComBSTR deviceName;
* ]" _/ P, ^, o2 x$ A" T8 ?                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

- Q+ T6 p' T; B# y
) _9 f7 P- p" C/ v4 R; n& o                        if ( FAILED( hr ) )
1 @) q& G( F, ~# i4 o4 b. V6 l7 T: P                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
% t+ H, N6 ?, H( J' f3 I: X                CComBSTR m_udn;
( t1 W5 W+ W) t' d* T        };
" J& ~) O7 \+ i' R  w' k" W       
) r) i% _+ d( `8 F; ^8 v        void        ProcessAsyncFind(CComBSTR bsSearchType);
6 z  d1 t1 S0 e, _' [  v        HRESULT        GetDeviceServices(DevicePointer pDevice);
, _) M% m5 D/ h: I5 @        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
  p- o1 i% B' r& h3 k/ F        void        DeleteExistingPortMappings(ServicePointer pService);
+ F( f  U, ?( F# M1 i3 H6 O7 u% H        void        CreatePortMappings(ServicePointer pService);
! y4 Z/ J( }! n" D* k% F% Z+ F        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
6 X; b+ ^) I* o        void        StopUPnPService();
) F9 }/ p4 ^. Y: w5 s3 v

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
# D9 X2 d: P3 O2 o$ J- e5 f        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
6 M  M- I+ f2 r) Q/ z3 _, F

' X& L* x' ]( E8 e1 Z1 D+ N$ R% D// Private members
: n2 @# z) F; y. q9 Aprivate:
5 A& r$ t# @2 h: C  Q        DWORD        m_tLastEvent;        // When the last event was received?
  i1 k( B: ?" v9 G/ \2 g        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
* l& x3 d# r( s( g        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
" N: k$ C3 n% P* w! @5 ?: L! {/ m# l        ServiceCallback                        m_pServiceCallback;

# E, R" R* d- k' r) |( V" R
/ D2 F+ T$ W) Q: p+ ^: N        LONG        m_nAsyncFindHandle;
, m8 O6 @/ P. }8 v9 \! p( c: a        bool        m_bCOM;
) a6 c2 f4 M2 `0 M2 B- |9 b  Q! D        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
) y2 N# ]7 _0 I) ?2 L/ K        bool        m_bADSL;                // Is the device ADSL?
0 N+ _  V: Y1 t2 x        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
( G# _  G* q  g- G! }- O$ X& S7 n        bool        m_bInited;
/ G# P' ^" G$ e0 \        bool        m_bAsyncFindRunning;
/ e. x' I  Q  Y( F* e8 x* @        HMODULE m_hADVAPI32_DLL;
5 y# A/ |* `" Q5 q* x* R        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
4 ~+ Z+ l  F$ Y' V# D( S" s        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

8 S3 B* W) G% P
+ q0 s1 F) N) U$ b: B};

* t6 s* c6 V, c& K  L/ P5 x
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
- |$ q* |, ^: O{
$ l3 y) t1 D- ipublic:
: {- g0 F. v7 y, Y, T        CDeviceFinderCallback(CUPnPImplWinServ& instance)
3 D5 s& k- J+ O8 j$ V, c                : m_instance( instance )
% |6 ~2 O" T  G. P3 J        { m_lRefCount = 0; }

& {- ^9 n9 Y+ T& o, c8 c$ m
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 l: R, K( R/ ]# }0 i9 E" X   STDMETHODIMP_(ULONG) AddRef();
% [' c7 ]! ^5 i* ]8 h7 n   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
1 j8 Y' @1 \9 M: Y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
8 I$ x, Y- r3 D9 U* x
/ `4 Q2 v( O  _- m
private:
* b9 q" j# b' Z5 G# t  \2 L5 j        CUPnPImplWinServ& m_instance;
% f2 Z$ W& o" ^& C* U1 W6 X5 M        LONG m_lRefCount;
) {( C7 ~5 R. U' i};

7 X: K1 `: u: \) `' J. T
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
7 i- E, e* I9 ^0 l  y5 {7 e! x. G{
public:
# ?  s0 ]0 J# h8 Z& V& p) Y( `        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
) y7 j3 W4 |6 v3 w   STDMETHODIMP_(ULONG) AddRef();
! `; f/ U7 T1 P+ V! p) m& a2 n   STDMETHODIMP_(ULONG) Release();

$ f& w" M$ e+ ^
// implementation
1 K4 I0 Q0 y, t) }% n0 Z$ y* Dprivate:
- |4 C$ H" i* k: o        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
% n$ P7 E$ V; Y7 |7 I% ~5 x

6 s2 D. L7 b" V. b  r0 Tprivate:
# j8 B7 l1 e0 h9 o        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
. W6 z5 b2 l5 V% e) N  e};
3 ~7 |, {7 E/ x* F
! V# H" w0 y. V" o" k: q* Q4 u4 X
/////////////////////////////////////////////////
" w, x& O3 Q% {# p- H( L
6 t" P( [$ t: v! |
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
0 \4 r; ^3 o0 b8 i% G9 j0 XCUPnPImpl::DeletePorts删除端口映射.