UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   ) d: U/ l2 ^8 I
4. #pragma   once
   ; V# E: s; v, W0 G
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    ) ^, C/ {" [! E+ j0 Y4 @! ~
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    4 {* ^9 q" m/ W
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    6 m% U& k) g3 F& g  o  J- P5 ]
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    - \# G/ ^! \* Y
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    , Z( G" e5 r- ^* w5 \* Q9 E
17. }   UPNPNAT_RETURN;
    
18. 
    ( I* s+ f! A8 A& B3 S
19. typedef   enum{
    7 O+ n, E- I! o% D. d5 ?' `  h
20. UNAT_TCP, //   TCP   Protocol
    * \2 e9 Q! Q* a0 [- X
21. UNAT_UDP //   UDP   Protocol
    * P( _- b. c4 w: {/ G& t
22. }   UPNPNAT_PROTOCOL;
    * U8 J: d  z& o
23. 
    
24. typedef   struct{
    ) M" r' {' P0 l/ ~9 I  o( f3 N/ i; r
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    + x; o0 E3 O- E/ _* j9 l
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    3 @; u5 X, l5 c  e. f8 v; C, H
30. 
    % O) l/ R% [2 r9 N' v3 ?
31. MyUPnP();
    0 N  T: X) a8 Q1 z4 u6 v
32. ~MyUPnP();
    
33. 
    & r6 B: ]  X& A. S3 _
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    : n# o" g) r$ g
36. void   clearNATPortMapping();
    ; D' j8 M0 U' y- H0 s3 O! u
37. 
    
38. CString GetLastError();
    ( w, ?8 K$ [7 X9 K' ]# ]
39. CString GetLocalIPStr();
    ( j5 d# g' s3 Z2 r
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    + i" z( @; B$ D' Q& U: f, f
42. 
    
43. protected:
    
44. void InitLocalIP();
    1 I# Y/ V& k0 {; n6 `
45. void SetLastError(CString   error);
    4 q# A' d3 B! p1 m4 J
46. 
    1 b+ V# T+ h/ F; |- |' f& A9 O
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    ; q: s2 ~3 g( T2 P
49. bool   deletePortmap(int   eport,   const   CString&   type);
    0 `4 n* e; s% _, T+ z
50. 
    ! J9 k( X: V9 P2 B: ~
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    $ z* E; g. g6 Q! O
52. 
    3 \, }, |! R7 S. ~+ {/ K
53. bool Search(int   version=1);
    
54. bool GetDescription();
    ' p2 j2 f  s& j( g$ r
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ) _7 a+ q) R6 d
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ' ]# \$ Z* k* z6 x5 [
59. bool InternalSearch(int   version);
    & p, ]+ t9 v9 K, k+ v; ^+ f
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    9 z0 Y0 H9 s  T/ s' g
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    % }; B1 U& B, r6 A4 S
66. CString m_modelname;
    ! V. s. u4 S7 G5 x% U% ^
67. int m_version;
    
68. 
    
69. private:
    ' |( {5 q, S& ~6 E! {9 k0 O
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    7 N; [/ j7 ]0 {: a, s2 |( |
72. CString m_slocalIP;
    ) K  Y' t  ?4 E. ?$ w/ V
73. CString m_slastError;
    ; y5 j& ~& A1 t# V3 }) e
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    ' X5 e4 v, Y1 b: i' j' A) i5 h  Z
77. };
    " Q8 C) s' _# u# n
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * ~( y5 Y$ d+ _5 N
2. #include   "stdafx.h "
   
3. 
   ! z6 }8 H; [  G" u. m
4. #include   "upnp.h "
   " u/ c2 e' ?# _0 d3 o$ _3 `  o
5. 
   1 a( D( y1 {# X, X* S$ n
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   , a0 @/ |7 I8 K& M% Z
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
     R9 `! p: |: \( K5 \* k
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    3 O, d3 e. N8 Q
11. 
    ( R3 ?" d$ q& }! Z: Q
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ! j& v+ j: Y) ^
13. static   const   int UPNPPORT   =   1900;
    6 r' q# _# N( n6 F
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    + v8 F: A2 M5 ]7 f, k
16. const   CString   getString(int   i)
    
17. {
    ) |/ L7 E" v1 I1 @: E6 c
18. CString   s;
    ' b8 g8 D0 V' B3 u
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    9 G3 b  n8 \# D/ J; b9 I! a
22. return   s;
    # C8 s9 n5 a" n
23. }
    + d2 x* A0 F* ]1 g
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    - F( R$ `0 S+ e9 d7 ]( `* Z0 S1 h
26. {
    & a0 I9 x$ m( t# r' b0 o3 l+ S- n
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    ( z9 n  y# j# v8 g
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    # Q  K1 g' e0 a
33. }
    ! h  _# G  V  M8 E& n7 A
34. 
    % E* F: T6 t  b3 _
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    - `4 ^/ E/ U# o
36. {
    % H8 `, }/ S  [: H6 s6 x
37. char   buffer[10240];
    
38. 
    6 P: l7 G3 p/ d+ F8 D4 t) k
39. const   CStringA   sa(request);
    . W/ q, I; O+ |3 r9 J
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    7 b  ^& O2 Q3 m4 I4 i
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    5 C1 R! X, X3 |9 k' l4 u! h
44. struct   sockaddr_in   sockaddr;
    2 E; z7 n  J8 x- @- @( x% v
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    3 s! ~" Q" M- q
46. sockaddr.sin_family   =   AF_INET;
    # ]3 T$ S2 _/ g
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    " V1 Q9 G( t, J
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    3 R0 |; J; l+ r1 s* y9 q5 R9 A1 {
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & q; r2 M, D+ l* Z9 v& Q
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    # b. ?( Q( K4 L& y9 L, O
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    , z( I% B9 y, ^# A- r5 h
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ! z' {1 P, \& ~$ J, r0 E
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    # X! }6 g* V* W7 d$ x3 {
62. 
    4 f7 @& P% J! Y: v$ S
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    7 h- h2 K9 E5 |2 j! f# W" c
68. char   buffer[10240];
    ) B4 u2 f# Q/ _
69. 
    " B1 F4 Q$ T- k9 u8 W/ t
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    5 ^& }# h4 P: P: P& e9 X& w! c  g
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 `0 b7 u* t% K% S! g* Q
76. sockaddr.sin_family   =   AF_INET;
    , g$ u; u" x+ K( Z
77. sockaddr.sin_port   =   htons(port);
    % G0 d% Q. g8 X7 I2 [3 T( z
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    , T( A, _# I# i3 S
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    , Y. I' e9 W: ]* x8 Z7 m  n
82. 
    2 o+ E$ h* g8 ?
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    : T: t; z; U$ @& v
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    2 P- L7 U( G) z1 z* u+ p4 A
89. result   =   response;
    4 f: T/ ^; o2 ]' g# H
90. result.Delete(0,   pos);
    
91. 
    $ j4 ~( P; u+ Z5 S. D' d
92. pos   =   0;
    7 C! ~8 }; {, ~
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
      _" ~" M- S5 v! y3 U' I( V
98. 
    & r) R( W) Z& K* u+ Q4 t- }) {
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ; T, _* W7 D) p
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     & e% \: H5 @) C0 ]3 V) Y3 w
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     8 ]" F' p1 u1 j& }
106. if   (posStart <0)   return   CString();
     
107. 
     5 \; t" V7 C0 {: D% G% v  j
108. int   posEnd   =   all.Find(endTag,   posStart);
     / E8 C! @6 S  P
109. if   (posStart> =posEnd)   return   CString();
     * c) ~5 z" @. C. n$ g2 l1 H
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     + w4 z3 Z5 D' j. X& n2 |
114. MyUPnP::MyUPnP()
     ! X$ v2 R, v$ a" W4 O( l7 K
115. :   m_version(1)
     
116. {
     ) _% U  Q5 l7 V/ v
117. m_uLocalIP   =   0;
     * C+ D% K; e$ B) C+ ]% c  c
118. isSearched   =   false;
     
119. }
     : N8 E0 a. X  A0 d: _0 @9 x& L" a
120. 
     % s! }# X3 u8 p5 i* s- z; z. r
121. MyUPnP::~MyUPnP()
     ' _' \5 O/ J& z1 |" C( c- P1 P
122. {
     
123. UPNPNAT_MAPPING   search;
     8 H1 t, ?* Z& n. \) b
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     - h$ w5 J; |% M5 g: [2 w5 r# C1 n
125. while(pos){
     0 y; o+ }0 g; M% z- R
126. search   =   m_Mappings.GetNext(pos);
     " s& D. `; ?: H# z& }& \8 n. M
127. RemoveNATPortMapping(search,   false);
       H2 t5 @" c- _* y9 s% Z
128. }
     
129. 
     ( s3 i) I* f. J: ~/ ^& j
130. m_Mappings.RemoveAll();
     
131. }
     - E& s! k7 L9 H+ k1 G( A5 }
