UPnP

天云上人

/*uPnP.h*/

1. 
   ! A+ m: ^3 W* _4 Q
2. #ifndef   MYUPNP_H_
   
3. 
   % b/ ?. @0 E7 j- A" A* L4 P
4. #pragma   once
   * _- p! n9 T3 N4 z4 b
5. 
   - ~+ C: ^, R2 G. A
6. typedef   unsigned   long   ulong;
   + i! z  z+ O+ @, J  M! \
7. 
   ; |! f. F3 w; X' f" U
8. class   MyUPnP
     J- p* l, E7 Q3 L  ^- A
9. {
   ) R/ Z. c4 R, S# n" G+ G3 g
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ( e2 p1 T! \  y) d9 b
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    - o+ _- o% j# Z6 e9 M9 e; ~* c" Q
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;
    8 F* s0 v# u7 S
18. 
    - t8 d% u# m0 \% q, y* b, @0 h, C
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    % t% z+ @4 k( _3 P& n* a+ L7 e$ }/ [
21. UNAT_UDP //   UDP   Protocol
    $ q' C& V! V, S, z& N6 ]( S- ~
22. }   UPNPNAT_PROTOCOL;
    ! F; L5 Y: Z3 I2 B2 B
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
    
29. }   UPNPNAT_MAPPING;
    2 ?* |+ m! }/ z8 L5 q
30. 
    
31. MyUPnP();
    ) i7 p6 M5 _( T
32. ~MyUPnP();
    . H+ q4 T/ N6 i5 U1 l# J  [6 `
33. 
    0 B! f) T& o5 O3 b- s
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! I7 p2 J/ O+ Q/ j5 d: Q9 R* R
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    5 ?3 u. G6 o, Q6 L- A: r
36. void   clearNATPortMapping();
    ' \, @' P  O$ d. }/ d6 S/ D
37. 
    4 J7 I9 _) z; L; g) `# u
38. CString GetLastError();
    + I' I! p; w% o, e; c
39. CString GetLocalIPStr();
    2 f" Y, r/ c. |* F; K
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    % @/ M% a, m6 D5 `) H" J$ \4 \) \
42. 
    ! P6 |3 l, C2 |  I& p
43. protected:
    
44. void InitLocalIP();
    & F9 H  P/ \" l# I, X& _. M1 Y
45. void SetLastError(CString   error);
    3 i4 C$ I) H" s$ ]0 s! ?1 N  {
