UPnP

天云上人

/*uPnP.h*/

1. 
   : Q0 p6 _$ f6 ^7 X8 j1 E
2. #ifndef   MYUPNP_H_
   + J  D* B+ y+ n8 A1 |
3. 
   3 H+ f$ y6 }6 _  `9 _, E
4. #pragma   once
   - V5 E3 [. a$ v5 I0 B4 Y
5. 
   2 \, l' X) U/ A5 x
6. typedef   unsigned   long   ulong;
   & a0 v+ e7 f) n; \! Q! P* T0 @
7. 
   # S6 G" _% l- u3 ^4 u- j) g
8. class   MyUPnP
   ' w  Z. O# l/ i
9. {
     \: v( z) ^+ o& w( ?% x
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    6 Q0 A& A  e' }6 ]
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    6 w( t& D# k$ ^' G1 U' O/ r. i% l+ T
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;
    
18. 
    * ~1 M* D/ B) S" D1 {+ F
19. typedef   enum{
    # ], t, a3 ^' X/ z: _6 U- @; X; a
20. UNAT_TCP, //   TCP   Protocol
    9 X7 w/ c0 }" G, h
21. UNAT_UDP //   UDP   Protocol
    6 u! x7 A9 U0 M9 \* G! ~
22. }   UPNPNAT_PROTOCOL;
    3 q( {2 H2 O( S4 M# ]0 d
23. 
    ) {6 Q2 c" g5 h* x0 n
24. typedef   struct{
    : A7 v/ G1 }+ w5 y, ~0 n
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
    " y1 q. {- N! r* k: [
29. }   UPNPNAT_MAPPING;
    
30. 
    * W* W- t* I5 y
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    # h, V6 z9 k9 D" r) K7 ]) O0 T
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " `; \8 J5 p% Z; }$ v0 o1 r( @( I
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    9 l; ?+ a( u& J
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    9 ]3 _9 ^2 k6 F! H
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    ! Q) ~0 J- s8 b2 N! d, i  a
46. 
    * q( X5 y* T3 y, T! C
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    $ F  v( C9 M# E7 ]: o
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    # b; D, y) Y3 r* t# i* k
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    9 d% R) E4 h( r9 A
52. 
    $ Q. _' |2 B) o# d1 r* \
53. bool Search(int   version=1);
    - P2 q% {4 \4 i& l
54. bool GetDescription();
    6 Y# m9 n5 t. d" |* k' N" d# v4 H
55. CString GetProperty(const   CString&   name,   CString&   response);
    ' M$ z" @+ e- h/ V$ X) i
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    : |( n, F4 ]' l: p3 d. `
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    4 m# c( p# C2 W8 d
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    . i+ K" a8 n) J) @. \
66. CString m_modelname;
    1 a6 c) x, i9 G; {8 _9 h
67. int m_version;
      m' [" I) k: s4 F, d$ o
68. 
    0 B3 e' B! s  _' O6 N4 H
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ! ^% K7 f0 ?8 A8 V& X9 z! i$ f
71. 
    # d+ L7 E7 z7 |( H! J0 c; {
72. CString m_slocalIP;
    - _9 i6 Q# A. s4 r5 p
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    & A3 [6 u% h& ]( j, K5 j
75. 
    4 I7 E6 G2 x0 }+ Y( z, M0 l" C
76. bool isSearched;
    8 s) [- T8 \) v* H! w! I
77. };
    9 X$ h4 ~* T2 s/ {( e
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   % T' D8 `, R7 f* ~& H2 E
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   - c! e. f) N1 Y
5. 
   : Y9 \2 n, @4 U# W% r# N' C
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   / O  _7 E9 l9 w  C6 t. C+ o6 z
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   1 j5 V" y, u5 x* a; U
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    # q% |& C+ Q3 T; x. ?! \
15. 
    % T) D. i# f0 C" u; C1 y& b) S) j
16. const   CString   getString(int   i)
    9 X6 Q$ |& o- g  q
17. {
    
18. CString   s;
    . \' g7 \# H4 {
19. 
    * [& @8 ~% d. G  f4 z1 L! e
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    - [& s1 E$ D" S" M* L9 @' W
24. 
    ( c& @5 x1 Z! y  J
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    & t% q+ r9 h- o1 q* _" z5 e0 s5 n
28. }
    
29. 
    2 j; [6 ?; X* K
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ) O4 p- e2 @" A& T9 J: k4 m
31. {
    6 D( x+ z# O+ n+ U! O: l2 y3 L
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    : j' y* o4 d8 `2 ]) |
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ( h3 Z  z7 e3 v! y, V
36. {
    
37. char   buffer[10240];
    
38. 
    4 P" S! ~* a5 r2 c* k' o8 Y0 r8 _
39. const   CStringA   sa(request);
    * t" I9 d1 C1 k. J' s
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    3 ]% B# D% @2 l& c+ {; x$ F7 ?& D  J
44. struct   sockaddr_in   sockaddr;
    % q. R4 i' }0 Q* Z! Z7 v
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - ^% l5 t; m! Y& Y0 m" Y7 e
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    : `" W1 G  g3 A* b: }$ l
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    . m) R4 k' J& _% P$ I+ X$ ?
50. u_long   lv   =   1;
    ( ]8 v& a4 x( ^) L: }) P+ S0 e
51. ioctlsocket(s,   FIONBIO,   &lv);
    " I$ j7 M. ~' \* Q, i5 I! `) C
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ) d. C" Q- X$ R6 s
53. Sleep(20);
    5 v5 o* ?2 ?2 ]7 y  o& ^# e  {
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    8 k/ _7 Z: B6 k* y2 A
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    $ Q8 D1 x1 G3 h/ U2 S4 u
59. if   (!rlen)   return   false;
      k. y& Y4 E) n7 z5 q* }' B7 j
60. 
    ( D& o' s/ ?' p. m  {! a& I
61. response   =   CString(CStringA(buffer,   rlen));
    ) T: R9 f9 S2 Y9 w
62. 
    - P' s- k; F9 Q# i8 I( N8 u* q
63. return   true;
    
64. }
    
65. 
    . D2 z) c. S$ d( w+ a
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * B. M! m' u) |6 j
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ' ^3 M* a8 @4 S' f' K! x
72. strcpy(buffer,   (const   char*)sa);
    : Q! h$ n2 P* b0 M1 b5 U* x9 }% F
73. 
    
74. struct   sockaddr_in   sockaddr;
    5 f, `2 Y- X, Q/ W) b8 N9 X2 }0 M; m
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    * S9 i; x7 T3 h/ M
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 e+ y; X4 ?: @1 h& t
79. 
    / K4 V8 y( w& S" t8 S+ T& B" T. q
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    7 n# J: l) `) a4 R& ], H6 [
81. }
    5 K! i0 }* {6 C' E
82. 
    5 ]/ {: o& x7 @4 ~! O" u+ \: T
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    $ u: I8 x5 C5 g2 @* R4 j. ]
85. int   pos   =   0;
    * q' M5 i/ D7 w2 f
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ! F+ S# _) I2 a
88. 
    8 m  a0 U% ^( E. u0 |3 z
89. result   =   response;
    4 ?* X3 c/ S" W
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    - W) _1 W6 J5 v5 o; y' x
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    - ~/ a# J; `: q. ?0 Z" F: G5 O
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ) S% j( T+ h8 L1 {! n* m% h
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # ]; k- m) P7 s+ O( q+ l
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     3 n. E- \$ R& X4 H. }; I
106. if   (posStart <0)   return   CString();
     : X+ L( V* k% U5 f* }: Q  G
107. 
       K6 G$ B, ]. O