132. 
     + D, I5 ?0 n6 e0 _' G- P
133. 
     , E  d' G0 C) N) w4 q2 u6 D; k
134. bool   MyUPnP::InternalSearch(int   version)
     * F% s2 m; a) _$ g
135. {
     ( i0 S& u3 V. ]7 y3 ]
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     / @: U9 q& D3 n* B7 x( f* s
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     . A/ `3 Q9 j# q7 X6 _
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ( u7 |1 Q& V$ `( w& ]' Z: a
144. };
     : q0 ]2 ~3 u$ {( p2 Z) n
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ! _' W+ t9 [1 U8 \
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     % l* o6 |+ Z6 Y: w
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     3 c6 H* k4 v2 n; {! U3 j
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;
     3 i  P9 h; {3 N9 F) f
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);
     : `6 G& o! {# c+ B
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     0 o/ U% ?" ]  q9 b. _' _
159. }
     
160. }
     
161. 
     , m& ]' ]' P- E2 g) s) b
162. Sleep(10);
     
163. 
     0 w: k5 [: r3 j& A
164. char   buffer[10240];
     8 Z6 |4 O7 b& h+ D& v- j: b
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     / n- E! I6 T8 j
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     + c, Q) ~+ I+ ^. Y. ~
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;;)   {
       B8 O0 I* f9 E# u- h+ i+ {
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     & t  s5 d5 u( i+ X
179. CString   name   =   line.Mid(0,   9);
     # y: a+ g6 N, }- r8 O+ E
180. name.MakeUpper();
     - m) c8 }- y' D
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     / m# _0 b8 R0 M0 J0 a( o
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     - Q) J4 m) F* V# a* m- Y/ L) A3 l
187. }
     
188. }
     
189. }
     ! K, s7 M: q7 U* q  {% R$ T( F/ f
190. }
     # e* T2 Z% ~) e: ~$ j' x0 c3 Q/ E/ v
191. closesocket(s);
     0 x; @% d8 A# H1 N- t
192. 
     & P( V0 w, _0 v, Q
193. return   false;
     0 T# q, Z  c( S7 x/ Y6 i) @5 w
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
7 S; ~# `/ G7 W- q# K
, |- p& e  e  h9 c( S
///////////////////////////////////////////
/ H3 y1 B2 e& O& j* C1 U//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


% }( ~3 A! ]. x+ Y+ W#pragma once
#include <exception>
9 E( d, ~# v3 s* G# n; p

  enum TRISTATE{
$ U% p. X2 s& A0 f9 N: S9 m( `        TRIS_FALSE,
$ Z; d2 R6 _, S1 D( P        TRIS_UNKNOWN,
3 C6 T& u3 d/ t3 \, O        TRIS_TRUE
};

7 y8 K3 d3 }0 l0 J: P/ [
! q1 e7 l4 [, J0 U( r2 k  I& Ienum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
# P6 f. q/ o+ \        UPNP_IMPL_MINIUPNPLIB,
  `% b1 H) k* \" V, @7 {        UPNP_IMPL_NONE /*last*/
};

' i8 _2 x; N. K" g$ \
( o' P) {* K# g

7 a# G5 @" e) g0 m) z" Tclass CUPnPImpl
8 `* Y* |  O7 W# G7 O! {{
public:
        CUPnPImpl();
- G0 L/ W5 T* d- x- {        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
. x, j( k) ~$ v; @                UPNP_OK,
                UPNP_FAILED,
0 ]% D! P7 L5 Y; R$ h* g1 ?                UPNP_TIMEOUT
* A0 ?" O) N* @8 d        };


2 f9 ?' t% d( G' h        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
4 n* u; G1 Q7 n3 [1 C- \% q: C! o        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
* j' @# o6 p8 t9 N; h9 G6 i/ [        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
; L4 q/ h2 S/ C' T        virtual int                GetImplementationID() = 0;
+ ~1 N9 a" k; m* ?       
) b; T( b: c. {6 O" E7 C" X. {. c        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


- y5 h5 A$ e8 L        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
0 L4 S4 F; B5 l0 K; Y/ x1 h        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 V6 y6 G+ T7 l6 w) J        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
* O+ r( [, n' g# M) \

// Implementation
1 S0 o$ p  S9 T8 j9 Bprotected:
8 f. j& o- @1 \8 F; Q        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
9 P# ]7 C  H& n0 w        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
6 U/ E4 S1 C  p& {        UINT        m_nResultMessageID;
. J0 b. j2 M# G  U: N4 \

};
) A+ B# f, X. N
) h; p3 N* ~1 ~/ d2 S9 c1 s/ X
( A+ a, X7 X: c2 S// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
$ t; _+ m4 l2 c! C1 X{
8 x5 d2 H9 d# V6 f$ a6 Cpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 j; ?# `% L1 c        virtual bool        CheckAndRefresh()                                                                                { return false; }
! u0 T: `7 }5 O" R0 K2 x        virtual void        StopAsyncFind()                                                                                        { }
  u$ P0 x  ]2 d  [$ r2 `        virtual void        DeletePorts()                                                                                        { }
3 n: d3 T$ J( g. y        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
) w9 X( ]+ ?- P* o, W, V};

* ~; }  o% b8 O7 |, i3 x3 k$ `1 }
% L/ V% \+ `5 \) {7 o/////////////////////////////////////
" c. O1 l4 Y+ ?2 p. l//下面是使用windows操作系统自带的UPNP功能的子类
9 |! }1 r) @! c. w$ y' C
' ]* I8 d. c5 v4 q* o! o6 W4 i
#pragma once
#pragma warning( disable: 4355 )
# w! O+ |: p! O) t2 y" [" a8 T

