UPnP

天云上人

/*uPnP.h*/

1. 
   1 Q+ q) Y0 ?8 e# f* s; c
2. #ifndef   MYUPNP_H_
   5 q2 F7 G7 b6 L+ o2 ~/ l8 P& L* q
3. 
   ) ]  k+ q- ?5 L* D6 X8 ~8 v* j+ G% Q
4. #pragma   once
   , R6 O, s0 o7 t$ H8 I
5. 
   % h* X9 ^0 o$ |2 k. G
6. typedef   unsigned   long   ulong;
   ( a: D; M9 i4 u# W; j
7. 
   0 Y0 {; C" ^& R  B1 ]8 H
8. class   MyUPnP
   , T, }9 n. v1 w' o+ d& H
9. {
   7 J' f; T1 P% w$ |/ r% K
10. public:
    
11. typedef   enum{
    # U( `+ V1 c; Q
12. UNAT_OK, //   Successfull
    * F, G& ^2 Y" b# e; N# L1 Z
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    / B! f" R3 q1 Z9 y+ h5 s
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    3 B: A5 S" G: T* @: F2 W
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    3 F8 _' A! `' G& w( P$ V% T
22. }   UPNPNAT_PROTOCOL;
    
23. 
    0 ^* s- ]' X4 M2 x0 s2 t
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    $ p; y+ \% l: `8 T
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    * x: ~* l- l: A( U
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    , d' n0 I) k- k- M/ q1 j) L
33. 
    3 U6 R  b0 D2 ?7 H
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    8 g6 ?' G6 w& `+ B& l2 Z8 D( l
36. void   clearNATPortMapping();
    7 {8 {+ I. s3 E3 f9 J
37. 
    
38. CString GetLastError();
    & J4 ^0 g5 z; D7 I+ V
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ! x0 b; L0 T& u# F8 Q: y: w
41. bool IsLANIP(WORD   nIP);
    ( A4 Y' J2 C0 p. @( d7 m
42. 
    1 y3 o$ m- s  w. ~- f( \
43. protected:
      y! O# M, I( P5 u( Q" }# Y# K
44. void InitLocalIP();
    8 f2 q$ K6 a$ h
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    # y0 x* T/ E7 I3 {8 V4 Q& K
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ! q2 c, r5 p7 W; s. J' O1 P' {& @
50. 
      i" ]. k, j. Q" p- E; T; o
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    " j5 g! m0 F5 c$ I
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    # D& v' D+ |( U. \$ o: ^9 r3 y" U6 E* [
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    . c; t5 y; a" [# w
62. CString m_description;
    7 M/ z" s" }, k1 c& p+ F7 h
63. CString m_baseurl;
      w1 L' q: M) i8 ~8 T" @
64. CString m_controlurl;
    4 P" K/ W/ ~7 _" X  a# C
65. CString m_friendlyname;
    0 A0 g( `2 c6 U! ?# d
66. CString m_modelname;
    
67. int m_version;
    
68. 
    9 A$ b! `& m* x! ?+ `" @# U+ @
69. private:
    + o' Z7 g0 ?4 E2 K  l1 x8 L6 V- T( a
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    # G6 N& @' p' t& ~! z
71. 
    $ h* U; z  Y3 \# f+ I0 h
72. CString m_slocalIP;
    
73. CString m_slastError;
    % y! U/ V' q# w+ O2 K9 O
74. WORD m_uLocalIP;
    # w0 X% v' Z7 |: ^! c
75. 
    
76. bool isSearched;
    
77. };
    3 S6 E0 }8 O& c; `
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   - |: J& j: [, q. D% W% @
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
     }# Q/ f6 @, R2 |
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   " V; _- t! }  m. X
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ) }) |" T$ l8 Z* F+ t* {
11. 
    4 N8 C0 F) i3 A( m4 r
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    5 C8 ~; Y/ ?: y1 Z
13. static   const   int UPNPPORT   =   1900;
    9 v1 J4 ^6 K+ ~6 S! j
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    9 }# x' y' t! Y2 D: P
15. 
    0 Y% C# F1 z  a0 N; s& S
16. const   CString   getString(int   i)
    
17. {
    ) m9 K* ~  ~8 s* u; O$ {  z
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    $ M7 P7 R8 V$ M  w1 T1 x
21. 
      S4 g6 O) S6 p4 y$ w) _5 i7 T
22. return   s;
    # m' Y5 X/ f: a) ]5 f% ]
23. }
    
24. 
    % Y1 e* v/ N/ `7 P
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ; p" }' R1 ~- A
26. {
    . Z% A+ ~5 E7 U
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    $ G/ r/ g) S5 c$ @0 O! R2 d
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
      R- x$ J8 o' ]7 p% A
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    0 w+ Q7 {' S1 T3 a+ T$ w3 @% f, a
41. strcpy(buffer,   (const   char*)sa);
    4 C! P8 s8 Y, e, R0 l
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ( ]3 ~$ j3 |7 Y# y+ w" Y; L. m
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    4 K  |4 N$ D2 s3 D! W/ a4 R
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    9 O3 X. i% h1 n. p  E  F
51. ioctlsocket(s,   FIONBIO,   &lv);
    - ~5 o5 M) U2 q8 t
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 L9 E2 q3 O4 P' m- {
53. Sleep(20);
      C, D6 H$ _" ]5 F2 V& b
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    6 i6 n# ^7 E9 d1 K/ T% q' {. {
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    / \4 P2 ~3 m) p  h; j4 t
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    . J% ]: ?1 h) b- G1 u
67. {
    
68. char   buffer[10240];
    6 p1 n3 [+ x4 L6 H- N
69. 
    : y1 h! c$ n% o1 f0 V
70. const   CStringA   sa(request);
    3 X& P: A3 n7 ?) E( x
71. int   length   =   sa.GetLength();
    $ _% z( P4 w0 i+ u& U" U: Z
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    * w9 [- r: {8 g
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ B( s& L  k' A4 U# \. C/ l
81. }
    + V: m- R+ p5 C6 I/ a# K. `  g6 Z
82. 
    ' n1 d$ C  a: o# ~
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    4 h  ]! ^* E$ Y; n' u1 |+ m% M
86. 
    " e7 D! G5 P/ d( N/ o2 T
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    7 \* S9 A1 H3 q* s" ^4 ]4 ?7 w
89. result   =   response;
    2 ?( F' ?9 C& M' [. V1 u
90. result.Delete(0,   pos);
    + r. I- v/ q/ O2 R0 o) ~+ E! @
91. 
    
92. pos   =   0;
    6 Z) ^$ ]) c. h* q: a8 D0 t3 O