108. int   posEnd   =   all.Find(endTag,   posStart);
     5 c5 h( o- s: N1 h
109. if   (posStart> =posEnd)   return   CString();
     7 I, L* i6 L2 }' {2 a7 }; R3 p# ^
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     ' i* r7 I. A# [, t! D, j# P
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     5 }( A- Q5 v# t7 D" c
116. {
     
117. m_uLocalIP   =   0;
     # g# G* V: r- G/ a
118. isSearched   =   false;
     
119. }
     
120. 
     1 G. m6 ?: g$ |  `
121. MyUPnP::~MyUPnP()
       ?, E. j. W8 T. D$ U; b& h' L
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     : X& N- k/ u' S0 j
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     & F) n, c5 ^9 ]2 M8 E* J
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     2 Q3 j2 W2 d+ f6 J- c( Z
132. 
     & H( z. Z: a* d
133. 
     + l7 a' X  }1 k/ `" z" Z1 P
134. bool   MyUPnP::InternalSearch(int   version)
       H1 G3 y- R* U
135. {
     - l  [, M! t8 l9 g9 ?
136. if(version <=0)version   =   1;
     / w! U9 ?1 \' n9 m
137. m_version   =   version;
     
138. 
     3 w4 q5 g" f. r' D' u6 k
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     / K# c9 M9 }: G
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     . h* Y+ n/ x& D
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     , c+ |* Y8 Y4 S3 n# q
149. 
     % G7 S4 {/ S1 `* L9 H( ?- [# B
150. int   rlen   =   0;
     2 g& n- ?& E5 N6 y( V
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     ) l- m' d( n0 X) H* I9 K% y
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;
     
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 "),
     6 |9 Y9 y& g/ B& e& p
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     8 `9 v) m" D7 S& v' l8 M
159. }
     8 O0 p  M1 c4 c, j
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     * I1 M9 R- n& {- \& J/ u* k
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     , l: S8 C% ^; w! C  T
166. if   (rlen   <=   0)   continue;
     9 E1 h+ I/ [0 p9 F  o$ D: x
167. closesocket(s);
     
168. 
     : l% R$ m9 N, V  q/ k, Y
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + U5 g1 x' [" X& k! ^* |" n' u
170. CString   result;
     " I$ M2 @' n. l# K& D8 H# v" \
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     % S0 i, d, [7 Y6 h5 p' c: I8 \
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ) l" p3 U1 }3 X2 p4 [
175. if   (result.Find(m_name)   > =   0)   {
     - q3 `5 w/ k3 c2 l, o- v' z
176. for   (int   pos   =   0;;)   {
     & h9 l% I+ _/ h9 p; P3 C
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     - Q% Z6 T) c; @( G5 K2 k2 J) u% @  q
179. CString   name   =   line.Mid(0,   9);
     + {0 `, B/ ~2 I+ W8 ?: B" S
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     1 q3 ^  v9 p' g( o" C
183. m_description   =   line;
     
184. m_description.Trim();
     5 N* C$ A+ V9 z$ U* x3 j1 h
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     4 c: U# U9 e$ D8 s( F" O9 j1 k( X0 ]% ~
189. }
     . Y  K: S4 o, g& [4 [
190. }
     
191. closesocket(s);
     8 z/ e( L; P; c2 w8 S
192. 
     
193. return   false;
     
194. }
     9 b! g6 {2 g9 {; Y

复制代码

天云上人

以下有关upnp的接口来自emule，
9 H* P7 A- b1 r8 [  i

( p6 g! k- g& f4 p* z0 t///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
( a) D$ r8 y3 s  q4 f

6 E% ~, N, T6 A# m/ w- B7 y#pragma once
4 D" k) M( F' s: s1 w#include <exception>


+ Q) a5 U5 Q7 u" r% y' H  enum TRISTATE{
7 c1 ?3 h& b1 z( n& j        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
. @6 S! [( q8 X6 i1 ]};

8 [$ Z' P1 [1 K+ j
$ j# s. T2 h1 G4 b# |enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
+ T2 n6 ~: l$ q. V! U        UPNP_IMPL_MINIUPNPLIB,
1 f8 z! ?2 k" \; ^% w        UPNP_IMPL_NONE /*last*/
};
4 w# o1 a3 A4 |$ ~
# Y# w7 S6 k1 t7 T2 h# q6 l2 H/ h


/ h7 b, z  q5 D5 a) b5 n4 ^. Qclass CUPnPImpl
# F6 o+ s8 g- j- g/ l* V9 ]. ]{
+ f( g% o' Y& G$ v* E3 }- d& G9 R+ ~public:
        CUPnPImpl();
; P9 b$ ?/ M( L+ B8 \# y* L" |        virtual ~CUPnPImpl();
9 |0 \& w) V2 B! Q2 h- f& K0 B        struct UPnPError : std::exception {};
- n$ Z! ~& u! p$ C7 L        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
, U; A+ n/ y. H2 V; o( Q$ K# K        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
: \7 F: \7 ^0 R% r/ F* E. L# s3 ?4 S        virtual void        StopAsyncFind() = 0;
* k' B: Z" \0 n5 ^$ O% m        virtual void        DeletePorts() = 0;
" [$ O( V; T) l* K6 q7 |) a        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

. ~5 I: C3 o/ B: c7 F" }
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
5 a( [3 s9 j. {9 d$ m        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

: r9 v5 ?7 i/ j9 X5 f( k
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
" @9 O4 I% c, i- {8 ^% ~* ]( T        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
8 ^/ G" f. R9 Y- P$ y& A
) h, K# F) l$ I; a
. F3 m% [1 T+ t9 `private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
- O, t6 i$ j$ l4 e0 \- K
: w+ t  K9 x$ L0 `) S: c
};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
0 Q. ]- d1 Q9 z$ h+ cpublic:
& Q/ I- l/ M; S/ u% ~" }1 y        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
+ b( V$ D% T6 i8 v) }' o4 M" X        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
& P8 h/ q- S* d+ r        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
" j( L4 Q  x) l9 V- f( Z7 ]: ?

/////////////////////////////////////
% B' T) b; A( n4 s9 W6 ^4 X- e//下面是使用windows操作系统自带的UPNP功能的子类
9 k/ l+ [4 ?1 `* H" |

6 o6 K2 n& S$ Y2 b) B9 U* y3 V#pragma once
#pragma warning( disable: 4355 )

0 y% C- R3 M0 b. L; b) [3 [! D' q# X( r
/ R: C% w- ^% k; z#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
# p5 L8 k8 h9 @#include <comdef.h>
6 }' A) R  O# @% g+ d& G4 k#include <winsvc.h>

, c7 T1 |9 a* x( ?# P
. L9 J" u% _% G3 C% J#include <vector>
& o8 o% y/ L: e& Y; O4 o#include <exception>
#include <functional>




typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
+ J) r) E( L+ Z0 Utypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
4 v7 T0 t# z5 i5 [$ p. Htypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
$ n6 |4 |% N1 a; x* stypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
, l! R( D+ D9 ]/ N/ xtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
# z1 m8 W  V9 \& f& N6 V; D  IPAddr dwDestAddr,
: o4 f7 \. ~  \, \0 ^0 `  PDWORD pdwBestIfIndex
);
% `8 q  o- U2 s# N& j- P# ^
2 N* Y9 m- ]8 O& J# h3 B! @
typedef DWORD (WINAPI* TGetIpAddrTable) (
, d) x! w0 C' h( m+ c  PMIB_IPADDRTABLE pIpAddrTable,
! l3 t" ^; B: w* S  @$ g# q5 A  PULONG pdwSize,
: y3 Q- P0 k$ o' x. v, V$ n$ z  BOOL bOrder
);
3 p6 T0 w8 N, |2 O+ o* H8 {* I6 j, g. h

typedef DWORD (WINAPI* TGetIfEntry) (
" H6 e9 u* t7 Y% ^, N  PMIB_IFROW pIfRow
, c4 b4 l# q! O- K);

$ _  A3 M, x! B$ U$ ?9 N
CString translateUPnPResult(HRESULT hr);
/ R& V2 s2 a0 ]# ]. @HRESULT UPnPMessage(HRESULT hr);
5 y; }: p/ C, p+ @( f. _

- [% ]3 Y/ Q5 t) L) ?: a! z( dclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
$ x* p1 A# }9 l% D5 ^3 [0 n: X        friend class CServiceCallback;
8 ^* X( Q8 A# e: O2 P// Construction
public:
6 W7 ?) d9 Q3 y+ h/ P( ]; A/ G        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
% ^' n  p; K" J
/ N5 H. \- F/ K4 }/ e5 L
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
/ T# ?; L% Z  B) R5 p* N        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
: D4 [0 n# m  S5 x" T        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


. ^( f! s5 ?2 S7 W        // 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
2 j' A0 A5 B" O3 c        virtual bool        CheckAndRefresh()                                                                                { return false; };
3 R% g3 [& d3 m0 ]2 m
8 H8 A% ^0 i$ q2 I% r  Q
protected:
7 ]: x0 B* a  ^0 L: w. c  C( B8 j        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
( i8 g9 A9 n. h( e0 @* `, {. c9 g        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

1 m" l5 _! C: W" Y& z
        inline bool IsAsyncFindRunning()
        {
) f1 z; Y* Z9 k" H7 {3 G                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
7 q) C* T5 x% F2 `" Z8 K# C7 B                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
  p8 |7 Q( w; j$ u& h                        m_bAsyncFindRunning = false;
                }
                MSG msg;
/ t% r# S8 [8 R2 ^% S6 _5 N7 |                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
& |7 o% u# k: L$ y4 Z3 D/ _                        DispatchMessage( &msg );
. F1 t! s5 z4 U# ?1 p) \3 z5 p                }
6 M7 R7 W3 J. ~                return m_bAsyncFindRunning;
        }
: X" a% A% i& `! U, {
; C. M& R7 _* l% M" `% o
        TRISTATE                        m_bUPnPDeviceConnected;
  T2 @0 q% v# n8 v# j1 p- Z, B9 n

- q/ ]. {' z6 p, ^3 G3 J- n! @. T// Implementation
0 c5 k; h2 C. _/ E4 o' b        // API functions
7 \2 W' \( S6 E( p7 Y        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 p2 I) ~" X! M4 i8 G) g        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
6 w) ^4 g+ `% ~/ j3 w: M3 m        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
; |: ?( c3 b, c1 o+ E, Q5 E9 _! @  X) h        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


: {$ |3 W% h/ u4 O6 V        TGetBestInterface                m_pfGetBestInterface;
/ F4 {% T- G4 c3 W3 P4 i) f        TGetIpAddrTable                        m_pfGetIpAddrTable;
# N/ J  t: p) B- C        TGetIfEntry                                m_pfGetIfEntry;
  i: P" |! w$ c- v% R1 C3 w/ d8 u

        static FinderPointer CreateFinderInstance();
& x; q' A8 r# K% B4 m, b        struct FindDevice : std::unary_function< DevicePointer, bool >
0 u2 k) q$ z- ~        {
6 a# Q/ S. M* y: K                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
3 C8 a" H. W  `. i8 j9 d4 H                result_type operator()(argument_type device) const
                {
0 _) U4 k# e$ n                        CComBSTR deviceName;
0 F* i% x/ B' F2 w/ [0 T3 n                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
8 g2 r/ ]6 e/ T' _, \

) N) S6 v$ x0 [6 \5 V                        if ( FAILED( hr ) )
" P; p6 a' k) C0 f' m                                return UPnPMessage( hr ), false;
0 \5 [% V, ~# G) O% ]6 f
  X4 Y# Q( x  m! f" I
2 n: p# [3 o5 F& A) m/ V                        return wcscmp( deviceName.m_str, m_udn ) == 0;
7 k' i4 \, |$ F3 V+ P. A                }
                CComBSTR m_udn;
. F* s- E3 U4 A# R6 k# y        };
& v4 [) K- u' F/ Z: V3 ?' R       
+ @! e- s  `- l0 }5 C; C        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ j) ~6 S- i, V2 t! j! |        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
: u8 p, P2 C* T! H9 X, V2 ~                LPCTSTR pszInArgString, CString& strResult);
+ d. ~* x# }4 b+ @        void        StopUPnPService();
; ^. L$ Y4 P" ?' X! z6 w
. }' _9 V' e. e" g* y# X
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
) G( ~$ O$ n2 P, C: D8 y. |        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
' S6 a9 p0 u9 }3 i8 H' ^* g        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
4 H* i/ ]! {& `% L1 K& @4 a        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
5 D3 r5 W2 v$ [% U9 `( u        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
4 p3 T" w" K0 c8 k9 A, k; B2 ~& j        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
! y* M) i! B) [  [! {/ e5 g& T        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
$ d" c( r' ]' V/ e' H) [) p- y2 L; R        bool        m_bCOM;
. Y' o1 }7 `2 t6 Q5 {9 N1 U5 I8 i6 O        bool        m_bPortIsFree;
. Q' p- B. W4 q' k# {8 s4 E) Y        CString m_sLocalIP;
        CString m_sExternalIP;
" Z8 K2 N2 _( ^) i6 _  A( P        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
! _0 S7 @6 T& `1 d: j( C        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
3 _  J- y1 w2 j. p8 s3 x+ r        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

4 J1 y2 N  ^- Q8 `; p
* G- f3 L) ?. y) ^7 F% g4 l6 w) H2 g};


// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
* \3 f% F4 J9 a$ v( n  o, N: e' }( E: ~{
8 \8 L: P5 A' A# i6 S/ Npublic:
( d! D6 f2 M4 O8 ?        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 U( o  V3 o& b' W6 C9 M   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


, O0 ]+ |9 {. [) b// implementation
1 B: q2 x8 k# R' E6 D  fprivate:
6 L; Q8 J* h; S; p        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
" N& ~7 V& N( x# H) ^8 _        HRESULT __stdcall SearchComplete(LONG nFindData);
' K+ x, R6 V  x. Z3 e( U! F; q1 M

& v: {$ |# r  {! oprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


0 R( I8 n: O$ G6 P// Service Callback
class CServiceCallback
- w8 h7 j2 ^% I" k9 x9 ^6 t5 Q4 c$ I        : public IUPnPServiceCallback
; i* W/ N4 z+ s+ n{
* N- R" `( g! D, Z/ y( s% r( K* npublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
2 q+ q  F! O' m+ T8 {   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
, J: b- Z; B& Y. [8 a% {   STDMETHODIMP_(ULONG) Release();


/ Q* b4 L8 f% L8 z, w// implementation
4 A* x' Q7 T- Y+ x( _private:
6 b6 Y, s5 l' R# B  e) Z        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
* G, N0 j& f! M- x        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


/////////////////////////////////////////////////

( I2 ]& j' O2 n7 G1 c8 q; S
* X# O$ M! `! n. L; g) ]$ }  `使用时只需要使用抽象类的接口。
6 w# x/ w2 ]  G& u. lCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
$ @; U7 Z' p, Q- q( a. S6 oCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.