#include "UPnPImpl.h"
#include <upnp.h>
4 s, I6 x4 `. t#include <iphlpapi.h>
! i9 Z% P  o. c; D- g#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
  M) Q; O9 f( _( R: ~5 o#include <functional>
2 H# t) K' b1 D# E. p! `

: y3 g& C& I9 [- ^. F* ^8 @0 h3 W

1 j8 Y2 Q# u8 n; r' Mtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
+ \$ g" |6 ^+ e7 _; Htypedef _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;
# `) ?8 ]3 |/ I! L- btypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

6 a1 g6 o4 n/ _; Z
typedef DWORD (WINAPI* TGetBestInterface) (
" N; m) ~* v6 D  Z8 U  IPAddr dwDestAddr,
' O" q5 u4 o# ~- k% g  PDWORD pdwBestIfIndex
);
0 v8 l2 K( f# x# N/ l$ Q

: |* i: |$ g4 `typedef DWORD (WINAPI* TGetIpAddrTable) (
' J/ W" {/ n3 S3 J2 O3 v4 Y  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
# M+ N7 f+ F" {, [! l% i  BOOL bOrder
);
$ y5 v7 N( p9 G
; N( g) V5 J+ b3 y
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
  K, z) @. F: M2 I/ K( H5 C" s
' J6 N$ U7 J' D$ B0 I, n7 x
CString translateUPnPResult(HRESULT hr);
  G# `; e* m0 E! }! yHRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
{
) M' P  Y2 z0 L2 n& b, [        friend class CDeviceFinderCallback;
$ L5 a9 V; ]6 Q; Y, |' `        friend class CServiceCallback;
9 M$ T& _/ E. c$ F# g// Construction
public:
$ v6 [4 A& {- Z& m        virtual ~CUPnPImplWinServ();
8 R" Z& `! a' C2 P' S& H, ?        CUPnPImplWinServ();
+ K' L: E5 J/ w4 N

8 T7 K  t  p  y% w4 J        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
1 L4 o! f6 R% |        virtual void        DeletePorts();
        virtual bool        IsReady();
6 t) c& U0 ~2 K( G2 D6 i        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
$ U1 j6 ^1 g) C2 f
3 \5 X( @1 f7 [$ h
0 e- ^8 E% q1 V+ |! j+ 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
' {" h8 [6 W8 p( T( ?( ?( S        virtual bool        CheckAndRefresh()                                                                                { return false; };
! [0 E8 |8 d' Y- R; A3 s
2 C" b! k9 r7 Q7 a* ^, {
protected:
- ~( [  @6 b7 Y4 G: K        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
6 B* S: {" E( y- y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
0 ^- a8 g- i! z$ V: G8 j        bool        OnSearchComplete();
        void        Init();

/ p4 T. [6 V% I9 Q. B! u
        inline bool IsAsyncFindRunning()
4 \- U: `* Y% Q5 s        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
& Y! o" k; d$ \9 M/ ^                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
' K, o9 X0 q6 K1 F% [: m6 E& o                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
9 U: P! {6 Y, e- g% \                }
5 F5 X, h6 w1 u  e1 a  p6 f                return m_bAsyncFindRunning;
        }
5 J2 s; a! I: R5 o) e$ y
) `. T; {) k; Q1 n  C
        TRISTATE                        m_bUPnPDeviceConnected;


) K2 E- w0 s. F3 u  D$ ?// Implementation
        // API functions
) i% T1 Z( E/ D: ]1 l) Q' i# g        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
' {( \9 i3 i8 J* c( i  z3 B) e        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);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


3 F& b' W4 j9 z* G5 D  r/ l        TGetBestInterface                m_pfGetBestInterface;
; ?' o' }2 W/ H3 B3 T* |8 ^        TGetIpAddrTable                        m_pfGetIpAddrTable;
9 c; H& U' R9 d) Z- g        TGetIfEntry                                m_pfGetIfEntry;
5 i, E0 I) \( Z. \+ v
% W5 B; |% n: w" ?2 C& W
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
7 t0 }2 ^8 m* C7 g3 |! x$ @                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

4 o# V. m! u4 m* \! x
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

+ E7 [! U+ d: \' _
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) q" {( d. K+ A, p; I- M& c, L* Q                }
! ]9 b* r: R* k& T" Q, r$ Q                CComBSTR m_udn;
7 k* ^1 z# y& _( r7 I5 k9 v1 m9 u# Q# N        };
       
. i4 h) D1 A4 I! }) Y# Y8 m8 u1 l; x        void        ProcessAsyncFind(CComBSTR bsSearchType);
; Y4 L2 T* _* N! |4 z% L3 z3 ^        HRESULT        GetDeviceServices(DevicePointer pDevice);
) |; _2 B8 h$ P9 g. V        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
- d% u8 x9 D- e2 k' J. [        void        DeleteExistingPortMappings(ServicePointer pService);
, C. l( t3 ?, B7 Z7 O        void        CreatePortMappings(ServicePointer pService);
( O0 w  w) ]( d7 ~& @, r& \7 W        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
5 m3 J0 W2 m% C+ U- @5 R$ v! A        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
6 ?2 O9 p% e* Y7 U# f        void        StopUPnPService();
" d1 N1 L8 }  [5 E& M7 A

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
9 g% A" b1 g4 d2 B* K; T- [        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);
% y$ A7 F2 @  V6 g  v# R1 V* L' ^        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
0 J/ Y$ o# `, f2 L( K

. r1 V2 E- o1 j8 N3 s2 v// Private members
/ G8 J! Z7 F8 u" h- C0 Rprivate:
6 j( M$ Y1 y4 h8 d& q+ J/ E+ H        DWORD        m_tLastEvent;        // When the last event was received?
- p2 |- I( |* d& D        std::vector< DevicePointer >  m_pDevices;
; s4 j5 e3 |* b/ W. f& R        std::vector< ServicePointer > m_pServices;
7 W8 m3 y" ^6 ~& e2 b) b        FinderPointer                        m_pDeviceFinder;
$ `5 Y% F& n! V3 q; ~) x- {        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
6 V& X' ~3 @7 i' o4 l5 L1 W/ o

0 T2 Z6 ~, x1 j        LONG        m_nAsyncFindHandle;
$ o! b. R& c( R* X, j5 ~) k. h+ j        bool        m_bCOM;
! e: w3 ~; Z, C2 X$ }, n9 q+ o        bool        m_bPortIsFree;
1 k3 h6 f: ]$ s        CString m_sLocalIP;
        CString m_sExternalIP;
9 u( D$ v" Y8 ]( m$ I4 Q, F4 m        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
  Q0 {0 ^+ l4 g5 a        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
# \2 L' |. p3 ~" ^% z% r        HMODULE        m_hIPHLPAPI_DLL;
' y0 k- ?( x2 |" p8 m        bool        m_bSecondTry;
; d+ W/ C7 G: U        bool        m_bServiceStartedByEmule;
4 C5 `5 N0 `; ^7 W6 _        bool        m_bDisableWANIPSetup;
( `) _0 s1 o: k6 Z        bool        m_bDisableWANPPPSetup;

/ q* u: Y8 T; b/ r  q. _+ O
};


// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
; a0 i  V% f1 P4 X  `& ~        CDeviceFinderCallback(CUPnPImplWinServ& instance)
* m; V; I0 C( {                : m_instance( instance )
        { m_lRefCount = 0; }
# _! D- B  r$ ?7 `
/ `2 Q* P8 I6 E- @* {
, ~! x6 d$ W( Q7 L8 \. |$ ]) i   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ o- [' G/ t, C# u/ ]   STDMETHODIMP_(ULONG) AddRef();
% g* e  W  u2 l# j- `( d7 ?; _   STDMETHODIMP_(ULONG) Release();
2 ]- J) y8 x' ]# J3 n7 _$ ^
' x: Q( R9 V! y2 _+ d9 W; x
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
6 g2 z/ e: o2 S: k' D6 z, @& P) h9 p        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

