UPnP

天云上人

/*uPnP.h*/

1. 
   ' I+ C0 k; b2 G
2. #ifndef   MYUPNP_H_
   " f+ @$ v$ S  p" s( _9 s8 i
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   5 W9 C9 M2 X$ d9 ?9 n" j; G
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ! E1 ~$ g" e4 c, ]& ?5 K
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
    3 t7 D" v1 s1 H+ R
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    8 N! O+ e2 T, s* }% E8 o7 y  e, C
17. }   UPNPNAT_RETURN;
    ) y; ?: l: u( D' s  |
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    7 c* }2 L# M! E( @2 v7 k
22. }   UPNPNAT_PROTOCOL;
    9 ^, b$ b. p, B6 X4 {# R% K
23. 
    6 L9 K$ G! D4 p' Q2 D, o
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)
    0 P( H$ C' y( Y% S
28. CString   description; //   Port   mapping   description
    6 c! _: T  }5 t5 K& N
29. }   UPNPNAT_MAPPING;
    3 a% C* V6 @( E) d
30. 
    
31. MyUPnP();
    * w2 f/ U+ k+ g$ _1 F2 D
32. ~MyUPnP();
    0 A: s9 R. Z% D4 E- Z+ r5 [
33. 
    1 u" ?3 {# V: k' z
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    9 O' s; B1 `" x! R8 b, t
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    3 y8 s) A( P( c: O8 u# c% S
44. void InitLocalIP();
    9 c5 Q7 C( W* \7 h1 h# ?8 r% j" N# t
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
      n) r, t" @9 a" X' K
48.       const   CString&   descri,   const   CString&   type);
    , E: B  }6 u3 W) j$ j
49. bool   deletePortmap(int   eport,   const   CString&   type);
    4 s. _! X2 e8 M
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    & _5 d! b  Y+ j. w
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    # R7 k. s9 D' @( ~
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    / I9 ]3 v1 z/ f
59. bool InternalSearch(int   version);
    : d6 I% f+ c2 J" A
60. CString m_devicename;
      p) J5 Z' ?1 ?, [8 J
61. CString m_name;
    : X# h6 B. Q0 f  @
62. CString m_description;
    
63. CString m_baseurl;
    ; X0 M) p# E" W8 Q7 x2 V; z
64. CString m_controlurl;
    
65. CString m_friendlyname;
    , J3 F+ s/ x7 B( D7 U# R$ d
66. CString m_modelname;
    
67. int m_version;
    8 q  z& k4 [4 B) K
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ( T4 ^1 X) l7 z6 J
71. 
    - C; `4 G$ I8 c
72. CString m_slocalIP;
    8 N1 n5 f, N8 \9 @# g' n
73. CString m_slastError;
    ) _. h0 E: J' {8 p7 ^) t; [0 }! {
74. WORD m_uLocalIP;
    . U) W' q8 G- v. u& j* N) }' j/ I
75. 
    6 {6 i, R( x/ I7 m3 c
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   / a- k  B; t3 v0 @9 P, }2 e
2. #include   "stdafx.h "
   
3. 
   & Z: T, y! k" R/ O; u2 w
4. #include   "upnp.h "
   
5. 
   5 G  n% Z9 a( h
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   & a4 \3 X6 R0 D1 D" {7 \
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ Y, a% M% x  Y
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ! j0 i' |3 U0 C8 J
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    9 j: |# r6 Y! i
13. static   const   int UPNPPORT   =   1900;
    % S( }: ?: M5 i1 M& a4 J7 r4 {
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    7 H/ A5 Z( ]! s0 z) p, s1 h
15. 
    / d, _. G- ~! l* T: n: z' E0 I
16. const   CString   getString(int   i)
    
17. {
    $ S2 E0 m& y$ r+ l# e: }
18. CString   s;
    
19. 
      K% ]  Y0 o+ X3 O* i0 e
20. s.Format(_T( "%d "),   i);
    
21. 
    ( k8 ~8 q8 V+ m+ C# G  K
22. return   s;
    
23. }
    , l- I! p/ g; s8 H* t( r% W' w
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    1 \, O5 X3 ]1 z: l" j
26. {
    ! b5 m3 e: n. W: H6 f$ c( b+ l
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    4 C9 U7 j% i9 a5 N- g
29. 
    5 I# I2 |! o+ z6 {2 |
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ' |9 u$ l0 g: j$ p6 P$ Z  J+ L7 b
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    # o; L7 w7 |) v, z2 t9 K$ f. A
33. }
    
34. 
    ; n1 @: ^3 O3 z& F
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    : W! y, X1 Z5 @3 B
37. char   buffer[10240];
    , L* G8 H5 U0 U. A4 Y8 J6 `+ l( q3 t$ W
38. 
    
39. const   CStringA   sa(request);
    ' N. `7 e6 E! ^7 F* j* u2 x3 C- `$ b: C( C
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ! z3 h; ]1 l+ D, A' K: E
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    6 G" U3 W4 Z7 [- S0 Q. X& f
44. struct   sockaddr_in   sockaddr;
    - j* F: _$ ~/ G! B
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    5 }) n0 M9 e/ N
46. sockaddr.sin_family   =   AF_INET;
    3 h& @4 i, i4 }
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + K7 H* |6 Y% Y: Q3 p8 _! u; J
50. u_long   lv   =   1;
    5 W* T/ D5 \6 y) K