46. 
    $ O1 c/ L& D9 s- z4 }$ }9 _
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    " Q! Z9 f  o0 |5 K2 c
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ) G. k& A% f( q0 U. q! L
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    6 C7 y, k& d1 l: ~4 M
52. 
    4 m2 k" K! o8 a3 n
53. bool Search(int   version=1);
    1 p$ Z' P2 B7 A
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    6 i. d0 s$ N3 X) v% ]4 D
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    : N& v, O* O5 V( H. m$ U+ F
60. CString m_devicename;
    ! g! h/ z. ?/ R0 D0 E  g
61. CString m_name;
    ' I; {& R  |  {6 A
62. CString m_description;
    
63. CString m_baseurl;
    ' \5 l+ j( V0 X6 R  b
64. CString m_controlurl;
    
65. CString m_friendlyname;
      v4 v8 u& I" v+ @1 N' }$ ?
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    ) _7 \1 m& S2 F, {5 V
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    $ ?/ j' y  ~: z8 x; j; K9 E
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    . E  N+ _' o7 N0 c# S
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ) K( n! o* T9 B1 S" V
3. 
   1 n$ H1 G9 u9 R7 C6 B
4. #include   "upnp.h "
   
5. 
   5 D* T8 Z( T; P- o1 o2 X0 x: k
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    0 {3 N6 _* ^2 A8 D; M& B
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    , l) [0 R& h$ r/ r, R9 ?
13. static   const   int UPNPPORT   =   1900;
    ( @6 p# ?& n" B+ }- }. U6 W  N
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    7 h0 y' b  [. V. R6 k0 n3 E% |9 d
15. 
    # h: Y' k" w- g7 p1 O8 K" i
16. const   CString   getString(int   i)
    . U5 N; B; Y" i+ ]5 Q2 j4 u
17. {
    % X' p! I) I& n
18. CString   s;
    ( n4 W& o$ A5 _+ U4 z1 `
19. 
    0 z: F: L% Y! L
20. s.Format(_T( "%d "),   i);
    2 Z; _7 T; F% U1 {7 r* ~; j6 P
21. 
    
22. return   s;
    
23. }
      l& b$ Q  f/ c6 z8 j- @5 c
24. 
    ! x" E' z5 k, h. ^$ R' K. }4 K
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    3 L: [/ y4 d! c% q7 S
26. {
    5 _7 C* C5 `1 o* J+ V4 q3 q
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    8 I4 J7 [' V0 C5 r# G
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
      d7 R9 D/ Z4 b* U" `& Y8 j5 N: v+ O
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    1 `! Z3 l; B4 u7 q6 d+ h
33. }
    ; ^! e/ a: }( ]/ n9 d# M' U
34. 
    9 ]+ ]4 m2 z2 j( ~  K" r6 ~. q( N
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ( T9 B- I) ^( }# ^, D  u
36. {
    
37. char   buffer[10240];
    
38. 
    & V' g6 D1 F3 o2 I& w
39. const   CStringA   sa(request);
    # v. F; q& l: I
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    5 a  G2 e) w: v2 u4 L; x$ }6 w
42. 
    * n6 E9 H1 Q1 R7 o( k  y
43. uint32   ip   =   inet_addr(CStringA(addr));
    % W  B- `9 g! `/ F( ~2 H9 H
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    & C! d& Q2 U. S: E+ J# ~0 h
47. sockaddr.sin_port   =   htons(port);
    3 j; ?! o' x" w& K
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    % k- a8 {- z1 q! J( Z
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    $ }6 L+ G0 o9 T' V3 B# R% [
50. u_long   lv   =   1;
    % X+ q: k- j) p8 B+ T, Y
51. ioctlsocket(s,   FIONBIO,   &lv);
    ! f: j; o& J, d' C
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ) N, U/ p# i0 y
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    1 s# Q0 b( M, R+ q/ Q" {# v
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    / U4 }) Z& @3 s+ e
63. return   true;
    # Q+ K8 f3 z) I2 \# I3 \) L
64. }
    
65. 
    % k/ g' U2 O& B0 r) h( u
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    4 ?  F- \; L# h+ O) q! K! B
69. 
    ! n0 Z8 Z9 ~% `: I- n! E
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ; E1 n- e* G1 i$ W- J. N$ a/ p
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    ' j: t6 Q+ e1 L7 o
74. struct   sockaddr_in   sockaddr;
    # \- p9 L# _( {% q
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. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    6 q0 ~5 b+ ?; @7 Z
82. 
    " [# g0 F+ E  B0 R! e" k
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    1 T3 r# h) h+ x6 x0 d
84. {
    , E3 b* v6 p& s% w( T; J
85. int   pos   =   0;
    
86. 
    " w" S& ~1 \1 o7 k( m& T
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
      Q! X* U$ _; ^1 e
88. 
    5 J# e) j8 x' ~; k/ T% {: L
89. result   =   response;
    # i. ?; M# L; H/ n9 _: q
90. result.Delete(0,   pos);
    : P+ V  N- s+ j6 j1 h
91. 
    & N6 d0 Y6 R3 u4 D8 ?+ C; c+ P3 C
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    # @8 o# d4 c, x2 U% S
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    2 e* n4 h- w& r0 g8 v1 x4 B# i* j
96. return   true;
    3 C+ F( H3 V# U, K* K) Z0 h
97. }
    
98. 
    / p3 ~6 P: A7 }6 \' u' c4 j
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    . p$ W- J  _8 P
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     % L  \1 \& u: R5 A7 i* Y
104. 
     6 ?) z- {3 b' Q# p9 ^
105. int   posStart   =   all.Find(startTag);
     & Q# c  h1 l- j( V4 @- ]
106. if   (posStart <0)   return   CString();
     
107. 
     / x3 s' `+ A" l3 J$ d
108. int   posEnd   =   all.Find(endTag,   posStart);
     9 A7 T. T( @6 p$ {
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     0 [0 L$ `2 Y( V* \
112. }
     $ {7 D7 l7 r1 T2 s
113. 
     & y$ a; f' q2 G/ m: l$ i
114. MyUPnP::MyUPnP()
     ) g: J% _/ \% N) l5 L. [
115. :   m_version(1)
     ' V" Y" ]: H: m. Z% I; ^
116. {
     . P9 X$ f6 H0 a9 N) G& u4 e; V, h$ y/ n4 h
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     8 f" E: _+ K' f+ Z' f& o
123. UPNPNAT_MAPPING   search;
     , [: f7 s. W6 b' m0 C  X7 v9 y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     4 J9 ]3 X  S5 m. r) S! i8 e% P6 {
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     ( k- ~; v" j7 L. x% M
130. m_Mappings.RemoveAll();
     
131. }
     / F# I* e/ i) Z1 P" L5 k9 Z$ S
132. 
     ' n5 f) k& [" Z' |( Q
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     - [5 D" |9 `% H: b' o; G3 {
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     8 h: i5 g& H4 A8 q6 Y( E: x
138. 
     $ a: ?) D. w2 y2 Y' w- `0 ?, G7 a
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     + q! n. U5 i- ^, ?! x# d! K' {
142. {UPNPPORTMAP0,   _T( "service ")},
     * n" G' {1 G! u/ _
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     . ~$ ^+ L, k4 e, u4 D* Y
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 g- n: ~# v; r  Y2 B3 g
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
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++)   {
     + D" E$ n1 m$ q8 N
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     , S+ p& z6 R9 Q+ S
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. }
     % |+ U* O" c5 Q" K. }
160. }
     ; M+ {2 k3 {; a" b+ y0 o
161. 
     , T! `8 b) G: _) d7 q
162. Sleep(10);
     6 m" I- Z. c3 f( e3 }
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     9 v9 E& v5 i) U1 C0 z% x7 f
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     " A  J  g; i1 P1 X; T
170. CString   result;
     / x0 N6 K" z  w( g
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     4 C2 K3 i  D, n+ T. c1 W+ ]2 g
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     5 O; z5 Z5 B8 z  Q+ x$ I, `
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ( e+ v6 n- V4 o. w" U+ ?, R: Y
175. if   (result.Find(m_name)   > =   0)   {
     6 k9 K. d* H2 H# G0 Z& c0 J
176. for   (int   pos   =   0;;)   {
     ' ~! s6 i/ N& ~
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     : ]) F+ O: K3 f" y% |9 v. ~% y
179. CString   name   =   line.Mid(0,   9);
     , Q/ }5 c/ N" L* x6 t, r  h7 }
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     2 J5 c4 `9 b: l9 r  [8 A$ _
183. m_description   =   line;
     
184. m_description.Trim();
     / E! Z$ O0 Q) |8 p- b4 f: R
185. return   GetDescription();
     " j* U) W* V- h. c) K; I; j9 m
186. }
     / Q& M1 S2 u0 R% B2 M/ I
187. }
     ) L+ X5 u' F3 U! O4 i7 S: {2 c/ Z
188. }
       b" L5 e0 f% v/ y. C3 E0 H  C
189. }
     
190. }
     % q- n( X; t4 y, J5 x
191. closesocket(s);
     
192. 
     0 s( `- q$ J0 a9 C' T
193. return   false;
     
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>
7 w# r. U% a% b

3 A  m, |1 y! i  enum TRISTATE{
        TRIS_FALSE,
4 D0 I7 c: V: V. m' f) N2 o        TRIS_UNKNOWN,
; C$ i0 f  y7 z        TRIS_TRUE
" Q8 A! r/ q6 r! P" T/ ?/ W0 a1 N};

9 T* m0 Z3 e3 _% v+ H
" N% [- n- @& l+ N+ nenum UPNP_IMPLEMENTATION{
  ]' c$ P, [9 Q  p9 O1 N        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
. \7 C( m1 t+ \6 j};
( ?! n) W! m, _6 f



class CUPnPImpl
3 }1 e% m: X( z) `" h# `/ c{
3 c2 T' [$ J! u% d% {$ h# I: x. xpublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
0 [$ z' l& U, v9 H, D+ |                UPNP_FAILED,
: V7 j( k' l- \. t                UPNP_TIMEOUT
        };

" B) ^( X" R+ `( E
; `$ n5 t7 v' M' H4 Y5 i        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
( I2 r- B( K  O# `        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
# X' Z, U# g/ }1 c1 G        virtual void        DeletePorts() = 0;
7 n+ j6 `0 @1 y& H        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
  t) N+ J* \" Y& c! k1 j       
# K' [7 n% Q$ W; s; X3 K: z        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

$ Y! \) a. v# u( ?0 B' l
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
( I- [: Z# i4 q* v) L        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
/ D$ x- J& i9 i) V- |- x  ?        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
8 R% `# j& o  t. e5 U' v* u7 j
& P9 L8 m, E2 A* Q
: d. R$ f7 Y$ g9 |# r- y, T// Implementation
protected:
* g. z" y/ O  P' B) \        volatile TRISTATE        m_bUPnPPortsForwarded;
: E& V0 W: A7 r& ?" k: `( y' r        void                                SendResultMessage();
: J! B1 c8 E- D4 [( T        uint16                                m_nUDPPort;
; \! K6 P0 T9 A3 ?1 d+ _        uint16                                m_nTCPPort;
6 l; M$ w. i; K; X0 n        uint16                                m_nTCPWebPort;
7 `3 C" M6 l4 c2 T        bool                                m_bCheckAndRefresh;
% b, q9 H- o2 }. q, X; ]+ R" g
6 ]0 M2 A+ N9 d) l' f
private:
        HWND        m_hResultMessageWindow;
1 @- f$ u1 G+ [; t+ [        UINT        m_nResultMessageID;


};


// Dummy Implementation to be used when no other implementation is available
2 r. Y7 N1 E+ _class CUPnPImplNone: public CUPnPImpl
{
" c' t9 o, L7 K- q9 K5 Dpublic:
* `$ C" p9 ]7 V9 g) f  g) e        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
( J+ n1 }% i# ?0 }# ?        virtual bool        IsReady()                                                                                                { return false; }
$ V9 `5 r4 @2 _4 w        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
. E1 _5 b; Z4 e, d) M8 l0 Q2 y};
/ R4 n4 y: m. r& ~3 z5 Q& P

& F' G3 ]1 j1 I& g  _/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

! J4 ]  x: y# r5 Y# n3 s" ~
#pragma once
#pragma warning( disable: 4355 )

* f& V' A$ G3 a. Q0 ~: K! f
#include "UPnPImpl.h"
( I: G- g* j, x9 G) H1 G#include <upnp.h>
0 w0 I: Q7 g6 {( V#include <iphlpapi.h>
#include <comdef.h>
8 U/ P& I; b/ i#include <winsvc.h>

$ _/ P6 c" g2 j! P8 M/ F
- I5 l' C. p; c' R' H9 i' a: U#include <vector>
5 B  \2 g. t* P#include <exception>
#include <functional>
4 J7 D( Q" |4 g; `+ C


; _# S- I7 T, d6 s9 H: T- O
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
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;
6 r5 w1 ]) v1 j7 ^8 H( [4 ?typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
' w2 {5 w' V" Rtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
- @. W9 Q& i. g1 H

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
+ ^) B  }$ p+ x3 u);

% n7 Y" d! w8 M2 m5 Z
. j* s$ o: {# G  q/ Ftypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);

  x* l8 Q2 V0 w, Z
! g6 R5 }# n) r( ~CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
9 ^) I( R: M2 ]$ s
. l7 e+ \$ E. Y
class CUPnPImplWinServ: public CUPnPImpl
4 ?$ q, |# r0 k% e% t{
8 |: K+ |8 D2 J' X# |9 J        friend class CDeviceFinderCallback;
& U; e8 U3 a! J# e1 }) k        friend class CServiceCallback;
// Construction
: j& }5 V: ?5 Spublic:
( c4 j# U3 f% Z        virtual ~CUPnPImplWinServ();
% I, [5 h% R& O, t        CUPnPImplWinServ();

: e, o( `1 h! N2 p& \( k/ p
9 |! E  f% L* Z9 P' w        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
* J( z, |. b, N7 d+ n- g        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
4 O" Y; {$ a  Q$ X        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
! U$ H( _2 G* g

        // 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
  g( j/ Q) f% j/ D6 R  s. k3 ]        virtual bool        CheckAndRefresh()                                                                                { return false; };


5 X# H2 X, _# L% z  M& X: Q$ Y3 C) Fprotected:
  b; K! N4 ^% B$ ]: W. T        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
5 b! {- y& u9 ]. j        void        Init();


9 l6 b% A1 O8 g$ d  _        inline bool IsAsyncFindRunning()
: T& S& ~: t4 b# f) }' ~        {
0 q- D8 _2 S  K9 H6 v+ F1 {                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
- T6 d1 Z! X8 P& W5 R  u5 m6 [0 k$ U                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
7 y7 ]6 o4 E3 e; Y4 m- I$ Q, S6 y                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
2 v/ J  f6 A8 P) ?& `4 ?                {
                        TranslateMessage( &msg );
5 w5 R$ W. }! p: O' @4 v+ c                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


        TRISTATE                        m_bUPnPDeviceConnected;

" l6 i" F2 R: l+ b+ v) @
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
% \$ F7 l8 A# G9 c0 J        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
# T% p  t& H7 ^  i' V        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
% y- d6 `1 F  \8 Q& d' H# l. h# r

5 \/ H# D2 ?$ M& B/ i7 k        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
7 p! Y* g% S% C. X* S9 ?; b& t, p
" W7 ]( V/ v; e) x  t2 r$ _9 l4 X
- b& |$ }2 T9 q& `2 j; X                        if ( FAILED( hr ) )
2 K2 [5 M! M6 }" B+ k                                return UPnPMessage( hr ), false;


% k' `9 {* A  h                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
* n, \9 Q8 v& J+ o$ W7 s5 Q       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
0 k' `( H% w6 b0 d; [& _, Z) {# Z        HRESULT        GetDeviceServices(DevicePointer pDevice);
2 X4 i) y/ U3 u7 s  Q        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
  U+ c. G' S1 z3 I/ g9 k        void        DeleteExistingPortMappings(ServicePointer pService);
$ m6 O- M3 y# w        void        CreatePortMappings(ServicePointer pService);
) t3 c* o3 y- l/ L        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
" ~+ K3 V5 J( z% B. `        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
# `5 Q$ C$ [9 N* p/ F4 C0 j                LPCTSTR pszInArgString, CString& strResult);
: f" M3 N7 s4 z7 Y# u4 Z$ h; H        void        StopUPnPService();


9 b* B7 x- r* J1 H0 ?! O( ]        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
: q+ M, U0 }' Y+ t4 ], ]        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
. S* I) I, y  n        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
4 |. W% f3 _. X, p% q        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 n6 ]3 K' v3 T8 {! k# N        CString        GetLocalRoutableIP(ServicePointer pService);
, |6 d' T: A' G$ A/ J
( e9 V) |+ z! t9 U# l
( K; m( d: |- T2 m$ _: e; t// Private members
private:
* H3 C7 O4 h6 {0 \0 @/ M: y        DWORD        m_tLastEvent;        // When the last event was received?
) S5 S  m: T$ x( {9 d% p* t; n! u        std::vector< DevicePointer >  m_pDevices;
. b) A8 \3 D9 _) R        std::vector< ServicePointer > m_pServices;
# c% J1 ]. @" `        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
- Z4 j' [' ]1 H
* O; ~: Y. w! T2 M  W
5 G9 G  s  B# Y% O7 O- ~        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
$ g9 E% n3 m+ i% C( c/ C        bool        m_bPortIsFree;
9 D! m0 b! h4 N0 D        CString m_sLocalIP;
        CString m_sExternalIP;
8 H: t. |5 a& G        bool        m_bADSL;                // Is the device ADSL?
4 O" @2 S6 f  H9 G. e        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
% d: c" ?+ W& I; K) C3 a& z- K        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
! H6 B8 O" `3 n0 P8 E: Q        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
7 r3 L2 |" o* V- h2 [$ {/ d        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};


+ \: m, ^. I$ C3 w; b& C// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
1 F5 r  Q; r2 ^, V! N3 ^public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
) m8 N% F& Q( \0 S7 y        { m_lRefCount = 0; }
( f! Y* o. b% I3 X# E# Z" a
, w8 R' e! o1 g: y- l+ s& H. T
. J3 N( }0 c+ ~, M! |7 j' R6 C   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
, T4 Q  i8 h0 p, ^   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
1 b$ N( i, T% a* |# F" \
0 E5 h7 s/ k9 J7 n& X. P
# A* D6 i) w3 G: i; F& l// implementation
private:
1 G& s' G, E) W: a$ H: @' S2 |# a        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
. {- O! t: c& {; t  _1 m5 {        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
% R+ [( I1 F( Y: b5 f+ u+ X1 e        HRESULT __stdcall SearchComplete(LONG nFindData);
( n/ p, d$ q7 [5 O( R7 v6 r$ ^

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
. h; j! L+ U, e: n- x+ W

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
% _0 U+ ~6 o$ o- S5 J* Y& m1 r                : m_instance( instance )
9 t# h) H/ U8 S5 Z! M4 ^        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- F6 ?  a; v: ?& z& p: [   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) h5 n1 H  ]! H# c

// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
2 b: F3 b. e/ j  w3 W        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
8 S8 }; r( e5 D/ N/ I' {( T};
2 O. C* w; q0 Y5 @3 b$ e3 R& s# Q
2 Y3 i" m# _6 o
5 g0 x. X3 @8 D  e/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( t+ k* d1 M6 S. \& SCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
2 k+ S7 u2 Y1 Z7 J8 {CUPnPImpl::DeletePorts删除端口映射.