UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   + G1 }* V0 e7 ]9 g/ ^2 F
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   4 h7 r6 @! |& x0 y. b; N+ ]
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    % F( `  E3 |5 W. M
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
      n  z3 `& D( c+ N( L4 ]9 A; e
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
    % i, n7 s' N( e+ d9 N! r- J
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    " U1 b, s: U* B7 I9 D+ u
18. 
      _6 x( |+ o% Z# ]
19. typedef   enum{
    ; ?0 ?0 [9 x+ B/ t, K( y7 r
20. UNAT_TCP, //   TCP   Protocol
    * _, X4 q8 c: W. _" q3 i6 t
21. UNAT_UDP //   UDP   Protocol
    ; v4 ^2 q, {2 X+ L! J5 I2 ]
22. }   UPNPNAT_PROTOCOL;
    . B, S9 N. J  @! i, w2 b
23. 
    
24. typedef   struct{
    , f  t" y! v) l4 s) f2 M
25. WORD   internalPort; //   Port   mapping   internal   port
    ) r5 X7 A5 @% T1 o9 u2 Z* P1 y
26. WORD   externalPort; //   Port   mapping   external   port
      a% F! w* e' e2 j
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
      c9 k7 y  N' I! S& A
29. }   UPNPNAT_MAPPING;
    5 e% N# D- p( R
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    , H- _) Y( X: Y; [0 Q1 \  _4 B
33. 
    % }/ h. ^. O) `  ]/ W
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    - V- V9 V& P) w9 z9 b
36. void   clearNATPortMapping();
    # r) I) V) [1 C  D% {$ \* z+ ^
37. 
    ( i6 Z4 T8 j& P' [
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    & z# `( A3 S4 B" ?
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    ( G8 `5 d. a: O" l& t
42. 
      E$ \; ~* w) c8 Y6 a" a9 d. S
43. protected:
    
44. void InitLocalIP();
    % B! q. B5 q, s  m
45. void SetLastError(CString   error);
      w5 O$ f8 ^8 L/ V7 B3 K2 ]: t
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    " v7 i1 k1 c0 L* ?: P5 i
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    . H/ r2 U  E7 G: |" [& k9 y
52. 
    . y9 E3 n+ }6 K. y  }) Y  [% L
53. bool Search(int   version=1);
    ' C- I8 t- F8 m: z$ D
54. bool GetDescription();
    * F5 i' I* @7 ]. \
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    " [1 G* i8 k* h! t: k
57. 
    " p$ M2 e) E' O6 j' f
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    4 R6 D; E- L4 ~9 O1 \
60. CString m_devicename;
    7 B6 G2 t, V$ `0 s: p6 v. E$ q
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    # k' Y( W( X/ |/ f$ C) s
67. int m_version;
    6 I- @4 v3 e. _( \9 ~9 D' x
68. 
    
69. private:
    " }& w8 L+ W' ?# x
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    # Y4 e- a4 A. `; d' t; n
73. CString m_slastError;
    ; I7 L0 t; _0 f; o: x
74. WORD m_uLocalIP;
    5 V% J& S# |8 \- y5 e$ j# ?
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   , k" B% o; ^  ^1 j" N2 Y( @- J8 C1 B
4. #include   "upnp.h "
   
5. 
   0 {$ t+ Y  m& r0 X& k  t
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ; n4 C- S. g6 R) a/ k3 Y" G
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ; n2 F2 A( M; X0 o& H6 R
11. 
    ( h8 R# [; ^5 I6 l, i& J3 f9 _
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    1 v& z1 b1 l2 }
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    , s- T# X- s* O
16. const   CString   getString(int   i)
    ) N6 D) Z1 ?8 r8 O# B3 T+ I* r
17. {
    $ M5 r9 r. M6 M0 S$ i' t/ j
18. CString   s;
    
19. 
    1 }8 S/ ^- u5 f* ]* v/ a
20. s.Format(_T( "%d "),   i);
    
21. 
    # F( J- K- [3 M% a0 c  w
22. return   s;
    " T* [( o! C+ r3 s5 a( a
23. }
    
24. 
    ( Z4 O* \7 U* |- m% g( Z$ d; d3 p' e
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ; }  u% J! x0 m/ Z6 `0 C
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    % `; K0 h# i$ \  Y9 x
28. }
    1 A9 V' D7 }% W, r
29. 
    / s9 W6 ^& E' r# ^8 u* [+ z& d2 c: h- `
30. const   CString   GetArgString(const   CString&   name,   int   value)
      G9 U1 h1 V; P# e0 s6 q& a
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    4 A4 `4 J, V' Y
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    & L& b5 s$ w4 z/ H1 h
36. {
    ( U8 H! l# P( R/ U9 K
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    7 J+ H% A/ H6 G+ x, [/ I; G) a' n, X
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    & z9 M% x0 U1 Z) W
43. uint32   ip   =   inet_addr(CStringA(addr));
    - ]' R/ \$ ~) e3 ~% c# b
44. struct   sockaddr_in   sockaddr;
      `0 S# V- G- K4 l" B; M8 ~
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    - h1 a; @7 `' r
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    5 t. P. d* J: C* m& R7 K
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    " Q1 P; i: e  z' U/ A! _, X8 w1 K6 n
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( A2 U  _! B  M5 E) l
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    : A7 c- P" M9 k2 H3 R( t' H
55. Sleep(100);
    % `/ m0 k  W! V8 _
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ' o; J; D; B/ h0 @
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    7 L. Z- I3 Y. P# v9 ]- x: J  A
62. 
    : y/ D: D% u$ g  ~; ?. \
63. return   true;
    0 s6 ]" L! @% u8 @$ g
64. }
    0 w/ R9 F. |! A, t! \
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    9 h' d( u5 D9 a* p, G
67. {
    5 ]' G6 w6 e% t& `6 J4 K
68. char   buffer[10240];
    
69. 
    7 @1 G) \' E; b/ T7 b: Z9 P. r
70. const   CStringA   sa(request);
    - f& I8 s; A1 M2 i: V
71. int   length   =   sa.GetLength();
    / P. |5 Y6 B6 _; ?. U, |
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    2 A( m8 [$ y: ^
74. struct   sockaddr_in   sockaddr;
    4 I3 k0 L- Q3 ]5 f
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    - }9 K$ V/ D# B  _. `
76. sockaddr.sin_family   =   AF_INET;
    4 g( u. ]0 ^7 u; l% t
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));
    % k1 Z9 `; Y9 \  }- G) v# u
81. }
    - `! [  L/ G3 f+ @: g
82. 
    1 o* J' ]" W  W( h+ b3 u9 X/ Q* {
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / @& [5 x" T7 i+ Q
84. {
    
85. int   pos   =   0;
    
86. 
    & s. [4 {3 I/ w% P
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    1 q% [7 W/ [1 K6 Z8 ?$ l3 e
89. result   =   response;
    ) C$ R2 t- F2 E; n
90. result.Delete(0,   pos);
    - t- H" o. b) D# N) W3 G  H
91. 
    
92. pos   =   0;
    1 y+ s) R& b& o" H
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      Z1 a; s* \9 A6 _, o
96. return   true;
    ! J( n1 E9 G& `" `$ G$ t
97. }
    . a, @; ^3 l* N, R8 y- z
98. 
    0 h6 W2 L4 q9 v( d" k
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    + ]. f3 V* T4 d  ?6 n; X/ `$ e6 P. ^
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     3 I2 d* z+ Z5 M( d8 |4 y- G7 F& U1 F
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     0 i8 k, y- {. h7 `
105. int   posStart   =   all.Find(startTag);
     3 W0 R& ?+ t1 Q/ c) ]" G& _" |
106. if   (posStart <0)   return   CString();
     1 o9 _6 S5 \  ]* U2 k& N
107. 
     # {9 A9 L2 n4 H) y8 y/ i4 @
108. int   posEnd   =   all.Find(endTag,   posStart);
     4 @7 C/ ]  X$ z; n: G- T9 L8 r- i) V