93. status.Tokenize(_T( "   "),   pos);
    / A8 r) }+ p( I* q+ e& s5 Y* X! b6 ^
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    , H, D6 X! q9 n+ T
97. }
    
98. 
    ) |& L' ?* i2 n  b7 L9 t" e$ J  [- m
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ) s' [4 g' P& j
100. {
     9 ]* K$ }5 S+ {7 E/ q, N9 L
101. CString   startTag   =   ' < '   +   name   +   '> ';
     2 M7 x& E0 R2 i1 }
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     8 K& S- V4 N' R" m/ P
104. 
     
105. int   posStart   =   all.Find(startTag);
     & b7 R2 B' ^: @4 w0 l
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     , P5 y, k! {9 z) U! S
109. if   (posStart> =posEnd)   return   CString();
     1 ^5 x/ @7 ?2 Y) o/ A$ i8 w
110. 
     4 ]5 I* U; Q& L1 a
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     2 G9 W/ ~6 \' G! v: b
112. }
     
113. 
     & {. i7 D% g7 y) f! A: ?4 f
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ! L8 D  |0 t/ r
117. m_uLocalIP   =   0;
     ; y" i: f/ j1 f
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     8 _) `; n+ @& r8 [: e- i7 V( u
123. UPNPNAT_MAPPING   search;
     6 I3 t1 M6 k, N0 M
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     / ?) i0 d% ?5 W8 C4 u# v5 W
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     0 E5 X% R: f: M( i% @. y( E
128. }
     
129. 
     7 p% S5 c) x2 o) }/ ~# u4 ]
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     9 u3 o7 ~& |: Y- ]
135. {
     
136. if(version <=0)version   =   1;
     $ H4 U' l7 Y, G% Z7 ?! h
137. m_version   =   version;
     + k6 x# @1 Z+ C: T2 S
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     4 @9 ^2 b9 R- r. n1 |. R
142. {UPNPPORTMAP0,   _T( "service ")},
     ( \) D2 u! @1 U# z7 ]* G
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) i" D6 j8 U1 k0 \8 u" S* {) g4 r4 E
144. };
     
145. 
     " Y8 s, L) u% s) B
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     " l& K0 |( |2 j
147. u_long   lv   =   1;
     6 H$ Z' J& p; I
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     / c/ `2 b7 |+ M+ F
150. int   rlen   =   0;
     " Q% Z1 K8 h  L2 m
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     * d' e: @# }/ S
152. if   (!(i%100))   {
     " N  s5 f" X, C9 Y; b
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ) y7 D" j4 _1 C. Z0 t% [  e7 C
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     * h7 R& b- I. o' m- a
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. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     # Q4 G# c$ T' @
164. char   buffer[10240];
     8 v8 x7 r, M9 ~& K2 L
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     - |$ Y# O* v! p2 ~, S/ q9 E
167. closesocket(s);
     