51. ioctlsocket(s,   FIONBIO,   &lv);
    ; n* F: s; f: B; B0 a
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    0 Y4 i: i$ C) _
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
      \. p6 k7 }5 J7 u6 z! M
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    3 W+ d+ n1 K8 h9 v' q7 @
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ) h; R' h6 Z, U& Z: ~
59. if   (!rlen)   return   false;
    7 O, d2 _# O- m# x+ ^. m
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    5 u+ C' J* O" g4 _
63. return   true;
    
64. }
    
65. 
    1 Y) D) L: q" u5 Q& ?
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    * |" {7 Y8 Q7 I+ o
67. {
    , t0 O. O# Z9 ^( V, B: k) g! T
68. char   buffer[10240];
    : \5 E$ L. k4 H! e/ E4 Z7 k- k
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    * A  {% L9 G+ f" Q% ]
73. 
    
74. struct   sockaddr_in   sockaddr;
    ( A' c6 k) E3 l  h( k, r' D
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    1 ^" X# K6 V+ y. t7 U* P
76. sockaddr.sin_family   =   AF_INET;
      Q3 D1 R1 g# s/ t" d6 {7 |$ L
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    + H" L4 k6 l7 `
79. 
    & B8 M2 i3 Q! H& J% K
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    - B/ `9 l* `# m0 k4 n' d" |
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    / r: s$ l, u! s' Q- t6 ^. H+ `
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    7 T/ B8 a8 z& F, A$ q: _; w, I
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ' G$ E5 @; y3 q4 o: V. S7 y6 r+ y
96. return   true;
    . `/ a4 P! r* Y% V: M
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ! I& g7 L8 M6 v1 ^1 E# ~
100. {
     0 W! Z0 A$ v$ f3 k6 i* @0 v3 X  D
101. CString   startTag   =   ' < '   +   name   +   '> ';
     5 z0 o; R% I* o& `# N1 H1 z
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     " B1 p- r6 p* ?6 V* d
103. CString   property;
     ( w; H( _! E( V
104. 
     2 z' ?+ y& b3 B9 G2 S1 S
105. int   posStart   =   all.Find(startTag);
     , Q2 }# h5 `) ~# J6 v! s
106. if   (posStart <0)   return   CString();
     ' W1 l6 z3 C" j9 o- e' q
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     ) [. d. Q2 m$ N  s5 a4 u
110. 
     * G1 p$ l# ~. K% Q
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     $ [1 y" ^) T4 N* u
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     : O" f7 p. O: m1 F
118. isSearched   =   false;
     ' p/ {1 i% r$ p  }1 ^  j$ G
119. }
     3 x" z' M, s' o3 B( l
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     % o/ O/ ^. X$ f
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . c* c! e8 n: u$ e; j. N# v. x) ^
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ( l" f7 _( M/ c  R# E
127. RemoveNATPortMapping(search,   false);
     
128. }
     8 ]  `: n" h( x0 }0 T
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     0 `6 S5 W' D, E
133. 
     % ~  L% i1 N8 i* H% Z) s5 m) @) g
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     / [: P5 P5 s( [4 R0 w, z
136. if(version <=0)version   =   1;
     ' F, N5 G4 `1 o6 z& F, f
137. m_version   =   version;
     ' W% o$ }2 W7 V- X; T' ~7 m
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     ' ?1 a# B& a8 g* e1 }; H  n
142. {UPNPPORTMAP0,   _T( "service ")},
     ; C' M8 j5 k+ E4 D( B1 \2 j8 {: G
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     0 v1 p2 g7 Y# g( @( e2 s
145. 
     7 P0 ~/ f5 P: l6 ?" a
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     . p, Q! m1 `; n3 ~1 a9 t
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     9 B* o; `# h( q4 b) c# A
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     % m0 b  j4 O+ ~5 c9 L( [/ [
152. if   (!(i%100))   {
     1 c$ u- _5 l4 Z
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     6 p" W) S6 a' ]8 V: m
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     : X  ]1 {! K: {. W. p7 j5 V
155. CString   request;
     ( L& E1 F6 B6 h: n. a. L
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);
     8 B& {8 {9 \  X0 k
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     2 }: x& B' @+ P6 L
160. }
     + Z! m9 M# K: B! \1 ~
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     % F' ]8 I5 f! Q2 h9 g
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     2 s) P- S! g$ b  |$ }( m) Q
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     0 j/ F, N& |$ V, X0 L' P
172. 
     ( t& L" Q, T6 A& h
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     - ~" K4 o2 F1 y* b
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;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ) I+ h. F' f$ X/ T! ?5 k
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     $ a4 [/ X  W% }' l3 q& x
181. if   (name   ==   _T( "LOCATION: "))   {
     / s9 [% O7 ?( v  U0 ]8 d; A' |
182. line.Delete(0,   9);
     9 K5 J+ x2 a  K( C) \/ \
183. m_description   =   line;
     : r7 X0 q+ ^7 K) a. o; Q
184. m_description.Trim();
     
185. return   GetDescription();
     ) q0 Y$ m3 l5 `* y3 i, P
186. }
     8 U& Q) g7 n$ h7 h" E1 [
187. }
     
188. }
     
189. }
     4 ~: u) n8 C: H8 m2 @9 Q& x$ C
190. }
     / m' O/ e8 f# C% N" ^
191. closesocket(s);
     8 P* W0 h  q* ^" i6 Y3 e) P
192. 
     
193. return   false;
       Y- n/ x( ~. O
194. }
     

复制代码

天云上人

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


1 x8 ]$ e. H4 j$ q- c///////////////////////////////////////////
0 ?# y% _, ^8 [//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

9 r. q4 V( l. z$ `6 a% Y
0 k* B9 g! L" _* x9 S#pragma once
#include <exception>
. n* `: a( b( ~7 z) ~) n8 ]
7 l8 |. t0 ]  j7 j$ s$ e
2 o( C: m! s! m" _7 H4 P/ P" g6 X  enum TRISTATE{
) v" r- b) F7 n8 u$ y        TRIS_FALSE,
        TRIS_UNKNOWN,
+ G  {$ o" g/ ^+ z6 l, t$ e        TRIS_TRUE
8 B3 V$ a- b  m5 G9 x" f* C9 U};
+ R4 u" j: Z& M7 ?

; b6 l- @5 b" m* L6 Y8 J0 e8 genum UPNP_IMPLEMENTATION{
# J6 O+ _) U: }" k4 m+ C' g        UPNP_IMPL_WINDOWSERVICE = 0,
" ~3 h! ~& f  B; c" ]- {: Y        UPNP_IMPL_MINIUPNPLIB,
* H$ J' R) w! U7 T# x; |        UPNP_IMPL_NONE /*last*/
};
7 x7 H( f" N' Z! N" J


! h2 W* p9 V* D: s
class CUPnPImpl
" g% G2 |% j8 X7 l& I0 R4 h{
public:
% r* {1 h4 [# ]  F, @7 C  ^' E4 K        CUPnPImpl();
0 J% P4 g3 w" `0 _9 F" Z; S5 H        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
" ?+ C6 G, a3 {. d0 F: _8 t                UPNP_FAILED,
                UPNP_TIMEOUT
        };
. v$ e# x/ j. E& m8 n: L

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
. M1 S9 g- ]" j# s+ E# e        virtual void        DeletePorts() = 0;
+ F5 }6 g6 N! D+ I* V& W0 B8 P5 K        virtual bool        IsReady() = 0;
( m" r2 i! U8 n! s% q$ _" H        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

7 Z% e8 m* H8 d$ P2 Y  G
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
1 ]/ `6 U0 ]# i( S! ~        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


2 U+ ~# {$ q5 |. K, H- d6 N6 V& O// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
8 a# A8 s4 |; {0 d4 X1 T, m# {        uint16                                m_nTCPWebPort;
: K+ H% U' v" T8 \2 P/ Z        bool                                m_bCheckAndRefresh;
- {: H; o* q2 C; i+ H
. [( ]5 t  c# e- Z  i  N. m2 \2 c7 C
private:
        HWND        m_hResultMessageWindow;
1 B, G/ r* L  H, G        UINT        m_nResultMessageID;
) N3 M0 S* x6 m6 \

};
4 w. i8 J" T; S
# v, \7 g0 P7 B
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
% o' K6 }6 D5 D- |{
0 ^$ ^# _3 x% Fpublic:
) q- u4 t8 ?  K$ T" j8 G- V        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
# @2 u. M3 a) }7 W5 s2 Q        virtual void        DeletePorts()                                                                                        { }
5 G3 {; r  k) ^; U- u; b        virtual bool        IsReady()                                                                                                { return false; }
2 x0 ]: Q, Q' q" V: K. r" i0 [        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
* R$ T& c, W- V' `

1 b# }/ |( l& @7 `  y/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
5 P6 c8 o; y! t; e: Y+ b' c
6 a  p2 R7 B. E8 a# p
#pragma once
#pragma warning( disable: 4355 )


8 T' n5 l. r9 p- r2 p% ^+ V#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
" g/ l9 w, [: h" {#include <winsvc.h>
7 V/ I) n" B, G) [  \% {
$ x; w* q0 M6 I8 P4 P6 d" ~
+ t* W& U! w1 k4 F! H) X: d- Y) i. m#include <vector>
#include <exception>
% x9 a; G- T$ r; r$ E' J6 a( {#include <functional>


: V: @; _* b* X" y4 ~
/ P$ W( W' }2 a. D3 [
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
* u8 P0 T9 t* _, G" f0 n  Gtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
) u# B9 X1 O8 Ttypedef _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;
! Z3 q% f2 d8 g) d; Y; Htypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
* r8 _0 D: g5 l/ N; X, u9 k
' g2 M. Y. j: H
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
: M9 a1 _0 z3 P0 _  PDWORD pdwBestIfIndex
# h$ y! [7 a2 }: b);

$ ?" E( C7 O& I+ a  p
, x1 p" K- a5 r' a' ?typedef DWORD (WINAPI* TGetIpAddrTable) (
& _' T2 }. J- c2 L/ ]' A  PMIB_IPADDRTABLE pIpAddrTable,
+ V5 _" @; G! c0 k6 o  Q  Z  PULONG pdwSize,
! f, s+ O0 p& R( v9 a# R4 M  BOOL bOrder
, Q9 V& H) v  G);
5 a; e0 m) A, g: L, T8 B2 {, |0 ^
* w8 P. ?. R$ L- G2 |
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
2 ^3 A/ t  \7 ^7 T8 }9 a/ n