109. if   (posStart> =posEnd)   return   CString();
     ' B- s( X2 K% j/ e
110. 
       ^1 ?7 c5 t2 {
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     & r0 W1 D; A% K7 S& R: L+ @* _, A
112. }
     
113. 
     $ _8 P  L0 }/ s. a6 T
114. MyUPnP::MyUPnP()
     & p' L+ a5 l6 w" K
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     8 d/ i+ R5 {1 q. V1 B) l2 Z! H
118. isSearched   =   false;
     ; I6 \9 I7 w  S) [- @2 S
119. }
     8 e" L% ~) c1 k% S0 O
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     8 P7 o( C' d1 g% n, X7 `% V
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     : h1 P. n+ |. q* A
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     ' X9 {* |8 C5 T3 |% _
129. 
     % v! [# V* l3 N8 y  f, D5 P
130. m_Mappings.RemoveAll();
     $ x0 i+ U$ u$ d! ~
131. }
     + q5 h; Y; O- ]
132. 
     - F3 F5 L/ Z$ m9 o- p
133. 
     & s: Z) r5 t3 ^. N5 l# j# k. k
134. bool   MyUPnP::InternalSearch(int   version)
     9 c" V% e  W) T, c* U
135. {
     ! z' V' E9 l: i, S0 m9 e; F
136. if(version <=0)version   =   1;
     " n+ L5 |) `+ f# o  L
137. m_version   =   version;
     
138. 
     $ n$ A9 {! e+ q
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     " i: ]- g. G; F( d4 s
141. {UPNPPORTMAP1,   _T( "service ")},
     2 G% D8 q/ l& ^
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     1 x6 @0 D5 _* k+ X: X1 [
145. 
     - ?( e: u2 q  N3 S1 L' P; G6 b' y
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     . K- l1 m0 s; v, m' H3 Y
149. 
     0 l( }5 Y" n, `- B/ D
150. int   rlen   =   0;
     ) N7 [; X% M8 E! V1 u: G
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     $ J* ]# g1 ^  C+ I. @$ ?# c
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);
     
155. CString   request;
     / T) a& [/ u& i( h
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);
     1 L! o, G! b+ z/ ], j
159. }
     1 e1 J- |; r9 l