168. 
     % a! u; m+ H! M$ S: @
169. CString   response   =   CString(CStringA(buffer,   rlen));
     5 R2 T7 f* P7 M/ F9 A3 m
170. CString   result;
       }. b. S6 j# D0 V
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
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)   {
     ' Q% h9 p' a: R/ B
176. for   (int   pos   =   0;;)   {
     # D$ w( k  w* Z: ~! [
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     + m9 Z* _) c8 t/ x; ~& x* p
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     & o6 m5 ^3 N6 Z0 `! q; c( b, S
185. return   GetDescription();
     + Y$ ^+ ^& }; r; L8 e4 A9 }; I' `
186. }
     - @. h. Q6 x% F
187. }
     $ q' S! v+ V: e  R4 N- R: X6 S( x+ {
188. }
     
189. }
     ) D" N' K) n; }- u2 ~9 {
190. }
     4 C/ Y0 U' F. ~9 q, I5 T
191. closesocket(s);
     
192. 
     7 R) g3 h# M+ c2 j
193. return   false;
     
194. }
     

复制代码

天云上人

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

6 L8 ^: R8 l; ^# Q/ I9 S; O& \
///////////////////////////////////////////
& v, D  ~- s' U2 j$ I: n//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
6 U; X. }6 l& N1 O3 Y+ I8 K

; Y# c' `1 m: p+ E6 T#pragma once
#include <exception>
) \* Y1 c2 |7 ~+ e" i4 T

4 U! U( q1 I# u# `; m  enum TRISTATE{
2 n. r+ t  ~6 G( [( ?4 x9 D        TRIS_FALSE,
        TRIS_UNKNOWN,
: s6 V& o, S) h3 \9 `        TRIS_TRUE
4 L! ]# |0 f1 i; j, g8 i6 z! T};


* T8 W# a2 k. F& [enum UPNP_IMPLEMENTATION{
( }. A2 {7 l: u; h- k        UPNP_IMPL_WINDOWSERVICE = 0,
) U" k+ c  \; y2 g4 Y5 W        UPNP_IMPL_MINIUPNPLIB,
" b* r& B0 B8 n5 x: c% J% M        UPNP_IMPL_NONE /*last*/
5 E0 I2 O! `0 ?, q) h) l$ x};

