UPnP

天云上人

/*uPnP.h*/

1. 
   ) o% V- F) [% Y7 m
2. #ifndef   MYUPNP_H_
   
3. 
   " o* ]. Y% D6 ~" F7 c2 |% e7 i
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    % {9 Z! Z* e3 [& V) w- X& M& W7 d
11. typedef   enum{
    - U' [' f" k7 o% s0 T' X  r$ c. ]
12. UNAT_OK, //   Successfull
    ; Y0 x) @$ ^. n% o( F* y& G+ [! C
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
    7 \# R. b0 M7 z
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    2 F0 o% t6 A0 c) G
19. typedef   enum{
    4 r' \. J. }4 x6 e" E2 p
20. UNAT_TCP, //   TCP   Protocol
    7 b& J& B% I0 n" j" n  m! C
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    3 n( U8 F* ^: j/ E# r6 m
25. WORD   internalPort; //   Port   mapping   internal   port
    : k" u: c) U8 [/ C1 l
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    " p5 V$ M" u4 C9 f' E
28. CString   description; //   Port   mapping   description
    8 r- A$ m- B  B; H7 v! `, t
29. }   UPNPNAT_MAPPING;
    % b$ G( P& w8 k/ P% f. \
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    3 E- {8 c  h. n
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! l5 @* i1 T. b0 I- s
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ! U5 N0 K" \  p5 t
37. 
    ) @+ F# y7 Y5 t& a& L
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    & g, j+ U! M) T" D& B% ^
40. WORD GetLocalIP();
    " I4 o0 B1 z" b, E5 ~
41. bool IsLANIP(WORD   nIP);
    6 S2 |2 D8 b% k# U6 {7 v3 S9 v/ W
42. 
    
43. protected:
    ) Q* z, _9 A/ n$ l; x( a
44. void InitLocalIP();
    ' U. U9 @' a3 s
45. void SetLastError(CString   error);
    $ b. N. U1 u$ V. Y( |
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    " r- Y3 n( i; F; l, D- m* e+ L" ^' C4 K
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    3 K# l0 T6 V2 j! h4 t! t4 k
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    # f5 d4 ^" w5 J! R
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);
    + f4 L9 i- _! I# D3 k" `- ?
57. 
    4 [0 d6 [  `7 D' |. x, ?% q/ l; z
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    0 Q; m( H, p0 T+ N1 T5 U
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    0 W3 D4 H: I" M: N7 @' G! l. t
64. CString m_controlurl;
    2 }8 [9 g( F6 M# {' G; s
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    & Z  f; T) i' Q  R  h% Y) m
68. 
    ' D" b; o6 G; Q9 h# [! j, I
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    , T2 v5 K, V- a3 Q
71. 
    
72. CString m_slocalIP;
    5 [" E. E  d7 M( B: m3 q
73. CString m_slastError;
    8 Z7 r. C- @- Z" X  ]* w
74. WORD m_uLocalIP;
    $ V: `# K% ]; N' _: r- \0 |; U& n
75. 
    # ]& q5 ?) N( ?5 T8 w3 ^
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   2 m& h8 H1 z( N) [) z
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   4 M  w  ~0 I# H3 Z; j' I+ ]6 T* m
5. 
   4 N2 Z9 C, i: t+ v; `8 e6 I$ c. q
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   6 P' w1 J$ d; _' K; u
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    2 s$ B8 s- _0 l: B3 @% u) S$ L0 c
11. 
    0 p" L4 w" I' W) u5 Z+ L" r) B
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    / k% a% H* o; o1 [( U- G: p+ O
17. {
    
18. CString   s;
    
19. 
    3 e' R. ]! d/ u/ M: ^) x3 |- @
20. s.Format(_T( "%d "),   i);
    4 F) H8 y* k2 S! U: B, {+ Q
21. 
    # ^, o: h5 m" c: |' _
22. return   s;
    
23. }
    3 ]* B3 M$ ]. _3 O0 D. J) w( _
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ! l# S6 T# n# J. X5 j$ _
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    ) n+ w, H" @0 Y* }- ^
30. const   CString   GetArgString(const   CString&   name,   int   value)
    # b" C; y4 _. b1 R) e+ v3 G
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ( X! \0 `* C- t6 O
34. 
    % L$ C0 M' H9 d, z1 R
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    # @2 m1 H. v0 |4 S7 \! M
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    & u3 Z# b  L, g/ y% D
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' G* g& g  G3 I$ ~# J$ G" z0 k4 A3 t
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    + c8 h$ r4 H+ j$ e2 [+ T
51. ioctlsocket(s,   FIONBIO,   &lv);
    * Z9 ?1 s% u7 l6 e' _5 @
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " ^; Y8 G5 {1 H8 F3 ?% ]# w8 ~' R
53. Sleep(20);
    - }0 l9 O% L% u0 s7 w
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ) ^* }) q0 m7 l# n
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / u: `& d2 E' Q: l
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ( E# f) g9 H; j& y, P+ n5 p
60. 
    ! z; W! [: D7 I1 u& p
61. response   =   CString(CStringA(buffer,   rlen));
    8 q: U* b. c8 K) {
62. 
    
63. return   true;
    
64. }
    
65. 
    3 }) K4 E7 J7 {( L6 d
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    6 r( Z0 K* G5 Z$ u8 u' L- l
67. {
    
68. char   buffer[10240];
    : K0 s) V" \2 W( A. p% k) \& Z
69. 
    ( n9 i1 m, h' q
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ! ^& r$ f, Y* c8 T6 p
72. strcpy(buffer,   (const   char*)sa);
    ! V$ {. s0 }5 o* S* o/ Z
73. 
    
74. struct   sockaddr_in   sockaddr;
    - e; @4 P0 v- V  u2 A* |' V
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    : n/ d/ g/ K0 j1 p3 f
76. sockaddr.sin_family   =   AF_INET;
    / z- g: @; n3 R0 o2 k1 m
77. sockaddr.sin_port   =   htons(port);
    + @( A6 ]+ L- {) @) i' B8 s
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 V* i* M1 ?* N. ~. ?6 {
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    ; Q' k' v2 s; V7 B9 y: H
85. int   pos   =   0;
    , P1 M! z2 ], x4 o
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    + o" W% P. a  }( K" u, ]
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    # {" g" j# W2 i9 U  x! W+ A
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    ( H% m8 p8 N- W0 j  P2 P% q
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    # s6 u$ ?9 R, r$ S
98. 
    2 X# K9 {  Y# [9 m" i
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ( }7 E# I9 d' o! `5 d9 z
100. {
     6 X8 w& R9 [* @; C* }, S) Z
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     5 p) i" H5 V& I+ t
108. int   posEnd   =   all.Find(endTag,   posStart);
     + `: l9 X7 j+ z1 a8 _: E8 J
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     + }, y8 J% I% M- D7 ^& h( g
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 I6 ]# K$ k, X# D
112. }
     5 ?. |- a/ A9 @# U
113. 
     : k3 Y9 c( H# a$ T1 o, d
114. MyUPnP::MyUPnP()
     ) ?0 o! }0 q9 R0 R
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     * j8 F7 }5 R5 E, x6 `7 F
119. }
     
120. 
     $ M/ O1 A0 G, s! _- M6 g
121. MyUPnP::~MyUPnP()
     
122. {
     1 I0 b, {5 {& |+ {2 f! B
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     / G4 ]- c' k, _( g* e8 I
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     , M2 p0 {: i  M' ^
129. 
     
130. m_Mappings.RemoveAll();
     8 g6 D9 a) a5 V4 }. a3 p
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     5 l4 t; m8 K, R4 [; m% e* a
135. {
     
136. if(version <=0)version   =   1;
     # E8 d/ i4 K. i7 R" y
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     ( Q( L# v, ]: R9 U4 L, S1 q
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     ! O- G; I- ]) _7 `  O+ u
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     8 I& m  X) D6 Y/ w* ^+ Q2 G/ L- U
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ' a* o- n/ C9 o& ^2 ^0 [( {
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     : v8 P* |3 I: R) ]" r/ K
149. 
     1 T" g" g  y3 v1 L9 a1 z
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     . S) S( v5 x& B( F/ w
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 "),
       R+ x5 B1 A% A- W- ~- D0 q' @4 }
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ) p- N' @8 `! n9 J* o
160. }
     ' o+ `# s) o/ o$ E  a# @
161. 
     
162. Sleep(10);
     $ c& n( {" I, }  q* r2 P) Y
163. 
     ! a% q( P8 [7 U5 q
164. char   buffer[10240];
     % l! c+ `9 Q+ y! ^, H
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     % Y; _3 m1 K' r/ [+ y/ V6 g
167. closesocket(s);
     * b4 ?1 V9 Q6 w1 f
168. 
     7 N2 e9 c3 z! o2 w+ p
169. CString   response   =   CString(CStringA(buffer,   rlen));
     $ {' I. t) a+ Z& ?3 Y3 g/ @$ ?
170. CString   result;
     2 b0 y; f/ B) P  M* n3 z/ I
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     5 K3 {1 o: S* U+ ?, ?+ P- R. n0 r1 q5 h6 Z
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;;)   {
     9 t+ k3 g$ z, p; _  M
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
       E1 J+ F! I5 M1 I( Q# C% ~7 m0 d
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     8 F, v# r- M9 R/ Z6 g
182. line.Delete(0,   9);
     7 X' R  B, ]; {
183. m_description   =   line;
     1 P3 I( g' ]: N- E0 G
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     , I1 M# b' u% C- `
187. }
     7 ?9 P3 {4 E$ l" f7 j( M$ K
188. }
     % z# T7 r* K0 G5 g
189. }
     
190. }
     
191. closesocket(s);
     " C2 s* P2 [/ u4 U: S% H
192. 
     8 F" v; L4 @) s- s5 [; L
193. return   false;
     7 `3 Y' ^4 u, _% T) r+ w% C& P4 Z
194. }
     % @5 O5 q2 t6 J0 J5 Z( p

复制代码

天云上人

以下有关upnp的接口来自emule，
( F' d& S6 Y! J$ o
" p4 W) {4 `% c# j$ G0 ^% ?6 ^
5 Q+ u# [6 a  s: O///////////////////////////////////////////
/ A9 s+ J' x2 k9 _//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
+ N( W7 K5 [5 U: X' q" d
2 Y( @* W* m+ V: ?% U. s. v
# r; P  j( D, h$ z- a: m#pragma once
' ~. k! |: ^5 o, E  k#include <exception>
  B2 s4 O' ]% S! I

. m! G4 S1 I8 }1 c  enum TRISTATE{
' I9 X( z6 S* X" u        TRIS_FALSE,
        TRIS_UNKNOWN,
+ M- Y5 K1 R# \! W) D- X) K$ e3 U        TRIS_TRUE
& @+ U& ?. K/ \9 x  X9 ~. K};

8 S" L2 L2 w- o
- }( M5 h0 p7 V) v# U0 \" z% Q5 genum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
7 ~3 Q" z, W9 f2 D, v0 O        UPNP_IMPL_NONE /*last*/
};
( g4 \7 B2 v/ l7 S

- L' F8 q, v8 U* {$ z* O

! Y% D$ A1 f9 y& Bclass CUPnPImpl
{
% Q- M! `4 O! h: Q% x# kpublic:
3 x; f; k2 s6 I8 D. Y' Z  \        CUPnPImpl();
" N5 T( g/ _: P4 m, E9 O: M) Y        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
$ F6 Y6 e5 \) q; ^# z; ~4 Z9 O                UPNP_FAILED,
2 {9 X$ L2 P9 v9 i" I% Y                UPNP_TIMEOUT
        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
7 D/ X4 p  k/ [; b        virtual bool        CheckAndRefresh() = 0;
- M9 a1 d* w1 x        virtual void        StopAsyncFind() = 0;
' n+ V, B7 V: [! y! A( Q, M- g        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
" j& a" D+ j# T& Z: _        virtual int                GetImplementationID() = 0;
       
# r9 X1 K/ J* b. J, C/ w+ {        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


/ S' T  N* B* k+ \& n* j( r+ O/ a        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
; n) F0 r( |* `9 x$ V        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
4 x, i5 ^( r2 V8 c  o+ s        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
  \4 E6 T6 g2 k1 M

4 I! o; |, {% _! D" N; ?" k! L8 H// Implementation
protected:
3 |  P7 A6 G% |7 {: W        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
. I2 o7 z0 S+ o3 ~3 M3 H- o3 S2 b1 u        uint16                                m_nTCPWebPort;
1 Y( ?0 U/ A; D0 G! D- k9 f        bool                                m_bCheckAndRefresh;


; B' N. R9 p7 J$ B0 Jprivate:
        HWND        m_hResultMessageWindow;
' }+ U* Z# y% D' {        UINT        m_nResultMessageID;
" k- U0 A; v, C* T3 K

+ S3 i2 X+ O" b- r};
8 k3 X8 h: j  `6 d% Q1 \
+ z( Z4 ^& O7 Y$ I- i! }. a* @
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
" b/ J* r' q* h- Z2 O{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
& W4 S  [* I0 o* i( D  N        virtual bool        CheckAndRefresh()                                                                                { return false; }
& D$ \( t) }1 m& W! y: m        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
  H# a+ [1 ~! }; e& q        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
) D& Q' Z* T  E( k};

. Z& V0 B* g2 @/ k: o1 H" H& G
5 J+ ^3 b$ R  G8 C5 c/////////////////////////////////////
/ ~8 Y- V$ L2 ?- X, P//下面是使用windows操作系统自带的UPNP功能的子类
6 T  e2 q$ X- u; E. |8 A( q  {

3 @  M; N7 L, A#pragma once
#pragma warning( disable: 4355 )
( k* }8 P$ ]( {# e3 O
& i8 [$ r- I2 d$ e& w
#include "UPnPImpl.h"
8 U( I' F$ H1 I#include <upnp.h>
#include <iphlpapi.h>
# m6 K  I' ^/ P, r#include <comdef.h>
#include <winsvc.h>

7 m- w$ Z5 d0 w
( G1 M! ]- `# I4 A1 H& z& @: S0 \#include <vector>
  b& J1 v& y. ?; ?$ x8 v: c% {8 `#include <exception>
5 Q5 h% }7 B2 M#include <functional>

6 D5 x5 Q% S2 V" `/ ~. f4 |8 h
- P8 j* V1 u6 I2 @8 F2 Q
( }1 c. }  O" P- \% E# w8 b
0 {" H+ N4 Q! j! ~" I3 ytypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
- c3 C# ]; g1 ]6 O# x; qtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
; b; a5 A' r3 c  Ktypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
/ Z/ ]8 p* V& q$ d2 Z* t' itypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
0 v0 R: ~5 `* @8 d) {3 L* Ktypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
& \4 e! ~- I( ?) x. B6 S8 A+ ~

& O3 i. U- ]; ], L* B/ ^" r3 X' [; \typedef DWORD (WINAPI* TGetBestInterface) (
9 I6 _& z% s- ?5 _  v0 O  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

9 n6 ]; n# I! Y+ F2 s; [
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
7 q& D9 |5 I' d& d7 S0 T/ G+ X  BOOL bOrder
. D, A& x7 t9 l/ I7 ?3 v( w);

2 C7 ?4 C  n: E, V
typedef DWORD (WINAPI* TGetIfEntry) (
0 p$ i. R6 R$ E9 _( R# }3 V  PMIB_IFROW pIfRow
) n. F3 ]4 E6 G' l6 ]: e) I% G);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
) X- c& E* p$ v' t  V5 U! P

class CUPnPImplWinServ: public CUPnPImpl
" c6 z& ~& @8 h& {) n: F8 @2 a{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
  F. i4 n( H7 ~, u6 K2 Y" |        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
5 M( b) H* r8 l# y8 e: X        virtual void        StopAsyncFind();
- z4 U+ r1 T9 b7 Q# B. v; c) \        virtual void        DeletePorts();
$ ~9 R* l" @, s: S        virtual bool        IsReady();
: x  ]7 Q$ ^! ^( F' t% k. ^& G' N        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
. V! d# ~+ O# y7 p% @( U
6 ?6 c( t- `; c% x* U! @8 G$ O
        // 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 G$ R  w/ E) _2 Z0 w; k7 M( V        virtual bool        CheckAndRefresh()                                                                                { return false; };
% Y5 L# R0 {: \/ M
) B& j' U+ j, _
+ N7 k$ f2 f) f& m- P6 Sprotected:
) M$ F# U/ }  L+ e/ }  z( j        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
+ r9 |# d" _" k) i        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();

  u( d) @) h) U  t3 t
9 K( n" Y; A, [% l0 K        inline bool IsAsyncFindRunning()
4 E' q9 b# \0 ]        {
. e, H6 N! d1 a" B" F* M                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
1 U. [6 r# `+ [/ P! L. w! k/ t                {
- t( `# p1 U3 O3 ~1 m                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
1 e: z) s, A, S2 @* ?                        m_bAsyncFindRunning = false;
5 M1 V1 R( I: s2 s( i                }
/ ?9 }; r1 l* q                MSG msg;
# e  f- e; N  F% C2 t( ~( h* V% M+ p                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
$ b* A0 k" {5 Y, L+ w                {
& I/ X1 j, |6 {" {- k" p% w                        TranslateMessage( &msg );
; v) V$ N$ D$ _+ b9 g: t                        DispatchMessage( &msg );
" i9 t, y. I3 M/ h. t( M* a                }
- x1 ]" K0 _. d% @                return m_bAsyncFindRunning;
( T6 W: g5 x8 Z2 B$ r5 R        }

# }1 L7 d2 |4 E
        TRISTATE                        m_bUPnPDeviceConnected;


7 o8 U- P1 Y- T// Implementation
2 [/ m7 m0 A' r6 I( W0 E        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
9 o# `( R3 E) o, w7 O4 l+ V  q        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
6 p& ]% N: y8 V" Z9 b$ |        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
, J; @. c8 r6 C; S

+ {6 _! v- M  c2 i        TGetBestInterface                m_pfGetBestInterface;
+ R3 R7 u# q& ~) \        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
" w* [2 `: O8 E* j4 O! s  f( F
2 S# V' O" q- t: T
4 ^. n, u$ e0 _; P% x        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
0 D" ]3 K  {. b; P0 Z: B                result_type operator()(argument_type device) const
/ T! O# B: E8 X) W                {
4 c2 F2 Z1 L# Z2 N                        CComBSTR deviceName;
' v% I' m4 T. z3 D                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

. h( H2 {0 c; E$ \
: s: s( k$ H! k: G+ E% D% i" H                        if ( FAILED( hr ) )
6 m6 D& V$ H9 e. E  n' i                                return UPnPMessage( hr ), false;

! O9 |) S( ^4 ~5 d
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
/ ]% B$ L) @) ]        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
( U$ M9 k4 l- u5 ]3 `; ^3 K        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
0 n3 h# M' S) s. J  d* ^- D        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

% k, O; ]( x) o. q2 m
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
2 K  h: M' y/ S) U6 Z8 x        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
6 ?- k) d7 X# o+ T        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
, N! V' O( n9 c        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
" S9 v4 \/ d% X9 X/ O7 i' b
6 S5 B$ _1 b9 d' r4 s% Q
2 b5 \7 ~# E( ~: m# \% ~* W$ f// Private members
, h5 J- {5 R8 P; P# V: Aprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
7 }% C: @9 o3 Y* _; Q, {        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
  K+ ~- d7 n8 K( t) G1 K  X; r$ ^        FinderPointer                        m_pDeviceFinder;
1 F3 D& ~4 e8 c        DeviceFinderCallback        m_pDeviceFinderCallback;
6 v- p" w; Q( z- y        ServiceCallback                        m_pServiceCallback;

+ ?$ C, I# B2 z8 K4 O  d
        LONG        m_nAsyncFindHandle;
8 G1 O: f9 }3 V/ y, `        bool        m_bCOM;
        bool        m_bPortIsFree;
( \2 I, W( D3 [' X% ]7 X        CString m_sLocalIP;
        CString m_sExternalIP;
& S# C  C- T5 {: C, z        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
# N" m4 {1 Z, y4 n1 I        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
6 H& F3 y' [0 P% L+ l# m" u        bool        m_bDisableWANIPSetup;
1 E$ P+ C( O& V+ |  H4 f' t* f+ k        bool        m_bDisableWANPPPSetup;
. C4 _: z  |2 S' K% Y5 W% m
4 [; M5 N' x# ^$ p
$ C4 D4 p/ s5 [% [8 L1 Q  B7 s# f};


// DeviceFinder Callback
6 g, @" R% u; p" [) tclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
- y' M3 q' x1 @: c4 x7 O" @) dpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
+ B: P' @8 u3 z( c! J% J5 L
* I' u: t6 |$ W3 C2 W
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
' F) ]# B+ y" B   STDMETHODIMP_(ULONG) Release();

  C# G, w' U! N: t' d" R6 S
% s7 w7 ~. L; V$ C- E; p; p// implementation
private:
. L% g& V7 w2 }/ {2 w/ R        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
2 _4 m) @4 Y( s, o! J6 C( w! a. U        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


$ U1 c# C  u$ r' _* Rprivate:
        CUPnPImplWinServ& m_instance;
/ w, l9 Q" Y) w7 o4 @+ g/ x0 N        LONG m_lRefCount;
};

3 _& E9 c/ _$ f. m
0 P& ~6 N' k8 g" i1 D: B& @  ?// Service Callback
: ?4 H! `; l1 r( G1 fclass CServiceCallback
        : public IUPnPServiceCallback
% k$ G) R4 x/ ~. e1 w+ G{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
. ~5 J5 h8 g8 s4 o! o8 }' J1 a                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

) L8 W! D2 a; G1 M* o
$ T0 L6 D+ ?# g' u// implementation
7 K) J: [# r0 o" Xprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

5 Q& e. t; Z5 p8 F0 l) p
  G& h; f- r4 Q. ?: ]/ iprivate:
        CUPnPImplWinServ& m_instance;
7 j: O" Y8 Y' c- C        LONG m_lRefCount;
- Y  s1 j& U7 @0 Z8 U7 i: w};

" B; }% }) U. `( z
* y% Y6 ?6 `1 `6 x8 D9 m5 @/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
" ~; g. `! B, }2 Q7 m6 T+ LCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.