160. }
     
161. 
     
162. Sleep(10);
     4 x7 ~& s1 }2 p( [
163. 
     $ j: q# ]2 c1 f4 g3 M( \( {( K
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     # p- L/ }/ \: G  k2 ]
167. closesocket(s);
     
168. 
     ) ]+ [: ^! a7 f$ p" p2 l
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
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);
     " t: F1 {1 b# S+ \/ Q+ x
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     - s4 Z1 U) _2 P# l9 j
179. CString   name   =   line.Mid(0,   9);
     ! g' {$ V0 T$ x% c: o0 k
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     " A' N" p! k% f* B7 o
186. }
     ! {- M6 j1 p" w0 D7 |3 q  p/ Q" J+ Q
187. }
     5 A" W4 w$ K1 Q. j( l) w6 B
188. }
     
189. }
     " a# B3 s# {. x: L
190. }
     
191. closesocket(s);
     
192. 
     9 h6 w1 ^* h5 J& p8 j
193. return   false;
     ! B# V. K  _  T
194. }
     0 U) z. l( h  N7 N: O: W' c6 Y

复制代码

天云上人

以下有关upnp的接口来自emule，
: z3 a, l9 Q( v" ?) U

, _' B( @7 x- I# ]; h! [& \///////////////////////////////////////////
  i7 w% U: B6 \, O. j//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
. ~3 o) r6 G7 a  M& V2 u
" w+ }4 @6 a6 I" B
#pragma once
0 `: h  u5 _8 H- k#include <exception>
" }3 L5 M) X* Z  L7 k
4 R5 o) M* K' W9 A
: p6 _. X6 z6 B7 f6 ]5 W/ w  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
  |5 B" v! G- N% b+ L        TRIS_TRUE
& G8 `/ w) _, Y) k0 f};

( T$ K0 N* z. R$ B
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
; _$ ?  e: L* S1 T$ D7 j, M        UPNP_IMPL_MINIUPNPLIB,
/ P! W; F/ I9 g        UPNP_IMPL_NONE /*last*/
$ u8 x4 v0 X- \2 S* s};
: M: w9 U. j# @0 ~: u