4 _/ m6 o( b% e- Q
% G8 A+ E! {7 Y8 C1 Z, bprivate:
        CUPnPImplWinServ& m_instance;
( D; O% O* W) ^& @) D+ m) [* d        LONG m_lRefCount;
};


! h( a) h, T0 W+ y8 p: W// Service Callback
class CServiceCallback
. D( Z' @  s8 p( ]8 ~( Y* u+ t' Y. c        : public IUPnPServiceCallback
, ^( q; y& j9 _8 J{
+ m+ G3 [% c8 ^  j5 ypublic:
        CServiceCallback(CUPnPImplWinServ& instance)
4 q: M# x5 `, O% C5 `                : m_instance( instance )
9 O/ v4 S6 S3 }        { m_lRefCount = 0; }
, h' F2 s- ?% Y/ u( `   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


) ]; L7 e6 V# |7 e+ D// implementation
" \8 A: v, {; C. x* R4 @9 {private:
' |% _9 v5 B/ @& G1 }        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
( ~+ Q) H/ ^' h6 y6 E        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
0 g7 _- Z- s! @5 C
7 o8 ^3 d0 O6 G! Y8 C
private:
% ^1 K+ V0 R9 y8 t/ t4 m# {        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
3 z8 c% p" w0 _7 V9 A' C7 T* a
9 ]& Q+ I+ N( \$ v
- B* v3 A2 E- @+ i# N* o7 u/////////////////////////////////////////////////


9 I9 T* Z. A* ~4 v2 _9 f7 v使用时只需要使用抽象类的接口。
4 P" ?1 X+ C! h  F$ o$ _CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
+ a- T' S/ |1 P% n2 _5 X7 k: t1 \CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.