UPnP

天云上人

/*uPnP.h*/

1. 
   " e- L5 @( u' c) [" A' u3 H1 q* K
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   : n4 F& t% R! V4 ^% D  S8 L
6. typedef   unsigned   long   ulong;
   
7. 
   2 B8 w. D! |6 J! q$ x2 w0 n
8. class   MyUPnP
   
9. {
   : e3 R) _  W7 e5 \% x" K9 w; m
10. public:
    . L% L7 o5 ~5 y/ X, ?
11. typedef   enum{
    * ^5 i6 c5 ?5 M: c5 z- A4 _4 j
12. UNAT_OK, //   Successfull
    ! ?( y4 D) R, r& V# a
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    : s7 F5 C6 L2 w% j" X! t8 x9 U* Y) S
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    * c, Z# z& s9 i$ G3 u
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    # o" b3 ]2 Y8 d: j
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    7 z' e/ |3 s9 j! {4 w
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    * t4 S* p+ E; R  {
21. UNAT_UDP //   UDP   Protocol
    5 ~( e% {0 T2 b  L3 A* X8 p! P
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ( k0 u! _) u# o% c- x% f
24. typedef   struct{
    3 U- i: q* H& ]4 D& H  [( [
25. WORD   internalPort; //   Port   mapping   internal   port
    0 F1 x' |9 |% G, V/ h1 t/ E
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ) r* c) i. v% w& }$ l2 h' j
28. CString   description; //   Port   mapping   description
    ) r. t* h* _. H' v# R$ v" A
29. }   UPNPNAT_MAPPING;
      f5 I1 v/ B" X  ?. o* k" u9 o' k
30. 
    ; o# Z2 G, s. k& n. Y" B
31. MyUPnP();
    & @! C& w( z' j* {4 R7 |3 c: ?9 x
32. ~MyUPnP();
      w( H8 K4 y* {4 F4 D
33. 
    : z8 x% |8 A$ ?% d8 D/ r, N5 g  Z
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    + ^9 _" s8 T* P$ k+ E/ }1 x+ t
36. void   clearNATPortMapping();
    0 W0 C8 |/ J0 l6 R* w, i
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    : T9 \" {) y# y  ^4 F
40. WORD GetLocalIP();
      |2 U1 v$ R* V0 J; R
41. bool IsLANIP(WORD   nIP);
    0 l  M$ g  |! g. p) v! X) O
42. 
    8 M, G, _7 x4 j
43. protected:
    ( e2 G% a: v; a8 M: [* P
44. void InitLocalIP();
    8 O" N' Y  s: m# J5 @* o$ i& K
45. void SetLastError(CString   error);
    : S' X3 o4 a  M6 p& a& n
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    2 R1 D' M6 A+ q1 A1 `2 V
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    7 J! s# n# ]% N8 h& A2 [
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    % U/ {1 j# C8 l7 h# z4 ~
53. bool Search(int   version=1);
    4 `2 G  B% E* V% m6 o( T  v: h+ U
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    $ I# @3 F( n1 p; y$ w9 |
59. bool InternalSearch(int   version);
    . b) I8 @( F5 A
60. CString m_devicename;
    6 c! ^  K- Y* K+ o2 K0 Y) y
61. CString m_name;
    
62. CString m_description;
    , F5 _4 R0 {7 N; R3 f* s
63. CString m_baseurl;
    
64. CString m_controlurl;
    + ]6 m3 m3 _. D% J  O6 Y0 c
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    5 k# j7 h9 `5 V& @; t
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    $ Q- I1 d. l+ _1 |( K
71. 
    - o9 k8 ^/ p7 p5 K0 h
72. CString m_slocalIP;
    " ^& u1 A$ V$ ]' a& U2 Z7 D
73. CString m_slastError;
    9 z% X5 h2 P4 K! H- [1 g
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    4 l2 B: Y. ?: p
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   6 W4 \; Q. B5 \# d8 \/ L- h
3. 
   
4. #include   "upnp.h "
   1 x4 R5 J2 r+ U* Z
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & Y6 ?4 ^( R7 J' m6 N* F7 o  p
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
     |3 r# @# z1 q, g7 D" \( B
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    8 v( k* R' i+ U5 m& g% N0 E0 p
11. 
    8 _% I" c5 g; ~1 G
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ) ~; L+ k& Z1 \+ `& K
15. 
    ( i1 g3 v/ d; {
16. const   CString   getString(int   i)
    ! M8 n8 `. o& ]4 T
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    : P- }+ N5 F9 X
23. }
    . g/ M1 U7 Q$ d" i
24. 
    1 ]8 j3 x5 o& b2 ~0 a2 o- c0 r
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    7 c: V3 y8 H5 O6 q) o) l4 E
26. {
    ( {* o* n; g8 |' {6 ?
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    0 K9 X0 ~& n$ S
28. }
    ) T2 P6 D% X% C, K' P4 K
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
      T1 ?. l7 S* N# Y0 \5 ~, u% w  k& _
31. {
    + i/ g9 d9 T. C! p8 O: i
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  b  y0 R  A
36. {
    + o6 ^0 [$ q9 u2 a" U  h' w
37. char   buffer[10240];
    $ g& `* q/ X  R- s# u
38. 
    
39. const   CStringA   sa(request);
    6 u: W  ?5 N1 A7 G4 S
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    1 A( f# b% S  N, [1 D, q# D. S
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    # g" |: m1 p: Y  h7 t% q
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    / F. v! k7 v! ~/ s; ~) r
46. sockaddr.sin_family   =   AF_INET;
    - e- {) W" r: P0 L# G4 p
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 ?. k5 c# A# K% r7 ?' t7 b* e+ z  R
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( ^( G/ ?  f$ T! Y! j5 m
53. Sleep(20);
    ! ?1 \* r1 e8 U% x5 k9 u
54. int   n   =   send(s,   buffer,   length,   0);
    1 ~+ L5 B6 h% F
55. Sleep(100);
    ! {# ?1 s4 x0 B+ t. q# \8 a5 s/ R6 R. }
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    5 t" h+ G9 A$ L: S
60. 
    1 F/ N1 X2 v( a- l- z: F$ j9 `
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    ; O+ C) h/ C) |; u* D
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    8 {7 i3 q6 c( Y: j- Z$ `% B
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    # P5 \8 \5 Z) R4 L# M* e7 U* q
73. 
    
74. struct   sockaddr_in   sockaddr;
    ! ]+ T+ ~' ~! k; Q5 Q
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    9 v. B5 E# K. r
77. sockaddr.sin_port   =   htons(port);
    ) y6 J7 X; P7 S, f8 M
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & Y6 \( d/ x1 s& L8 I& b: d8 w
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    7 \" }# [/ u$ F% H/ j! L- w. V8 B
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / N. H3 {! c* k  p. y
84. {
    
85. int   pos   =   0;
    , q7 m  o: U+ e
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ( E' x6 Z3 o' D  `
88. 
    8 l3 F+ j7 g7 ?6 J# {8 Y; O( ^
89. result   =   response;
    7 ^+ j! `$ \" n/ U: k* [& c
90. result.Delete(0,   pos);
    3 f, s, Q5 X( ]$ @0 G
91. 
    3 v, M9 `' j. u) Z
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    # G  |. {, E% E3 a0 U
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    0 Q, D3 b" V- a5 X" \  L2 p4 p% ?
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     / t3 `" l4 A7 y
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     # j8 ]8 @, n7 h
108. int   posEnd   =   all.Find(endTag,   posStart);
     : g0 G2 }. g9 C) }; H
109. if   (posStart> =posEnd)   return   CString();
     , d# E* V+ w7 l0 K  Q
110. 
     ' e1 ^4 O8 F6 N6 `9 |4 C
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     3 i" O* H: n( ^+ n1 m( p
112. }
     
113. 
     # m/ H" s2 ]2 o. O- D
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     0 n! G; A6 ^( J$ p, S
117. m_uLocalIP   =   0;
     * O2 @. a+ M! Z  Y9 y. w6 Q2 J* L+ J
118. isSearched   =   false;
     
119. }
     & _& B: u1 F# v' u9 E
120. 
     
121. MyUPnP::~MyUPnP()
     5 l. q! f$ e0 C" H; P
122. {
     
123. UPNPNAT_MAPPING   search;
     ; u, T% w* s5 J1 C( K
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     4 w' q9 V  x9 h) x* }
125. while(pos){
     2 a, b9 L( _& ^* B
126. search   =   m_Mappings.GetNext(pos);
     3 q  \- |/ A/ S2 }
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     * _8 h: @3 @4 x! f% L$ I9 k8 a
130. m_Mappings.RemoveAll();
     
131. }
     ) G+ A3 `7 ~& \) _' }2 I- j5 f
132. 
     6 d$ k) F+ S. ]6 }2 U) j) `8 z
133. 
     ; q+ i) f# I6 Y6 o1 q: h0 Q
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     - [5 V+ K5 r1 [- e
140. CString   devices[][2]   =   {
     ( ~% {! K7 t- P2 x( n; f1 i# J7 M
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     8 I( R$ \* p8 d
145. 
       C9 z  {2 W* v9 b5 e$ c
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     5 f6 A7 |" q  C! R
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     . k- N* H9 d! X
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ; z# `2 E3 D+ f! d
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);
     1 [7 _# J  g6 C+ i* M' k! R
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 Z# A  m4 s3 G% x3 c: z" ~
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     4 s) K! D" \: `" ?( A
159. }
       G1 i% Q" G* o+ {3 {5 D, N7 O  y
160. }
     $ i1 I, [9 ?  b8 r" ~% V
161. 
     
162. Sleep(10);
     
163. 
     . L7 T& z' L9 x% n
164. char   buffer[10240];
     1 J/ @' F& a) D( h* W8 u
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     - g5 M' }2 M4 {; v4 E% M  d
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     # H& d3 ]0 r3 G8 L' r
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     6 K3 l$ m* H5 A+ M, y
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     5 w. S# R" M2 C5 Z
175. if   (result.Find(m_name)   > =   0)   {
     ( z- ~; s3 A2 @& X" A/ a6 D3 L
176. for   (int   pos   =   0;;)   {
       w( y/ M+ I% c" ~" @- v
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     & C( S2 h4 l) l8 B: t( N, A
178. if   (line.IsEmpty())   return   false;
     - ?+ g, n. ~4 ^4 `8 y, Z, r
179. CString   name   =   line.Mid(0,   9);
     + r7 d: P& g6 `$ _
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     ! Y5 ^' }0 K( t0 n5 k. F
189. }
     
190. }
     5 s/ d) {; v& K% Q1 `1 R
191. closesocket(s);
     
192. 
     : q3 g) \& E# R+ K! y, ?* ^
193. return   false;
     2 c* z6 p8 n% _; @
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
; S) i2 ]1 z& X& X- e7 R/ q/ B

///////////////////////////////////////////
; r3 {5 O1 s5 S6 `8 S3 d; x$ n8 V//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
  w, L- ]! E' t3 E& d0 ^3 `8 {
6 ]2 i4 @& Z# u; K4 |
7 |) i+ G8 D0 Y" Y, j' b#pragma once
: O) S4 A9 ~' l2 @4 `; c#include <exception>

, ?; y4 D) R8 t, l% G
  enum TRISTATE{
  Y9 \3 b; }4 e7 L" ^* I( }4 v        TRIS_FALSE,
        TRIS_UNKNOWN,
* q8 Y5 w8 B# x: p& S$ U        TRIS_TRUE
};

8 A# W- v( I+ \# W+ b% s
2 ~- c8 a" ^7 N. ]5 O: |2 D0 menum UPNP_IMPLEMENTATION{
( r8 ]+ G( A! r0 K$ n; z8 }        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
/ ^. G) x2 H6 v};
' J2 X- ^* W. ]! }0 a


" ~9 v8 E( Z+ s6 H2 G% |
class CUPnPImpl
3 V6 G6 i, B7 _" ]8 Q{
public:
        CUPnPImpl();
5 m# D* R- O1 s4 N4 M        virtual ~CUPnPImpl();
' n1 |4 Z' S0 A2 A4 g7 \) J& z        struct UPnPError : std::exception {};
# o3 f8 a8 A8 Q& i* n5 X        enum {
                UPNP_OK,
% q7 v- R3 h0 j* ], y                UPNP_FAILED,
                UPNP_TIMEOUT
        };
! {9 J) W- v9 E) q! W0 M

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
" P# ^: y0 x4 o: p; G" u4 W+ x        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
, o) I' I' U% A1 A! F        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
! G6 l8 Q, R- x       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

6 S# r  E' {0 P
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
0 a' P) N" y; S$ i+ K; S3 h        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
4 w* R# S! `+ q+ b, D/ L* ^! Z        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

  w3 D! _7 \" Z5 Y6 Z5 D- D
// Implementation
! T& R# \0 W( u, S; h% xprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
5 L5 ?" H2 k$ P* d* n% @/ g

private:
" @* j+ W$ q- M! T        HWND        m_hResultMessageWindow;
5 H; ?3 c0 O4 Z0 f        UINT        m_nResultMessageID;

- U, T8 g5 n( l' U
* g9 G5 m- [( |. Z& X};


2 D' V2 {. [: y' a2 E, e$ w8 m& H8 j" y// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
# X" u4 X1 g5 {: t% K; z  B5 w{
9 x$ u& x/ i, t$ v# zpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
* ^) Y. S  R* j5 f/ }$ S+ _        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
0 ~" }; V5 j: y* E5 s1 D1 r        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
- d+ u4 g/ _& e. L) x

5 ]4 L2 \, [; m$ z/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
0 Z% l; V& H! G9 ~

) j  ]$ K+ k# N+ e+ N* U#pragma once
! R7 [# \' e* M" p# a& ?#pragma warning( disable: 4355 )
# x" q4 ?: O* _6 \& V$ |% R( K

0 t5 d0 Z% x* |) ^8 C% S/ f#include "UPnPImpl.h"
#include <upnp.h>
; T/ N  \. }) C2 t: b$ |#include <iphlpapi.h>
) W* N( S: [7 m4 w  d#include <comdef.h>
% V; t. _( `! M, ]/ j  I#include <winsvc.h>
( r9 U  r0 X1 P
. {8 {6 w$ o4 R3 j2 V2 e1 k1 P. m
8 x( E1 }' R3 P7 J2 D#include <vector>
3 N. k. _/ e/ _9 ^6 {- m* b#include <exception>
3 |; p% V6 M4 s9 K7 ^9 V2 F#include <functional>
* c0 ?. w; |# Q2 h& |
. @6 `3 u8 r; i; k1 S

- B( V& ]* P" v
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
& r5 {- X+ Y  ^7 D6 m+ G0 U5 [, Itypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
! e$ d" w' P1 H! N# f- Y3 \typedef _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;
0 b/ b) r' R' Q) n' ~typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
' R- Y" l0 d8 M+ f9 }" ^
5 E( U5 `4 @0 L4 S& Y4 a$ k8 W
. v4 Z- p6 _" K- m9 m) T) d0 b+ l0 Ktypedef DWORD (WINAPI* TGetBestInterface) (
8 A* w# f3 D' v! T4 N  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
* O4 e, ^7 C2 C& X- q) `- T- `
3 B1 \/ Q# x9 ~3 I
; c1 U5 Y3 j4 S0 ltypedef DWORD (WINAPI* TGetIpAddrTable) (
! a- E; o2 A/ W0 |% y  PMIB_IPADDRTABLE pIpAddrTable,
# {+ a% T' ^& U# S  PULONG pdwSize,
  BOOL bOrder
  f' }) u7 A! G" A! M/ W! ]1 a) G);
* u* F2 ?9 D6 ], V

: J" J) `4 J( t0 z- U! y0 p8 Ntypedef DWORD (WINAPI* TGetIfEntry) (
6 d2 m8 Q- |6 W4 F  PMIB_IFROW pIfRow
);


' D8 U! r( X8 }CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

/ P' p: ]( c( ^: U( S9 x  X: h
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
% l" ?4 Y: x( f6 Z$ i// Construction
8 y1 _$ q  S& h9 X" a  t' Qpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
# V2 T9 J) d* m% _

) ]; D' R" O( L* d9 Z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
7 N* T1 t5 D1 C, D2 V6 t        virtual void        StopAsyncFind();
6 ?7 W! q' l$ u; T% {6 C& f        virtual void        DeletePorts();
4 s' J9 a6 P& T7 a& U        virtual bool        IsReady();
2 w5 }8 ~3 B. W$ `/ p        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

! ]4 L! [0 W4 p2 G' Z/ t* i6 g
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
6 q3 r( L1 N3 s4 G& f0 e        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
4 v( o8 q2 T- Y% P2 H. Y% }! h        virtual bool        CheckAndRefresh()                                                                                { return false; };
! d% U! q% P5 O: o4 {  m# o

9 `4 [8 x$ D8 \/ ^2 p- uprotected:
" i9 u; V/ j$ q( p# [        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
% `6 C" h7 Y" U, J# S+ L! a* G        void        RemoveDevice(CComBSTR bsUDN);
4 C- n4 Y+ ]  Y# z: ]4 h        bool        OnSearchComplete();
& K* a3 V4 j& g4 }        void        Init();
5 w0 y. K8 [1 y0 w, ?, r
# d4 w2 c! @) X- L5 c, n9 W& ^
; [" y  {. L& q3 [7 r        inline bool IsAsyncFindRunning()
        {
! M; |5 h# P: i) [8 o! V                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
) j+ Y* P; K( s                        m_bAsyncFindRunning = false;
                }
; ?- B4 P& I0 |+ b                MSG msg;
2 {; G8 x1 M- u1 J) [2 G                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
) {0 U" c1 V9 S: U) g                {
; c: I, f% U2 N" u                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
5 @2 M. E4 K, X
, ?+ O3 {& S: t
: p3 i8 N# w% b! d        TRISTATE                        m_bUPnPDeviceConnected;
# Y: q2 J- p6 n7 M5 {
) O& C  l4 n- {; G2 W9 w( x
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
  q! }# F* i8 @, @# O3 P% {& |5 z        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
# N" e" P! e4 j" p- s        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
) {! C" t& K; J- k, @; G3 r
* f! v% z' |/ ^" C' E* a( I
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
( Q5 w" O" [( f) B; A9 _0 v                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
8 J2 Z/ n5 L/ l! S9 Q                {
                        CComBSTR deviceName;
2 {! I5 w  F) h5 F                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
6 ]2 m) V0 V; ?
  W0 q6 z$ s' c: Y! B3 Q1 w
, O2 ?: T8 m8 s  D                        if ( FAILED( hr ) )
2 |6 ?* o# C0 _9 g. V* L                                return UPnPMessage( hr ), false;

% A3 S+ I3 o6 b6 k' E) T: r
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
3 b0 J; ]/ I7 S* m                CComBSTR m_udn;
        };
; S7 K6 X1 r" T" \9 h  q+ o       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
- c7 Y3 ~) I1 q, L$ e        HRESULT        GetDeviceServices(DevicePointer pDevice);
" Z! D6 g8 }' K        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
6 f3 I( r8 C  G; c) _! B/ n- R1 H        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
) [# Q3 `$ V% y; F% V$ r        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
) f) M) ]3 e9 H  P5 l                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // 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);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
+ ]8 [2 j( x9 F        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
6 k4 [/ `8 q  O2 R. O6 c/ D. T        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
$ _! D1 B# [% u! i2 G) n        CString        GetLocalRoutableIP(ServicePointer pService);
! W8 l- B( |  H$ Y- U

// Private members
9 \  t! E4 T- `7 Wprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
# n) h" e0 ?6 ^, K/ `. V# N) K' k        std::vector< DevicePointer >  m_pDevices;
% C! J$ V+ {- V+ E% |' P2 z' Z) Y        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
, U# j& t0 e9 R0 l! i! E        DeviceFinderCallback        m_pDeviceFinderCallback;
& v8 i" d0 S# p  d2 K9 v3 o' r        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
9 {) [2 O( h3 a/ |5 _        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?
' i( e) V& E2 D' Y2 P( F        bool        m_bInited;
" H8 ^1 [5 }6 y: |8 q/ j        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
* O( \& a  d9 }) i: J* h- S        bool        m_bSecondTry;
. w$ \9 S/ C! v  c        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};

. u8 o! _$ M2 }& _
! S8 f" x9 l, H1 `5 m% O// DeviceFinder Callback
/ P- e9 w0 b* n1 m- r8 r. c9 ^class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
4 D& u! p  `+ {/ a$ p{
( h5 h9 B/ h- ]) D# Wpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
! k! q$ ?- O- W% E! h. U
* \7 H- l$ S% [3 j& j
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
* L4 n3 T: A! k4 A7 K6 T4 y
9 I% V: |, E+ F, W! o1 i
: t5 {! v# z7 `' u( F0 i8 I// implementation
private:
+ H' f1 Z. P+ `% V( R5 s        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
8 q$ a1 K5 s3 z4 @, l/ M        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

4 I( U, u- `$ s) I; Y3 v
5 n$ u0 H- T, |3 U. Aprivate:
        CUPnPImplWinServ& m_instance;
: }9 o8 A( ]9 |$ J! R! }        LONG m_lRefCount;
) C- P6 I: V2 @3 ~9 a};


  ]  z9 ^' U- W2 ^0 k' M% u// Service Callback
* s* P: [) e- j" K( u; aclass CServiceCallback
7 e# ]+ G2 A1 I0 @( h: h) ]        : public IUPnPServiceCallback
) F6 @+ R- E: A9 G9 r{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
; c& V3 y4 Z  h9 u7 Y) y* e1 T                : m_instance( instance )
! n1 b9 m6 w9 I; d  D# x) B        { m_lRefCount = 0; }
0 q9 R7 U* _2 O# _% I+ U   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
2 S8 Y$ R4 e: I& v! F   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

6 X1 ]! i% x( Q2 K
9 o# g1 x& W; \) j// implementation
' Z# c$ U! ?! N5 jprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
% H% M( f" J4 a" X5 N7 u
% z: D/ F$ z, e2 p
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
+ c$ S- {& ^8 F% H! D};

/ o& z3 L" P  `7 D- x1 C  y
8 `( W3 Z/ O' @3 `. Q; B% g; k% ?/////////////////////////////////////////////////
+ O+ P7 P1 }3 R
5 S  r* T" }2 t6 f+ [6 r
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 e3 E) T0 Y3 @# F. N9 GCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
( w* n* m" t: d8 \! MCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.