2 {* S5 U" U0 P* k7 T9 F

class CUPnPImpl
{
public:
" z1 H2 {7 u- |$ I; Y, A0 C        CUPnPImpl();
        virtual ~CUPnPImpl();
& h6 ^: B- I1 U, f9 j: I3 E8 s, Y        struct UPnPError : std::exception {};
# W  |. A' ]% u" u$ `- {        enum {
                UPNP_OK,
$ Q4 ]! o, S' g& Q  S7 Y                UPNP_FAILED,
                UPNP_TIMEOUT
        };
9 h; ?: Z8 ?5 |- v) _6 J

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
1 o1 U$ G5 u% O6 \* s0 f& H- S        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
/ J2 L. k) @( [/ T        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
8 q6 z" D  p2 E3 i; z        virtual int                GetImplementationID() = 0;
: ?) M4 }& \( E       
( Q$ F5 e- R! }0 i$ ]6 K% S7 o        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

1 Y8 N5 W: G  i
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
/ U' Z0 O& U/ z        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
1 j3 p: q6 |; {, q& X        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
9 M' H( j; P& b* u5 g7 V6 ?. C; V

  ?% C7 ~, {; b1 Q# j" ~  W// Implementation
& j. R5 T/ D9 A" @4 K5 p5 d' Yprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
; q1 F5 R3 u2 k+ P, q6 `9 o  }8 s        void                                SendResultMessage();
- Q7 p9 Z) ?! I& o5 h6 t0 B        uint16                                m_nUDPPort;
; }8 w& ?: Y- o5 A# g) j        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
4 I4 Y: O6 U* v: A% M        bool                                m_bCheckAndRefresh;
5 x3 K6 m- w& o! B4 `0 y- W. I2 T0 r7 |

private:
; i. Q# m# L1 G: V' E        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
! P$ e+ S; ]  C0 {5 b
* `# V# }/ M+ |! w" }  F- z
};
. H4 s7 j1 }% }% f( [" W

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
# F4 \( ]+ }" U6 m0 \. M{
public:
6 x! L3 e% p7 n/ `        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
0 b1 V4 r2 e" ?' p, g2 R9 w; U1 O        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
0 f% x  Y# E3 O6 j: u8 S$ y/ @        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
! p9 }0 S6 Z# r  \3 I" C};

  T9 B% w! \& b: u. q( l# j
# A  ?+ v) S( j$ ]/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )
( w# j# b6 Z4 W1 L! }7 M0 z4 i
9 t' ~' B# L% n9 ~2 D$ r4 L
; u4 ]5 V6 u, U# R/ b: r1 ]4 f#include "UPnPImpl.h"
5 z2 l2 h) Z1 @3 p+ C) @#include <upnp.h>
1 u  Q/ P, Z6 l0 l- {) d+ ?5 ?#include <iphlpapi.h>
  O2 g( H, l  K' _7 w( ?#include <comdef.h>
#include <winsvc.h>
) X  {# F% _% V# H. n9 L
9 e3 r$ v, s2 ^7 Y$ Y& r4 Z, a
#include <vector>
#include <exception>
( s4 |* p! S) v4 Z* w#include <functional>

" c1 Z6 j( `% ^# g1 ?3 ~


+ c7 A9 P+ [+ T1 O" Ptypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
, Z& I2 d  T) X8 R5 u! ?. wtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
' {8 k% ~2 Z/ i8 Jtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
, B, B' N( d% k% H/ Dtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
) j! Z9 b; ~8 x/ r# {+ g4 _9 ?typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
8 m( {( _3 ~' z. ~2 I! Z. L3 {- qtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

' D6 l+ y" P& l; |7 F
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
; U- @4 `# a9 U) m);

+ d* f# s, F% t0 ~
typedef DWORD (WINAPI* TGetIpAddrTable) (
* Z/ E8 N  b" E" A. K( Z  PMIB_IPADDRTABLE pIpAddrTable,
+ w$ U0 ^3 X" b4 A- T  PULONG pdwSize,
  BOOL bOrder
6 l" O+ o9 k( l' d);
8 }6 t/ Q7 U8 H. _- o' P! y

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


" l0 w; T; u1 F: F2 A" X8 e5 KCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
$ |  K/ W" {8 u9 n
- z, p  b3 w# ], O+ v* y
class CUPnPImplWinServ: public CUPnPImpl
{
% e/ R6 G1 g, ~; ~8 F7 e        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
' v4 j$ F5 }/ C6 H4 j// Construction
1 `0 q+ V' L8 cpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
" K  J7 d% @+ U& H% L6 `3 A; j  ^

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
' \  R1 Z8 c# S9 a        virtual bool        IsReady();
7 ^3 P& `; C6 x. Q4 o6 U# q        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


3 a9 L; R" k1 G: [4 E+ w        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
) |7 H, d- ~  u9 U        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
& u& H+ p: k1 k2 a- u# p  M        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
" D( H* ^+ c% \5 @8 z* |! o& ?- n        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
, {0 ~# W  ~1 Z; H4 q2 ^        void        Init();
' |- b/ C# Y1 `3 p, }7 X- v

9 y# o1 g! C3 i6 i        inline bool IsAsyncFindRunning()
( J4 t" p2 W, G        {
( H# E3 f6 ]. r8 n                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
8 i( R- x' l; f7 |                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
% ]1 S2 u2 o7 {* G                }
. ~# }) Y  f3 U, J                MSG msg;
5 Z) ^. c: K4 c" N1 N                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
" J& d  i5 D( S! `" z                        DispatchMessage( &msg );
; K- D3 A! O) ]' h                }
' i3 K: S0 G7 a$ j$ u                return m_bAsyncFindRunning;
5 L: h, M: P# Q2 K3 n( c        }

# Z: z/ H/ S  G  }! S" Z! X
        TRISTATE                        m_bUPnPDeviceConnected;

8 ?) v2 c& t3 o4 e. c# d2 m, h
// Implementation
' L4 w8 n7 T$ V" f        // API functions
" o2 L  f) u$ W/ l/ n6 g        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
' @: f' P5 ~7 X/ D. ?        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
: {/ N  n/ l6 ^- d4 G$ L1 c: `        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
, R2 v$ k2 x" u) q        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

" y% D) \5 X& v7 d' f' K
9 C5 D* D# ]& E        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
% S# f  a$ u8 [, _        TGetIfEntry                                m_pfGetIfEntry;
, m* D( k& |0 t* [" {3 L( E# S* q
4 c, S* o( I( H( M9 \- u
" h5 p& l8 }8 m) n' H        static FinderPointer CreateFinderInstance();
1 S# W. l' s( b; [        struct FindDevice : std::unary_function< DevicePointer, bool >
4 z* R% \8 O% G1 Q: z- i! D5 ]        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
3 e( y" |  R7 K  a                result_type operator()(argument_type device) const
7 z, g2 R$ w: T7 s( y8 i  q                {
                        CComBSTR deviceName;
; m1 H  U: R0 h1 D                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
  A/ z' r5 ?  @% v2 o1 E& ?% l' O

& S+ e; d4 f& l                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

, }# \+ r, d$ |6 X7 y% g$ u
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
- G: d: n; Y3 {2 f                }
8 ]- t5 Z+ ?$ v+ F" V  j6 O                CComBSTR m_udn;
        };
* Q- c3 V/ m# a% N4 U       
' _% Z# \( z, B7 H' h1 T        void        ProcessAsyncFind(CComBSTR bsSearchType);
' s1 |8 F4 S7 A% s        HRESULT        GetDeviceServices(DevicePointer pDevice);
/ @; R! n3 u5 w( i. C        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
. y" f, N% t, D! Y$ v) y3 o5 J# B! f        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
1 o( p/ V7 Y* p& w+ ?        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
: {/ e! r2 F( o# M+ R) b% L; G                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
" G0 p3 E& M# x' ~, L

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
, _$ p8 b( Z5 f1 a9 W        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
; ?) z) Q  m" @7 |- n        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
- v/ f: w% f! s/ f% s. L        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
9 h' g# w3 g; C1 \        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
$ H+ y7 E# H$ t( o8 }8 |: W        CString        GetLocalRoutableIP(ServicePointer pService);
6 K( l# ]0 @, x5 I8 u

" w, z7 g0 `4 [% t// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
8 @+ ]6 n9 X% T) l. R0 s        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
$ E# b4 |+ I- A2 A3 @2 m0 c        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
4 r$ w6 A4 e) \/ C$ @  T2 v: _

