UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   # k+ F* A) H& k* H) d& B
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   ; S' J% R2 S! O( P; [, A' F2 q& {8 w
8. class   MyUPnP
   
9. {
   
10. public:
    2 L2 E+ f8 A# ^, H1 g
11. typedef   enum{
    
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    9 E7 Z3 x. _% z. A% H& P+ e  E
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    - x8 v7 t' Q( I5 c8 Q# F
17. }   UPNPNAT_RETURN;
    5 S7 B' `1 N8 B4 H. f& c
18. 
    
19. typedef   enum{
    ( M+ w6 ?" F2 v: M
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
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
    . a% p# H8 O4 R7 p
29. }   UPNPNAT_MAPPING;
    , z5 [* |& x& H7 Y; D. r7 d
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    : I# j& y- z! d" q$ s; M
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    & t- l! ]2 L  ^3 R1 g5 F
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    5 G7 W0 p; ?" J7 L3 b2 l# ~$ K
42. 
    ( A4 z& H- }9 q& o
43. protected:
    
44. void InitLocalIP();
    3 X* t1 P( O  y/ T+ g1 P
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    ( X" {: R1 K. W/ }2 O
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    : \0 {' J* y. G/ Q# }: j7 i& D6 }6 ]
53. bool Search(int   version=1);
    2 g7 v; F5 U" T: d, z" e
54. bool GetDescription();
    - D  E# s/ H3 }. x$ M) s# }
55. CString GetProperty(const   CString&   name,   CString&   response);
    . a* c3 I$ A5 P  A: X0 E) `
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    / x, V5 \6 Q3 g/ y
57. 
    ! p$ M# `: ~7 k( E8 l) d
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    0 b) y$ O  |! |4 S5 O5 \$ S
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    3 T' A' Q& _( Y9 n  c6 w6 V
61. CString m_name;
    : m7 l; G- n$ J& M/ z8 P1 M
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    $ x0 U: h( f, y: w, @  D/ G1 ~( o
66. CString m_modelname;
    % V, j1 o- F; z0 `, H# H
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    0 h# R$ H( I  [; E- y3 X5 ]" v
72. CString m_slocalIP;
    9 w; _# Z% o* a* P. s  e: |" w4 f
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    " c4 T& |! \8 f8 b% k8 K
75. 
    
76. bool isSearched;
    3 }$ \8 ^! v# q! w
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ' _+ k- O0 n6 ~' K1 L
4. #include   "upnp.h "
   / n' E  ]8 I( g, W. x8 b
5. 
   ' o2 d3 R' y! {# c* F, S$ ?
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & k$ Q8 x# F* \' h1 s9 K4 E
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    7 [: I' v" s  I, j
13. static   const   int UPNPPORT   =   1900;
    0 \( `' ?" a6 U- h: [
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    3 {& ]' N% t' a+ M6 q/ [3 ~* l
17. {
    
18. CString   s;
    $ e5 M& y( C3 x
19. 
    / e2 g4 f* v% b  D8 Z8 X0 P
20. s.Format(_T( "%d "),   i);
    
21. 
    7 @; m+ |% U3 R5 s1 x6 Z& ?
22. return   s;
    : E% U7 X6 n  m$ E0 q3 w
23. }
    
24. 
    . S6 ?: U+ @5 v0 T) P! M2 S( W
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
      W# ]/ y. L3 {% D  i8 w/ G' j0 l
26. {
    # `! ~* E% y# I/ `) f# M- P
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ' B$ k$ F* g2 h) f" ^+ Y$ O' ?& _
29. 
    8 o' `  F1 m8 X' V( b9 Z: Y2 M
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ) s; m6 d5 h# }7 L3 V5 w
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ) E; k$ H( H% W1 R* H+ U
34. 
    9 b. O# L' |* N5 P& ~6 a( B
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    2 n# w2 O& f5 b( N) U9 n. x* H
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    4 u" {0 z; ~7 k9 a$ N! @
43. uint32   ip   =   inet_addr(CStringA(addr));
    0 Q4 b6 x' J% w. g5 X4 G
44. struct   sockaddr_in   sockaddr;
    7 t* O$ C# o1 {
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 W/ D; q5 t  P4 {- C/ a: J+ |
46. sockaddr.sin_family   =   AF_INET;
    5 g( T/ b. ^0 \; r  m
47. sockaddr.sin_port   =   htons(port);
    2 ]1 N$ A8 x! u  d
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + L3 t) |5 }- d* l
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    , {4 m6 y1 f9 ^
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ) {7 t2 ^  Q7 F
53. Sleep(20);
    6 ?# B! o2 `" e/ X
54. int   n   =   send(s,   buffer,   length,   0);
    # q% P& y- S9 ]( ]' s
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ' C- j" h* s& a  C# U; e5 h$ W
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    9 [* \1 X- x7 B9 d5 v* R7 [! z
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    ! u  F2 n! ^1 @$ i6 T" w+ g
62. 
      S9 c/ k8 i7 A. W+ _
63. return   true;
    % z9 g3 u1 Y( v6 s  \
64. }
    7 A9 H7 g. h3 D
65. 
    ; P3 }) D$ P8 |( p- C0 j- j
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & C) n! a% c4 {# C) u
67. {
    " e& d) f8 W7 o; m1 ~7 m
68. char   buffer[10240];
    $ V7 \0 C$ h, n9 L; }; Y
69. 
    # G& C% h& B# k2 E
70. const   CStringA   sa(request);
    : J- l9 L1 u' \) U. r' d3 n- P/ n
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    + a' M; ?  u' H' m
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    4 q+ Z2 [4 x" M' C6 w5 P& [5 w9 p
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    3 W0 D; b" m/ d+ u- l
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; I( \- n" r  s) S# K# D" |, r
81. }
    3 [$ `2 B$ w7 w/ l3 K0 T
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / ^- t0 B6 C8 Q  t, \
84. {
    9 O! v7 _; W: Z7 S
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    8 B' g" h1 w% V9 e, J7 l3 O
89. result   =   response;
    2 v3 P% M1 u: z, @. \, K4 ?
90. result.Delete(0,   pos);
    $ Q0 R1 P8 h9 R4 U
91. 
    4 U7 L' |0 z" C% X0 `& W
92. pos   =   0;
    
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 Z% {% a0 g* X" \( y
98. 
    ' x0 l; L, ]5 \+ x& `* \
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    * u( S4 ^2 E& K  R
100. {
     $ ]- V& x2 V8 Y: J1 ^$ S+ s
101. CString   startTag   =   ' < '   +   name   +   '> ';
     3 }: [5 u& n6 u1 B! _
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     & c+ X9 i, q& v% h* i
103. CString   property;
     / c8 U) E8 P2 w
104. 
     6 a% Y2 L1 [7 ~8 A2 v6 w- h* a
105. int   posStart   =   all.Find(startTag);
     . [% |$ j/ u2 n! k7 A: _
106. if   (posStart <0)   return   CString();
     & k3 x' `5 {7 {5 D5 @: W7 f# U
107. 
     3 [4 g0 i2 t& z. o2 r  O
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     0 ]* E4 q' E2 Q* @0 o
110. 
     ; H6 d- k4 g8 o" I# q: Q
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     # U5 L$ q1 a4 j+ j5 E/ d
113. 
     
114. MyUPnP::MyUPnP()
     2 x4 E, v" L5 H% j: A2 ~( v
115. :   m_version(1)
     , d/ |& a1 o: h) r
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     $ L9 x8 v! S' s* @
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     1 F- a/ J* W6 Y3 K6 w
122. {
     
123. UPNPNAT_MAPPING   search;
     $ N6 j9 }% f  g& ?8 Q
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     - q9 q" ~5 W) {7 Z1 W8 {3 _8 _
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     * f2 J$ X! N/ B( w
128. }
     / K% w' c. o$ ~  N
129. 
     ' }/ F, p) l, c& C! q5 I
130. m_Mappings.RemoveAll();
     * o8 \& j2 O0 K# i) b$ G
131. }
     
132. 
     
133. 
     # u9 t+ ?8 X, p3 K5 e
134. bool   MyUPnP::InternalSearch(int   version)
     . o3 C" m; Y1 k. Y# i- e4 {/ g
135. {
     
136. if(version <=0)version   =   1;
     3 [, ]$ f5 K& N6 b; b3 }
137. m_version   =   version;
     
138. 
     " H$ p, i/ A8 c
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     # p6 v. ?0 z- E3 W6 n
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     0 G9 A% {4 Y* H9 ]* u
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " `/ |0 ^$ k7 C# E; k
152. if   (!(i%100))   {
     - \" B" X# X2 i, k5 }* @+ W
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     5 B" f; O5 t8 M: B! Z; o! m5 R$ |/ u; T
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     2 H! q! v6 S4 q- W2 }
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
       c5 r) X. D5 Y& I
160. }
     ( {1 f9 n7 E4 D: x1 i0 M
161. 
     ; O3 b/ y" k, t$ o, D/ C0 ]
162. Sleep(10);
     
163. 
     2 w, C& ~2 f, d5 u% [
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     6 }, z8 G- F: o
167. closesocket(s);
       I- [- ?9 n. V% {. d& k
168. 
     . m, B  M/ C; o
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     3 F5 }, E' g5 J& N$ M
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ( U, d9 n! T& ?9 ~
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     6 y0 d# F' \8 l' A: V
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     ( k0 v. z$ m$ g. P
176. for   (int   pos   =   0;;)   {
     4 z* h+ d/ W3 I. o) \5 n% v
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     5 r$ G/ @9 l9 T( L
178. if   (line.IsEmpty())   return   false;
     9 t: b! E1 G" ^- s! Z3 H7 H
179. CString   name   =   line.Mid(0,   9);
     ' Z% H$ g% o6 y- Q
180. name.MakeUpper();
     4 |8 @1 \0 E( v6 @  R0 a1 u
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ) U/ W' q2 d) h0 `* g
186. }
     
187. }
     # _1 E. O+ l" r3 R. W9 K
188. }
       x3 z8 D! ?6 [; T9 J& j9 p/ [
189. }
     $ F3 s. F' W4 D6 h4 N1 K- L
190. }
     - r! h' ?1 i4 {( b
191. closesocket(s);
     3 S( \% M0 w# B( p
192. 
     9 R8 [0 U1 p2 n! A% \
193. return   false;
     % z7 v# w2 D! _; p: L
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
' J+ N2 v# ~: R4 }- `' B) R9 K
4 Z$ A, Q* K' L: K% e
///////////////////////////////////////////
4 o" S% z4 a+ i, K; ^' G; p/ z//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
$ K/ ~* |2 A* }/ m" `! u- W

#pragma once
#include <exception>

7 P4 ]& w2 }% w' S3 u8 u. F, k) X
* y; O* y" u: a2 M2 e5 ?  enum TRISTATE{
6 p. z4 J! a7 o" P, U        TRIS_FALSE,
        TRIS_UNKNOWN,
& u# r" l% o: f        TRIS_TRUE
};
+ M3 G- ^, N5 T

% l" w3 H7 ?- p; cenum UPNP_IMPLEMENTATION{
9 B- A1 H4 c& q% @. ^9 o        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
8 l% M; e, K! d' J};
6 W1 I' A6 E* j( ~8 d% D. U  @
6 |3 w% X  i, L  P- G) q0 {5 C& B0 O

9 H$ ~; h& G0 r. V0 J# ~
5 p* s9 @8 P8 I* g+ Yclass CUPnPImpl
( I' h: S8 h" O+ |- ?" Q( e{
public:
- w4 O* ^! B9 Q; Y        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
! X% m5 h( H+ {$ O' }# P/ v                UPNP_OK,
' l- O# q' o5 J                UPNP_FAILED,
                UPNP_TIMEOUT
        };

4 f" a2 D& x: V
6 `: \$ y1 h/ b1 v9 L        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
% ]9 G) ~. s: ]7 a5 n8 f        virtual bool        CheckAndRefresh() = 0;
( @3 X# r* ]( J        virtual void        StopAsyncFind() = 0;
$ ^+ k8 H1 @8 F. i+ I% p1 X0 A        virtual void        DeletePorts() = 0;
  S8 ^& y2 V; m7 X% P        virtual bool        IsReady() = 0;
+ @5 \: I- f3 I& M        virtual int                GetImplementationID() = 0;
       
8 M/ C9 |, l& f, D1 B; T# a        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
# Q* L" k& k# V; u& `2 {        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
- X; V3 q& s. J6 U( `3 R
, ]# \, ]( i% O& y! L. ^$ _
// Implementation
  ]9 L' x# b: D8 b; Z$ }; Hprotected:
, `% N: i7 J& v: {9 h  o        volatile TRISTATE        m_bUPnPPortsForwarded;
6 L$ T6 ]+ d+ P& J1 I        void                                SendResultMessage();
        uint16                                m_nUDPPort;
: j4 `% c* R. {' c$ @        uint16                                m_nTCPPort;
. g* J3 I  u2 D/ w& ]+ i        uint16                                m_nTCPWebPort;
. y0 T/ o/ r1 Y        bool                                m_bCheckAndRefresh;

2 V. v/ [0 T9 _0 r. G1 t4 P) D( y
' T. U. ]( L, K% o! ]- }% zprivate:
1 d2 C5 S7 z+ v: [5 O        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

+ `0 d: f" I+ V& [* N0 e# X6 U8 r
};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
9 b+ C5 N! A* x- N3 C8 |{
& e+ c( R' k5 D: C( epublic:
3 t( ~. ?) G2 L7 s/ ]' g, p7 k        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
: e* H- F- f% y; p' a) x6 r        virtual bool        CheckAndRefresh()                                                                                { return false; }
7 s2 J/ b2 w& u2 ?* v7 U" x        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
$ N$ H; n) \8 A( U5 M- g        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


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

1 q1 u8 s; }& I7 o
& \0 [! F0 Y7 Y  x/ m#pragma once
#pragma warning( disable: 4355 )

- y9 A, T1 w8 b" z" G
$ U/ F& N' D# t! Y1 b# j7 z#include "UPnPImpl.h"
- Y$ {) ?0 s$ I9 D- ~# z; ?. g5 J6 w#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
$ K# w/ ?: o2 I  ^/ ]#include <winsvc.h>
* M0 E" W3 }- O- B. D7 u: i
- ^+ {' {4 {! X* J, r1 @- J, u
#include <vector>
#include <exception>
% e# G( x" v7 w6 K* d#include <functional>
8 P3 Q1 H. y( A$ y7 }& \


" _  O6 t) M: @' r1 y
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
" T" c0 G- ]$ \! `5 b& btypedef _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;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
8 V/ k8 {, q* u+ O  z
% H4 o0 |9 @: ^  n' p- ^9 e& X8 X
typedef DWORD (WINAPI* TGetBestInterface) (
5 c$ o! x8 F7 t% ~  IPAddr dwDestAddr,
% O' @& h  f$ E# s& ~  PDWORD pdwBestIfIndex
$ X( _) O+ n; C);


, s8 v6 X3 z" J& qtypedef DWORD (WINAPI* TGetIpAddrTable) (
# g) `7 D0 f0 v8 a/ f+ N8 y  PMIB_IPADDRTABLE pIpAddrTable,
, v/ H1 E% e+ x: P8 ]  PULONG pdwSize,
  BOOL bOrder
; x: L  I' p8 p& a( Z. ]. X9 R( o);
* w$ A- X/ X; V! T: t. Y0 ^, i
) v+ W$ m8 H3 Y) ?- W
typedef DWORD (WINAPI* TGetIfEntry) (
  @  x; }' B9 G2 e; k8 e  PMIB_IFROW pIfRow
, Q# E3 l' d8 D. w; @);


CString translateUPnPResult(HRESULT hr);
  ]* M- x: \6 a7 y+ XHRESULT UPnPMessage(HRESULT hr);
: t7 X; F# i9 S1 b4 ~
% l# ^) a5 u* s5 D
class CUPnPImplWinServ: public CUPnPImpl
{
/ L6 P9 Z, p% y1 B& O# O; p        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
4 K( J* @: I- S3 q( @3 R, P2 p  y// Construction
- ^& q. b6 l$ ^& `1 c; @8 p% \3 Tpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

7 l: q: o. M$ X) v% F7 ?
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
, {/ G4 o7 S# x( K        virtual void        StopAsyncFind();
( X7 W, Y5 T2 m        virtual void        DeletePorts();
. d3 |; _; L2 R7 a" C! c        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
% s. N$ Y) t- F# C( q& n

7 M  Q2 \& ?% H9 r0 w, K        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
. R, h& ]: u: z  T4 i9 ?: @        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
+ H, n6 k# r* O9 S2 A. M' l) r        virtual bool        CheckAndRefresh()                                                                                { return false; };


- ]1 G$ k; L6 Zprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
) W- B( B! ~- R2 d# O. r        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
6 G7 L  j9 T; y5 Y- n- u: B3 d0 f        bool        OnSearchComplete();
# s- L( i# ^) Y3 G6 z        void        Init();

4 a0 V& }* P, I4 a
        inline bool IsAsyncFindRunning()
( x' `' K4 A/ k7 I; C; }, d        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
" D0 `* k8 \$ _3 m0 L8 x/ n                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
* c7 u6 N7 ~2 V) N                        m_bAsyncFindRunning = false;
/ Y4 z, [# L/ W$ ^7 t0 K3 {                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% A' |' ^9 _* H: d8 s                {
* g8 P; d5 Y* w* s5 Z6 m* N+ l2 y2 O                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
* ^. x; b+ a! p" V0 P9 z                return m_bAsyncFindRunning;
        }

6 {& S# b8 \0 M5 ~: u
: S! s5 o4 {' [+ s# b4 E' [        TRISTATE                        m_bUPnPDeviceConnected;
! X1 R- z3 H6 R" g- N/ Q: y
( I; n, H! n* ?9 x  f+ i2 C
9 ^" X' Z6 E. j% o// Implementation
, i, g7 p# t4 x5 d        // API functions
# v: c' I: j! D! ^  A- u$ \, b4 D        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
) o. [7 O) M  C, _7 m        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
3 F' p9 T7 ]# y+ P% x$ z# K) r1 D5 E        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
. P! ^" ~2 N7 @0 Y, n
7 q- k" p' |' P1 B0 U% J: Z" F
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


8 g' X8 t+ E3 z+ ?        static FinderPointer CreateFinderInstance();
" B8 ^# ?5 m0 H5 A9 ?) T' I6 y1 L        struct FindDevice : std::unary_function< DevicePointer, bool >
& g, |: p1 |* y' o! J" T        {
  k  c$ ^% y/ A2 X' h                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
# X$ d- @6 W1 i) E                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

; i8 n. U$ T6 [* n. A( I
/ b+ ~, V2 C3 p( \& m                        if ( FAILED( hr ) )
/ f+ ^# ]8 ^, E4 U" }6 u& n                                return UPnPMessage( hr ), false;

! y. x  s( o/ G% f: N5 ~
1 V& F+ z1 S& O% k4 v8 _7 I                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
+ T4 D6 Q( f3 r$ d                CComBSTR m_udn;
6 b. n: R/ w  \4 Y* a* S$ M        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
  G& F8 w4 H4 o0 t, l        HRESULT        GetDeviceServices(DevicePointer pDevice);
; A9 M% ~6 D$ K  {+ a9 o( T        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
5 q4 L4 C: P8 o        void        DeleteExistingPortMappings(ServicePointer pService);
* K. r7 C" ], R+ L# K& n" \; W        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

5 K: O+ L* G6 w6 p. s8 R) P& R6 q
        // 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);
6 X3 A( q; |" N+ p7 c7 y4 R        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
. Z! ~3 r4 |  A9 x4 x5 ~

// Private members
private:
& d, }! J  _; c3 r4 s3 e5 h9 O) J        DWORD        m_tLastEvent;        // When the last event was received?
! C, e+ k; a9 a9 `! j        std::vector< DevicePointer >  m_pDevices;
# I3 D, G6 ~& V' K        std::vector< ServicePointer > m_pServices;
9 x. i+ ~6 y! t  J5 N3 m        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
& Z+ k/ z) H5 d, n3 S) s
7 Z  U8 g" m6 Z( l1 H" t& w0 q
        LONG        m_nAsyncFindHandle;
# m, A4 {$ F! i$ L# Q! D! Y9 f( S/ x        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
+ ~$ y4 j  j: F        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
4 _' {" X* a# }4 F$ o! R" X& K        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
! r5 Q& U& s4 d8 J, N        bool        m_bDisableWANIPSetup;
. m; L0 g9 [" F        bool        m_bDisableWANPPPSetup;

' o% G6 R7 Z# b' N! k3 j+ G: f
};
$ G% A! N) Q( C* |3 t
5 `4 S8 ?# [! d+ [3 n! g5 Z. I
3 p" x$ F( `4 M( _! d// DeviceFinder Callback
6 M! f- ?6 N- sclass CDeviceFinderCallback
6 B0 {2 N& t# @$ I2 @        : public IUPnPDeviceFinderCallback
, L: f+ z) F% L; R  U* \{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
" }2 u4 V7 C) i! h0 @                : m_instance( instance )
" x% J# t: r: S# p0 j8 A        { m_lRefCount = 0; }

: _! `( h4 k* t4 R+ V
8 t5 t! j& I' |0 i4 h2 R( ~7 o   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
2 ]7 T) e$ s# B

// implementation
# k- d1 ?: w2 Q- z$ ^( B+ _  Oprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
3 z- J, P6 T4 t4 T; R# l1 B+ @: B        HRESULT __stdcall SearchComplete(LONG nFindData);

5 m( B! |! g: o7 ^
- G3 ~! K3 K" W, sprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
; r# S" i6 f8 v: z2 W) ]" ~};

9 S( c/ E- \9 I# {% k
# r' J9 ]2 }, ~; E) T+ G4 S// Service Callback
& u# I& g; D$ t) l9 hclass CServiceCallback
6 {3 v! i  E& V+ A9 Z        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
6 m& w9 z( |9 Z1 ]6 i, q$ L% A9 q" a- B        { m_lRefCount = 0; }
   
% N) _' p. ]$ e' ]9 J   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 {6 T' V# F, `, d3 u: b" g   STDMETHODIMP_(ULONG) Release();


; b7 I1 a. ~5 M# I0 J! Q// implementation
' X$ G! X+ {% h* m$ c, bprivate:
9 Z$ e! H$ \: }; ]" @! Q8 N8 |        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


5 w( J. T8 F. h" ]private:
/ R( ]7 W* ^" k& H" K        CUPnPImplWinServ& m_instance;
/ D# y9 s+ H/ e2 C/ D- [$ d0 L1 Z# `        LONG m_lRefCount;
};

1 q3 ]0 s6 D5 [" t  N  V* Q
/////////////////////////////////////////////////
8 K1 ]( J+ ^" M  D& U1 Y
* `5 r' z+ @5 v& ]% P
# y" u( a5 w5 ~. [1 Q使用时只需要使用抽象类的接口。
" n0 {8 }' y2 S6 Z  k* CCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
. W2 m4 ~* X  ]' D8 R' ~: ]& kCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
0 C" C, z% R/ T( p/ U" GCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.