2 E% p# q2 T1 w- zCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
* }  s' V" C7 C% y/ C

" S4 U; t! L0 }% {; @class CUPnPImplWinServ: public CUPnPImpl
{
! t+ I7 a, E. C8 u        friend class CDeviceFinderCallback;
5 o1 K! e0 [* {& X4 f; A- J* s        friend class CServiceCallback;
4 A- p/ `! l! m// Construction
  J4 G0 U; M! n3 M1 ~- _! n$ Fpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
3 d% x+ [0 g, ]1 N+ K
' C8 z4 ~* B- x, y) F1 y4 c
( K, _. ^; J" V3 T5 z" ]        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
+ _$ q5 @0 @( U- {) L( R  Q6 _9 |        virtual void        StopAsyncFind();
4 a. o$ H$ k1 j. o4 t        virtual void        DeletePorts();
$ S4 g$ ?1 U+ W, |4 G6 I) S6 I5 n6 X        virtual bool        IsReady();
0 ^6 B9 \9 A- D# z7 Y        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


7 ]0 A4 w5 A* H) x# C0 J" _        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
/ u2 C( ^: t0 Q* g        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

/ @7 ?+ j9 N% x9 c! i+ g% K# T" T
( {& d  }3 T8 @, F+ Kprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
) T# z. [* s* ]; q, _, ]5 x        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
" g8 H! |9 e) [) d( |9 t0 k        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
) ^2 `% X, m/ [7 {& i2 z                {
& y+ W; z5 n/ E& g7 [                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
3 W9 n  {% d1 L% l                        m_bAsyncFindRunning = false;
" E. }- K7 b% K7 j2 d* `1 |                }
                MSG msg;
% ~' N2 _9 `! P                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
/ V& T* S4 X+ b  ?# i                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
" M- L# }, Q1 \# H: E                return m_bAsyncFindRunning;
        }


        TRISTATE                        m_bUPnPDeviceConnected;


5 E4 [# v- E- Y" }+ ^/ k// Implementation
        // API functions
: e* _' ^) x/ N2 V0 }; ~        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
1 x" b9 }- z" `' r/ C        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! ?1 K3 `) b7 [3 M2 s# Q! @        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;
6 w5 ?# N  V  X4 E' b" R1 `7 ~; H        TGetIpAddrTable                        m_pfGetIpAddrTable;
* ^4 c0 ^1 ]- ~; R$ m. i, ?. J# y        TGetIfEntry                                m_pfGetIfEntry;

2 E4 c0 ~- H  ~& k$ G; ^
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
0 @: q# `5 Y4 L0 o        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
( I4 _- O5 _$ b3 w3 ^) X( b                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
% w- ^7 v, r. m$ N# G
0 B7 Z& p2 y3 F4 E* V
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

& j6 w. }( O+ t$ a5 n: P- \7 Q1 X0 K
) ^8 Z2 @# V( ]% {                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  B$ P+ s) p2 r% [                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
0 ]' u# _& a% `* O        void        DeleteExistingPortMappings(ServicePointer pService);
! A, K- R* m# [0 F        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
- U' }$ S0 \( i' l2 T+ {8 c                LPCTSTR pszInArgString, CString& strResult);
" ^$ D) X. q# P8 D- E        void        StopUPnPService();


- J4 E& @) v/ W$ E; p2 y  h        // Utility functions
+ D7 a3 S: t$ \        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);
0 ~/ L# q) w5 S2 z  S. c        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
7 _: ?+ q& c7 u4 U) p6 R        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
  g: K) a( e: r
5 [/ i2 G6 a9 I' E
// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
' ~# e7 y5 n& y" ]0 l/ B        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
+ O6 T: h' q- S& D

        LONG        m_nAsyncFindHandle;
; A+ Z4 A/ F% ^8 ^: s6 g/ j9 l        bool        m_bCOM;
3 _  K8 o# Z2 b; j; z5 m4 m9 S        bool        m_bPortIsFree;
) [. b5 _( P3 W( e! s# X# n- \        CString m_sLocalIP;
$ w% r2 A* t+ w6 N0 m6 d        CString m_sExternalIP;
, c1 f" j& e' g2 E- U        bool        m_bADSL;                // Is the device ADSL?
; {4 |' d7 w: ^1 }% R        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
$ e+ K- _. s  Q        bool        m_bInited;
        bool        m_bAsyncFindRunning;
2 z  ?9 V: n# \& D        HMODULE m_hADVAPI32_DLL;
" F5 y$ \5 O8 a" g6 V        HMODULE        m_hIPHLPAPI_DLL;
- |( z% K+ v1 @9 A        bool        m_bSecondTry;
& E1 V- g/ w7 |8 x' }: m, t" m        bool        m_bServiceStartedByEmule;
' E5 u' k8 D! m: o        bool        m_bDisableWANIPSetup;
) c; e9 c# u; F% |        bool        m_bDisableWANPPPSetup;


};
8 l8 |9 V" w3 ~  A6 t
* c) x" A. I2 I5 E/ `0 ]: }
// DeviceFinder Callback
+ A+ `% g; J5 ]  \2 G6 B/ Oclass CDeviceFinderCallback
3 `4 t: P' s* L' ?) l* c6 P        : public IUPnPDeviceFinderCallback
* D$ ^9 `4 s# W6 T& `2 {! g{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
, ]" A7 x) q$ [" B8 |! _                : m_instance( instance )
        { m_lRefCount = 0; }
( M8 r# t7 M6 j5 e

; s: D! p8 t) `8 \! n3 _$ ^: \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
+ i! f3 d6 P+ G% @7 A+ l4 t  a4 |   STDMETHODIMP_(ULONG) Release();
: G, C9 G! h0 s, G' T" ?. a
7 r9 L7 @1 Z; n1 W% s( W
8 d$ H2 O) t. n// implementation
private:
8 ^' E3 |; L% [        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
' t9 f: D9 e7 ~- k: t        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
/ L5 u! b) t( p3 a; c        HRESULT __stdcall SearchComplete(LONG nFindData);

1 ^# f; B# ?$ ~, \. \! G( Z) m8 g$ t
! b0 Q2 l8 H, H& g4 M& uprivate:
        CUPnPImplWinServ& m_instance;
3 `5 C, i' L; n, P( s        LONG m_lRefCount;
' e, ?. J& d5 C/ e1 ~* s};


// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
7 O* {0 |! N- @' E% _{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
- G2 F) K8 y3 @$ s" k/ o2 d& V   
7 M0 Z1 j/ U+ H: s" |+ s   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
" q- X0 Q0 R/ @+ Y9 h4 |) W' ?3 A  g   STDMETHODIMP_(ULONG) Release();
+ h  z- \- L4 B6 ]

// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
# R5 V, ^, c* ?" b. R        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
9 f5 E6 B# f/ q# n. B( I
% _! k5 ^( I  A% O  c
private:
3 I0 C4 P# n  w+ p: a: V' h        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
$ |! |& Q* \6 Y$ t0 v9 S% D};

$ t+ V! D, @/ A: V6 [/ L5 |) D
/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
, a/ D$ r! |: T" g' K* A, @  x# FCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
/ C1 Y  o1 d' aCUPnPImpl::StopAsyncFind停止设备查找.
' }/ l) I' F/ ~7 TCUPnPImpl::DeletePorts删除端口映射.