# L8 W$ e  C) K( l& C9 R        LONG        m_nAsyncFindHandle;
9 G! P9 I, B# E" o$ X        bool        m_bCOM;
* A: M! h' e& ?        bool        m_bPortIsFree;
( d% E9 t  s# j5 E% C        CString m_sLocalIP;
        CString m_sExternalIP;
; X1 o0 y( y$ C        bool        m_bADSL;                // Is the device ADSL?
3 P0 c! u: M3 i' w' ?7 D( q0 D, c        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
' e) r$ p' `: n0 C- Y- u        bool        m_bInited;
        bool        m_bAsyncFindRunning;
& j/ }4 N) R4 S, g& P: b/ [# t1 c/ H        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
# d/ F8 n6 T' c# i) {+ z        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
' i6 m# d- |- b: Z        bool        m_bDisableWANPPPSetup;
. f1 c# w, n2 I4 ~% e  Y8 ^
& M0 Q& @0 H8 Q8 K; o/ x( o$ r
' D1 c+ p! Q) @9 g. J};
8 }4 A3 @7 M, w4 s# U# y- N
/ S; p7 C8 f4 w9 _
+ c" J+ B3 `: H6 t// DeviceFinder Callback
9 b7 T2 T# K2 f. Yclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
0 e) {- w  q' t# q5 D2 p! z0 q+ J) Qpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
) |& \/ V! f2 h3 \        { m_lRefCount = 0; }
" \  R, [6 W2 y4 B
; d7 R+ D: T9 P* ?& o  w
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! R7 E8 |/ y7 `& ~   STDMETHODIMP_(ULONG) AddRef();
* Z' G8 `9 ^8 k" J+ R3 b, t7 O- v0 k   STDMETHODIMP_(ULONG) Release();
0 B: s- P; D6 r" I3 A  J5 k
! Z0 \1 Z7 I0 W; n0 P; P* H
( n. Z; b9 D" z8 U4 V- R& A4 \; R// implementation
private:
5 x$ Y  d7 v, O  X; a, p3 H/ b: I        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
; R; w& s) Z* B5 y        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
2 h% ?/ ]5 j- }& A( v! D! c0 \+ X- P        HRESULT __stdcall SearchComplete(LONG nFindData);
, f6 g# B& }6 Y
( V, P* d/ [6 }' ^; z! i
  n& n; w& I  ]7 ~7 Yprivate:
        CUPnPImplWinServ& m_instance;
/ x" e: x1 e+ M, O1 M5 K        LONG m_lRefCount;
};


// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
) {' W, S7 _9 m: T$ X1 \{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
- Z9 ~4 K8 j+ l4 V. D/ E                : m_instance( instance )
9 Z4 y: k3 ?6 Z( A6 z& H2 c        { m_lRefCount = 0; }
   
2 i9 [7 g7 T+ y% S: V' `   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ u# N8 H6 v1 X" ^- x& B& O   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 [3 m& l3 T+ S& ~

// implementation
private:
4 O, p; {& e5 U  ~1 ?8 A% Y        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
5 D6 Z0 o# V% e6 }0 Q  c5 g5 v        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


1 C5 A7 a/ M& u9 R/ Oprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
+ q1 u1 z# }' i" }6 z
/ f! V* P6 H7 ]
' g0 J+ ^* ^, R8 W- [! e/////////////////////////////////////////////////
2 @6 e* ^- g2 O0 C) o' U
  B4 Q4 z( h# n- {+ D1 C
8 c: F6 A6 A" z8 Z3 ^0 k使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.