4 E! ^8 Z' X  V3 N. Z
2 ?$ Q, q' \) R5 n# w# r" x3 ?
) g0 T) g2 O) I3 v8 M2 ~
2 ~. D8 m2 ?. _- J; Cclass CUPnPImpl
$ I! K! w- [& y! ~6 k( M, J{
  n  O1 B; G, X0 I" \/ bpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
! n/ _8 `9 G; _9 ^        enum {
                UPNP_OK,
                UPNP_FAILED,
: P/ j: f. Z  I3 d5 \6 w                UPNP_TIMEOUT
        };
* C; i# d. M5 C) D

5 x/ F  o  Z2 R9 A        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
# p; k- f$ i- t3 A% [        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
* R- {9 M/ g# h# G        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


! |/ K- v0 a: m5 H  C; _$ A        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


0 f4 N6 h8 R  I2 o3 V; J// Implementation
: Y9 J) ^* l! Z# dprotected:
& h6 H# s2 O& y0 k/ ?: `" T5 G        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
& P5 g' p% q2 t( X# u3 S        uint16                                m_nUDPPort;
2 e# a% K, n9 d0 n  ~4 a        uint16                                m_nTCPPort;
& ]' p. ]' v8 F  A2 N9 V9 p        uint16                                m_nTCPWebPort;
% O, K  `  N$ l! [1 E* S6 @        bool                                m_bCheckAndRefresh;
9 W, B- U, V) k

/ W" {& S$ n3 j5 e1 Xprivate:
        HWND        m_hResultMessageWindow;
( ], g0 e3 x  D! J- \7 j; z0 F        UINT        m_nResultMessageID;
% C! h, @7 R% T
; ]7 K# n& r7 v9 ?
6 G# j: J3 B9 x4 k- G};


0 u( X: P& A+ W, z, p// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
% u5 n( g- x( q* h6 V$ h        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
7 Z9 S7 Q) b1 L" I/ O3 p        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
1 R0 {3 k5 X# n0 i( o( A9 o8 C        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
+ C4 W, M" F" s& @};
/ L8 c7 J; n! S+ e# ^
; P$ Y  Q- S: V# q
( L4 A( I2 u8 \; c- X- {/////////////////////////////////////
9 k: l' g. l  }  C  X/ T% e//下面是使用windows操作系统自带的UPNP功能的子类


, G: S3 J$ I8 p' w# }#pragma once
6 k4 L8 l$ P& ~0 Z6 I' W#pragma warning( disable: 4355 )

7 k1 Y& k9 H8 z2 }
% M+ @; |, d- p- B#include "UPnPImpl.h"
3 c( k; V9 E9 t$ o0 t, s' Y$ `#include <upnp.h>
#include <iphlpapi.h>
5 P' P! w$ i7 b; c9 z#include <comdef.h>
#include <winsvc.h>

+ R' S6 C/ p0 x8 |7 i
8 X4 [0 J2 Q' q, P4 l* T#include <vector>
/ U6 V; z* d2 ^#include <exception>
#include <functional>

; p1 z8 g% l( R* W* G7 ]3 V$ y
( s( D: q! G9 ]
( Q0 Z6 g" J# {8 o& \
( w8 }* K3 u/ Xtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
+ t7 `4 m  C4 n9 g' etypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
: M" U+ d: u' ktypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
; Z/ |5 A5 G# z$ b8 ptypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
8 \$ R% q- y+ g# W) L8 Utypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
1 y7 L( S3 S* n$ Ttypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

% B2 X: _3 v' L7 O
typedef DWORD (WINAPI* TGetBestInterface) (
" _8 K  c/ k! @8 c# t9 p  IPAddr dwDestAddr,
, z/ s. x6 M; p  I  PDWORD pdwBestIfIndex
);

( o+ f5 e' B2 V+ ^& i
typedef DWORD (WINAPI* TGetIpAddrTable) (
. @- Y" ]8 d; g  PMIB_IPADDRTABLE pIpAddrTable,
# [# E. m: W6 x. m- s: A" |  PULONG pdwSize,
3 `0 [8 H; A: a5 ^* ^  BOOL bOrder
: w- {* C) \- _; Q2 P);


typedef DWORD (WINAPI* TGetIfEntry) (
8 T6 x2 }9 J9 e1 Q  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
8 x3 X" n; o6 L# o8 Z/ b
* P3 o& p2 S; ]8 z. j
: [2 f; p! _) m8 w1 i. jclass CUPnPImplWinServ: public CUPnPImpl
) p& i: C3 c1 u3 s4 Z  |3 V' V( D7 x{
6 t$ G$ `/ G. z  \6 |, _+ G        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
; S" g# I+ ^: W- q        virtual ~CUPnPImplWinServ();
& \$ x; b* v9 k2 d/ }$ {. d1 m& j, W  w        CUPnPImplWinServ();
5 L6 u& X  W$ J

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
- x9 u! p- c& q. F% g        virtual void        StopAsyncFind();
6 V# k& y+ c2 L6 K! t$ D9 ]        virtual void        DeletePorts();
        virtual bool        IsReady();
' |4 Y9 v' `) o( T8 }" K& V- \3 U        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
0 y! m5 q3 }9 }6 {) }3 n$ M6 t! c* @
# N  X  b; G" @# y
        // 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
        virtual bool        CheckAndRefresh()                                                                                { return false; };

) ^; l( d! I1 i+ b% S9 p1 L
1 w% {2 W! J+ g8 G0 j2 J) Jprotected:
6 Z3 s0 Q: h$ J        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
7 W. i. s" H! r& C, r        bool        OnSearchComplete();
        void        Init();

( Z8 |' `+ R$ p5 m2 \
% m5 M6 w4 M/ I4 V9 W  @% ~' L        inline bool IsAsyncFindRunning()
4 b1 y. [# i7 O' s% I+ V) e        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
" u' g0 l- K( g3 [7 c                {
, [. [# v  x( |) t5 h- s1 v% a                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
6 T6 Q+ n1 {: F7 E7 ]- u, Y  g( r" h7 Q                        m_bAsyncFindRunning = false;
                }
                MSG msg;
! a/ z9 {  v; A$ m; V6 H                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
* J, i* A2 h2 W8 a                return m_bAsyncFindRunning;
5 Q# d) ~, t1 A2 Z* U        }
5 R; A( a  e+ @7 d/ `( b

$ I' j5 t5 l' l8 j  X+ a        TRISTATE                        m_bUPnPDeviceConnected;
. _7 Z7 b  ~( _6 w

// Implementation
( ]/ q) q9 Q, X8 Y; Q; b$ @- F        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
1 \8 @) K& Y" n9 m5 H( H        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! M' ]+ U+ v( p4 F* x        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);
+ k6 Z7 C/ N' g  Y

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
9 k5 }) D' e/ Q; f. _# r3 \6 g        TGetIfEntry                                m_pfGetIfEntry;


8 V+ A/ x5 c7 C/ R        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
/ w" V0 ~) R+ H                result_type operator()(argument_type device) const
0 [1 }5 j/ o+ O* O/ [1 {/ O, `                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

- ~  ?$ w8 J: E) c9 w0 z
7 }! B1 u) g7 ?$ f1 B  j; u' @% s2 X; f                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
# q3 E& \) ]6 c# Z* M, {  w, `

                        return wcscmp( deviceName.m_str, m_udn ) == 0;
+ P: N+ l1 Z: L/ s; C                }
# _; Q& X2 e+ i; x) i+ P" |                CComBSTR m_udn;
! S4 H7 R1 b5 D* F8 K+ y& W4 g        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
9 J0 S/ A7 a9 A* d/ }; W" g        HRESULT        GetDeviceServices(DevicePointer pDevice);
( a# I9 S, ?0 M" G# ]3 d# O        void        StartPortMapping();
) F: ?; Q$ _2 T4 {+ S; @* y        HRESULT        MapPort(const ServicePointer& service);
6 t3 L% G! c# ^* Q2 r) W& H; b        void        DeleteExistingPortMappings(ServicePointer pService);
6 F9 `  ~4 c2 t  O1 H        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

# a. z( z$ B* x8 d
6 m% ~: T1 ~* T& g        // Utility functions
4 d) K: r% w4 c( @0 [/ Q3 B% B        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);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
- c2 S0 D7 V1 V0 p9 o
6 c. y% [2 t3 z( U
' ?# w9 [8 T0 n* E5 L// Private members
$ k' k: N3 F) G7 S$ v# _private:
        DWORD        m_tLastEvent;        // When the last event was received?
, d* R- ^0 s" r% b2 o& `, ]        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
. {7 y2 V: |' {$ G$ u( ^        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
7 v" \5 v# w9 n1 s        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
8 k7 a0 V2 d. ^, [/ d% s8 d        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
: v, u9 z, H. c$ P' j' Y/ V% |        bool        m_bSecondTry;
  R/ I( a5 V7 Q* g        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
5 D2 N! r6 h0 M1 k$ {  n, B  w; \! ]

// DeviceFinder Callback
) e# h/ G. z! u0 ~6 n; K% C- L* ^  _class CDeviceFinderCallback
2 E% v/ [9 `$ u" ~        : public IUPnPDeviceFinderCallback
9 A* r% Z% U  O/ z% P( y{
public:
2 W# ?" E# ^  ]+ s: ^/ ^        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

# W) j, j- s. \; t% v/ l
3 y4 \, n% b, s* k. R2 l4 l   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
" y8 N  O3 \6 f- N8 w" c   STDMETHODIMP_(ULONG) Release();
2 L7 n! {. N1 j+ D$ d  N+ J
% A0 b% V* a; h/ [3 E3 Y; Z" N; b
// implementation
& Y% D3 @% J5 V% a- zprivate:
/ x) Q9 L- D7 Q$ r3 Q# W        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
2 g: r$ ^$ j* B' \0 N        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
  R- R6 Z5 A/ o8 F0 j4 l7 a
# V* L) j; c& M5 O7 ^3 P
* |3 `, [# c/ _! R* qprivate:
& G3 x2 |" P$ G. Q9 n1 E        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
1 U) r9 m  D) g* Z. t; H8 D5 W& S};


" C4 `% ]! f% e$ Q8 k* S5 n/ E// Service Callback
3 O/ [) R0 Z0 @) q( P) K3 a- P9 Kclass CServiceCallback
0 M$ O  O5 p+ }" _' E        : public IUPnPServiceCallback
{
public:
% P6 @7 k6 r/ V5 l        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
6 u5 w# u8 S$ C) B( z% ^: W$ w   
2 i2 n. T' \9 h+ q- Y, n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
+ z- M! Q2 T4 L$ {* S9 Fprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

4 S3 V$ V1 W% A' g/ y# S
private:
        CUPnPImplWinServ& m_instance;
/ t5 Q8 R3 H$ e6 Z        LONG m_lRefCount;
};

9 u! l( e7 ^& _5 {
1 }. F6 E. }, z, R! m4 d/////////////////////////////////////////////////

# Z% I+ _8 R8 b: z/ n( Z' c2 a2 z
$ ^1 U; S$ A6 v# \使用时只需要使用抽象类的接口。
- J) J5 `: h# B. |CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
/ D: E2 m; o4 p2 hCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
" O! C/ x" X& y" M( tCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.