UPnP

天云上人

/*uPnP.h*/

1. 
   . u$ e. v1 D2 ~$ p' i$ c: I
2. #ifndef   MYUPNP_H_
   " [) ]" g/ \  u) Z
3. 
   
4. #pragma   once
   
5. 
   . z9 H# [3 T# k$ ?3 p
6. typedef   unsigned   long   ulong;
   7 w# Y- d0 }* I; X8 ?) n2 _
7. 
   
8. class   MyUPnP
   
9. {
   - _7 L/ w) N. L. t
10. public:
    . h3 W1 ?8 w7 w2 T" r
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    , ]" m$ @5 b( s
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    9 ]9 Y( p) c" S
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
    : t7 w9 }/ N1 l' r* F
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    7 ^8 ^( K) ~9 b% V, H4 p- w' Q$ t
17. }   UPNPNAT_RETURN;
    
18. 
    5 K% t8 V* E+ m! P5 P
19. typedef   enum{
    5 r: P5 K( m# f1 K+ V* E
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    - s. A* s. H: K; i% ~7 w
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)
    , @2 ]8 Q+ `' {) Q8 Y( y6 F  w
28. CString   description; //   Port   mapping   description
    ) J: H: ^$ U* \  e
29. }   UPNPNAT_MAPPING;
    - x; ?* b1 M# i# x. }
30. 
    
31. MyUPnP();
      p! E) v- p0 E6 a8 D0 O, u# m
32. ~MyUPnP();
    $ V" |, X9 I/ R& P( j7 x6 H! a% ^
33. 
    
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();
    - z2 t) m/ n0 |" c  I
39. CString GetLocalIPStr();
      m1 B6 p3 a$ |. w9 ?& T0 D$ n
40. WORD GetLocalIP();
    . F0 t( M: j+ n3 A
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    - t) U, c$ L/ l
44. void InitLocalIP();
    : Y$ Q0 z5 l: j# Z6 `- o, \
45. void SetLastError(CString   error);
    
46. 
    ( n$ U6 ]/ h  g& b* T
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    % A8 S8 q+ e2 \. w: Y  O  g
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    * X' Q7 u* c  Q. O6 I
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    5 x$ f# S( q* }2 R
52. 
    ' M/ b( D$ O, T: k# J/ f4 e' h. o
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. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    3 M, M, c$ ?+ w. ?$ M+ r0 I
60. CString m_devicename;
    8 S1 T3 L8 j: U& W
61. CString m_name;
    
62. CString m_description;
    0 J% Y5 l) f9 g
63. CString m_baseurl;
    * O7 G5 i. I# N6 |
64. CString m_controlurl;
    : T9 K- A& n( h9 ?! N: D3 m1 F
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    4 z, q  @# O/ X7 E+ [
71. 
    
72. CString m_slocalIP;
    / r5 e, q- B0 D
73. CString m_slastError;
    3 J0 e& i' V* G7 O
74. WORD m_uLocalIP;
    : n( g8 d0 C6 v6 k$ u8 H3 i
75. 
    / _4 V9 b' b" E5 t
76. bool isSearched;
    ) u- f# A) m8 f) B# X
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   , n: H9 i+ i, G) v+ ?& m5 J+ x! H
2. #include   "stdafx.h "
   % K: C4 h) o* R4 `) w
3. 
   
4. #include   "upnp.h "
   $ T7 ~  ~8 _( Z3 r# P
5. 
   6 K0 ?- }# J7 J& F! s
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   6 }) t$ a$ k3 t- s5 W
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ' N/ z) y0 T7 |1 P: q* f$ `3 J
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    6 ?& v+ h6 [  }" ^5 s
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    ' v% t  f9 g6 Q9 @  z# g+ r
20. s.Format(_T( "%d "),   i);
    
21. 
    & p. n7 ^/ V, f" C( O& b
22. return   s;
    
23. }
    6 \) ^4 S0 r: w5 f
24. 
    6 U" o! W+ X# Y; q2 L
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ( x; |3 A6 A# e$ E  W' `( s( `
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    8 z, D$ f9 B+ I" S
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 d3 L  \* m6 E/ K3 o- o6 e0 r
31. {
    & m) F  q  Y' B: Q
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    / ~; A3 `8 _% J$ Q
33. }
    1 q3 j! r3 D+ \' j
34. 
    . z8 U' g5 M3 W/ ]8 N3 |
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    0 @  W; K0 x/ N  S2 P
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    0 w$ n# o; j  f2 F2 a/ a% G
40. int   length   =   sa.GetLength();
    ) k" D( B2 N: c
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    - V+ J4 w. Z: ?& M
43. uint32   ip   =   inet_addr(CStringA(addr));
    0 t6 u. q( D9 |# W0 ~
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : l; m1 [# Z9 f8 A* K0 ~3 L
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    2 M9 p5 w) j4 X: h
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    2 O  D3 _" U& o" }9 Y
50. u_long   lv   =   1;
    ( i, C: Z0 c. d& a+ }% b
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    , q4 o; u3 ?4 L' y. b. {. \4 f2 x" x
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    + Q1 T- T5 n* B/ c  S& h2 D
68. char   buffer[10240];
    7 O* i0 p$ J: X4 Z6 q/ o
69. 
    / T" W: @. f7 j. [1 j1 w, Q
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ u) D. P) f, C" V# t
76. sockaddr.sin_family   =   AF_INET;
    ( A* L3 Z$ u& M1 U# B: a
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    9 ^' G3 G. @, H  q" b( L( s: }, g
79. 
    ) H8 H& V$ W, {. [9 P
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    ( ~+ S0 t9 \+ d7 Q+ y
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    $ Q9 u9 T$ ?! H$ \6 A
84. {
    
85. int   pos   =   0;
    
86. 
    ; H3 B# k3 z* e4 @
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    " p  g" D0 ?5 v
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    # l3 \2 q1 t/ Q4 s( {
91. 
    $ r& c$ ~) F( @% X! R
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    5 K+ T$ X% F( `/ E* M& s
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    3 Y! }  z5 R9 `( X1 r% s
96. return   true;
    / }% Z5 k) s: i; C6 \& E
97. }
    , Z: f$ k- b9 S% z
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    3 ^3 `% x/ _7 C9 ]/ Z; p
100. {
     % }0 D# w# [/ ]3 T
101. CString   startTag   =   ' < '   +   name   +   '> ';
     " U; F9 Y9 [! \2 m: e0 |
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
       W7 ~( f5 l9 ]  |& N& x
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     # Q; p# X9 K" I
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     / l6 M, s4 P2 F5 F
109. if   (posStart> =posEnd)   return   CString();
     & Z/ o& g$ w$ x1 s
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     8 K/ e- m5 c  P* i, o$ z1 q& d5 {
114. MyUPnP::MyUPnP()
     2 U) E6 _2 Q* v; _1 x3 Y6 M
115. :   m_version(1)
     
116. {
     3 N5 |- j5 P* u$ P1 J
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     8 o# l2 m/ j7 d8 V; _6 D
119. }
     
120. 
     6 k. c% M; p- G" V( p0 ]3 p
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     % g. u# b% J) p
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     3 ^) L0 U4 f2 b1 t  m5 P. D, M! T* I; k) r
131. }
     8 Q5 d; f( Z$ j1 f! t4 w+ R
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     $ R/ H; e  S! ~7 [* y
138. 
     6 T7 l7 m$ X) e, e/ G) r
139. #define   NUMBEROFDEVICES 2
     * R- A* T% @+ U
140. CString   devices[][2]   =   {
     ; `7 F5 @) h7 E) o' a
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     ) r, m" w$ V  l& S; J* h
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     # {5 ?; r' k* ?1 w
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     / g( A4 N  |. _* R
147. u_long   lv   =   1;
     + a# Q: D+ i" o3 A) h3 X" y* c+ `
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     7 o) z% e! D8 j
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     2 ~8 a5 y! c4 y# u3 p; {- p7 x
152. if   (!(i%100))   {
     2 H7 }6 V* @- ~" g" g6 w6 ?
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; e! a& r* |+ ?$ _& g  s
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
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 "),
     
157. 6,   m_name);
     / y& Z; x5 j( s% p! O# d8 H9 j
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     # `# L& I2 {$ I0 Q% T9 Y
160. }
     
161. 
     2 \6 K2 h6 A4 _- j% [2 m6 H
162. Sleep(10);
     ; V4 H& _6 f& X1 R
163. 
     
164. char   buffer[10240];
     * l# q7 w& F1 v$ D  B: \
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     3 P* F7 ]$ a2 `! a. M; n1 }8 @
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ' f: ]# G- z8 W5 E
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     8 G2 ]: g8 D9 u
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);
     ) S" s. [/ P3 G6 O" {
175. if   (result.Find(m_name)   > =   0)   {
     0 N7 V8 t0 E( n
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     : D# \+ f7 X3 p4 ^: U- r
178. if   (line.IsEmpty())   return   false;
       E9 a8 u) }, s4 U! S( w) y
179. CString   name   =   line.Mid(0,   9);
     , m  H! I; c) X- j$ o& w9 r" s
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     * p( l, I# F, U- L6 C% m" ]& m
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ; E9 ]$ O+ R7 e  Z7 J7 ^1 q% i# Q& b
186. }
     
187. }
     0 H. Y( r  b! E  ~4 C" o& ]( M
188. }
     7 @* p3 X) h" H; u# a
189. }
     ! e6 [9 m9 e# }( r
190. }
     
191. closesocket(s);
     
192. 
     . J8 D* y( U/ R+ @4 z- R
193. return   false;
     
194. }
     7 J; m* Z! @& Y* g- s- g! e$ }

复制代码

天云上人

以下有关upnp的接口来自emule，
5 n) D6 c" J, R8 _

///////////////////////////////////////////
5 P( ~7 C6 X( j! C. b" T2 u9 }//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

4 G! d9 j8 h0 c8 S( I* B
7 q( @% x5 E1 E1 Y  F( X#pragma once
#include <exception>
3 @4 g# _) Y# O3 N6 F
% z/ Q6 m5 N+ s. l: ]7 E& p
  enum TRISTATE{
# x$ G8 n2 G+ T3 |        TRIS_FALSE,
- `9 {& }& ^! o& n6 L        TRIS_UNKNOWN,
; M+ E3 g& G* ]* a7 j% s        TRIS_TRUE
};


  Q$ h+ h" r& o* h% U. w* Denum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
9 S6 }" Z* D( ?! d
* i6 x% Y, @+ |4 r, r7 k/ ^+ ]

- K& X; j4 O8 Z. F  W
class CUPnPImpl
{
public:
8 {1 i$ J& n/ |7 t        CUPnPImpl();
! A% J, w7 D, w& @( C6 K2 {5 A, Z( W" s        virtual ~CUPnPImpl();
+ }" d) y5 l2 `/ l' \& ?, x        struct UPnPError : std::exception {};
: G7 O0 g' ~) X8 C1 x5 H+ ^. O        enum {
$ p) @3 l$ e" G7 u& a                UPNP_OK,
                UPNP_FAILED,
/ J; K1 {6 M, }                UPNP_TIMEOUT
5 v7 N  i7 Y' ]8 I7 E! U        };
0 o( Q+ y5 u) g9 M9 F

! u& x$ q. a2 ?* n% }        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
( ^4 H3 q2 @2 q        virtual bool        IsReady() = 0;
( Z* H: L$ L5 f7 I* B        virtual int                GetImplementationID() = 0;
/ V( k( O/ \$ a       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


# C* t6 h& v% {( I; l% Q; M! q* I) n        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
! q/ L% _8 {5 Q& Y; J        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
; l/ U3 j) A" h! i
2 G. j0 i7 L5 T; a, o. N" c
6 ?. l1 @* T- z. f. _0 Y// Implementation
( u2 j$ E2 m* m1 V5 aprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
) e3 e. O( `& ~" ]4 Q        uint16                                m_nTCPPort;
+ p9 T; d; X3 B        uint16                                m_nTCPWebPort;
0 Z8 a  i0 e$ {        bool                                m_bCheckAndRefresh;

8 W6 ?) H( h2 F: k4 a" N9 Z" y9 B
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
% O7 q/ {" h- a* e7 [
9 V: Z$ `+ I) x# P5 L9 k
; U& ^5 d9 k5 [/ f// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
+ ^9 w) b3 q: m" O! X: {{
0 w" N  r( B. m5 y" y4 Q- V+ [public:
- h) b& c3 k9 g5 E( f  Q+ I        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
1 K" I6 X0 c! S- E/ u        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
+ W% A. p# ~1 h" Z/ {& L8 E/ V        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
' ^; P- Y; x! L8 X& e/ m- e( P
6 `  I" j7 v+ X1 ^, M. b
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
+ j  T" [+ |- m0 C8 ~
' c2 I( a. g  n5 f9 H! P$ {
#pragma once
5 U) n; G5 X. ~, _, O# \#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
) q5 }6 T+ ^3 w1 |* g( Y( `#include <upnp.h>
#include <iphlpapi.h>
1 c1 \1 Z# {: c2 H: _" n; m$ i#include <comdef.h>
$ e1 C! [5 v+ O" \& N7 A! A( V: @#include <winsvc.h>

* Y& n& ]& i2 X/ p$ _* z9 H* g
2 c) `0 y/ `: V( e8 j#include <vector>
& G/ ~+ g9 h) T! w: @( A0 T#include <exception>
#include <functional>



7 |# S4 k' Q9 H+ e; k
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;
% v* d2 ^! A9 j. W8 f: vtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
6 L9 v' e0 @' K( Ktypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
" Z" a# b. k1 x; }# }% Y% e2 Rtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
4 C* h7 C* d9 v; @2 X( utypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

" v0 g0 c) l* d; G
3 G- D) Y5 s  z* M. q8 \" Y! ftypedef DWORD (WINAPI* TGetBestInterface) (
) `, i+ N. A  L$ V$ E3 d  IPAddr dwDestAddr,
3 x% E2 c" |+ ^9 _0 \1 I  PDWORD pdwBestIfIndex
% t- N4 z6 m8 c$ D* n$ a) p* \);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
4 K6 p- N9 E7 R/ W  PULONG pdwSize,
  BOOL bOrder
);
; u! |1 r) q/ M3 l7 E1 O: _) t4 j
5 G- m1 @# H- R1 K6 U
* ?# S0 R$ ?& E( h  h7 T0 e7 @3 Dtypedef DWORD (WINAPI* TGetIfEntry) (
- S2 v! i& ]3 A4 B  w7 J  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
( H1 K4 P- J, U

class CUPnPImplWinServ: public CUPnPImpl
6 W8 q; ^* i+ A: s{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
+ K3 N1 w- P# B// Construction
% u# B& [8 A0 `1 A6 J% }4 J3 D8 b( Epublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

" v/ o- E  l8 ^8 F5 h6 r" v
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
1 E7 }1 b& I' X# [        virtual void        StopAsyncFind();
' {/ e$ W1 W7 p; C& l  ~+ ?        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
, `! l: K" f+ S

( [6 g8 u" e& G        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
& x* x, z$ i3 e' C7 {, i        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
3 `+ Y2 x% {2 M/ x9 j% d6 _        virtual bool        CheckAndRefresh()                                                                                { return false; };
2 d: Q0 C1 f1 e9 h& L$ m

protected:
$ U# N2 i5 K5 d( s# U        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
4 X0 T, z  L" t( K& b        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
( W, R$ k& e. s( i: V/ `4 d% Y6 A0 ]) `        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
/ T0 d8 C, B5 L. z+ I                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
! ]  |& _1 S) [6 Y8 k% c                {
- z8 z8 h$ ?2 t" A                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
+ |: T& x) R1 Z  l' I; D                        m_bAsyncFindRunning = false;
                }
- d& D3 c. s' J* k                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
6 D2 |; ?# _: n1 G. M5 c4 n% B                {
7 O! d1 s6 Y  t1 C4 ]3 Z                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
3 F) M2 _% d( ?3 h4 u! j                return m_bAsyncFindRunning;
5 ]: D" \4 A1 d) E+ m- N0 G        }

4 n8 z8 A" [. ^' E& ]- ~) G; h
$ ~5 \/ o: O5 R& Q' H' J" ^        TRISTATE                        m_bUPnPDeviceConnected;
3 |9 `+ x+ o9 C8 @& t

# a0 B& [4 f+ {4 X* A4 u// Implementation
$ C' ~! r4 U( A; ?4 g        // API functions
5 u6 Y8 q6 C- ~: s. d        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
" H- z2 c3 x9 E7 u( S4 ~) @        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
& P2 [# z' Z5 J        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
9 M5 T9 D! H9 l8 L8 Y6 u( Z        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
0 F1 b- q$ P( k( m1 }        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
$ n- X; {8 t. ]; P+ ?
0 O2 M4 k, e# f: O
) E- `% q- f8 }: p5 g' `4 }6 I        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
* [0 w4 e* a2 H! e        TGetIfEntry                                m_pfGetIfEntry;
2 M; g. g+ z8 q5 [' T0 z

7 E$ |7 G# t; `( T: s        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
( u6 |5 b- a$ s) F) S        {
& s$ S) Q; q& }" V' y8 ?                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
& D: p; e8 s! S+ D                {
                        CComBSTR deviceName;
* ^. F6 d( q. ^- [                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
) ~6 ]: B1 ~6 U5 B" n$ h

                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

8 l1 N& p" M8 i' O% Z) n( C
( I1 ^- Y' s! u5 k                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
$ H% T- A1 e: T6 m: K                CComBSTR m_udn;
        };
       
' k" p. h0 a9 `6 v6 v& {        void        ProcessAsyncFind(CComBSTR bsSearchType);
) ]5 _4 ]. o5 f) ]( }        HRESULT        GetDeviceServices(DevicePointer pDevice);
- L, {' x7 p8 D# e" c* t        void        StartPortMapping();
5 V/ z0 ]+ e, D; L+ [! X        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
' D% D/ k' j2 g: i. e5 z        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
% Y# I5 c1 R6 ?2 I( h7 d: G        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* y4 I, B) k0 y& ^7 n$ I                LPCTSTR pszInArgString, CString& strResult);
1 ~6 q4 ^+ Y7 y1 I% |3 v4 M/ V3 w        void        StopUPnPService();
9 y$ Z( }0 f( ^: ?1 z' R4 L
% L4 S1 R( K6 e4 s3 z5 k
        // Utility functions
7 y4 M9 P8 ^5 r        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
# G: \3 s# W' O& g+ C! c$ }        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
9 y+ a% M" X1 b- D  {0 V+ Q6 M        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
& i1 I1 p( F1 u" D; p        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
' [1 {3 z6 D$ b6 {4 b2 e5 U1 D1 X        CString        GetLocalRoutableIP(ServicePointer pService);
; q4 {+ Z1 q  n  [4 k  s# g0 W+ g) j

; ~$ e" I# S3 ~# W+ }// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
5 Y5 Q4 `" k" M( \        std::vector< DevicePointer >  m_pDevices;
* j" Y" y+ s: b        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

3 L- L* ?% h6 g/ }
1 J( A% m) O" }) o3 Q, _        LONG        m_nAsyncFindHandle;
2 Y4 i3 x$ e7 Z) f' X        bool        m_bCOM;
        bool        m_bPortIsFree;
  k) }0 N3 U; J' _4 [9 N9 v        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
4 [! [9 n* n4 Y$ l$ X4 _        bool        m_bInited;
' q5 N- E; W; X" v( L5 [        bool        m_bAsyncFindRunning;
- |9 @; S. y3 I& _6 }& E        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
. a" g$ F6 U% B% t        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
5 u. r' y# w5 E, Z9 H        bool        m_bDisableWANIPSetup;
% i+ ]: U. k/ V        bool        m_bDisableWANPPPSetup;
/ C; f! B  c* E" n7 ?" j

. W3 u8 A0 o6 `* E3 l};
- o% M# e& J4 c4 m

// DeviceFinder Callback
# F! q' C% l4 `2 \  xclass CDeviceFinderCallback
: L# I0 d: A5 L; \        : public IUPnPDeviceFinderCallback
, e' v! y3 l; P3 g3 b9 G{
, K  _5 x% {# e5 h# S9 ]/ Fpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
' c& N/ w& @  K' K        { m_lRefCount = 0; }
$ G8 I( Q4 x0 Q9 }  I3 r

6 N% B9 ~' [9 s" N8 C& x   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
7 m  l; d7 M: G) Z   STDMETHODIMP_(ULONG) AddRef();
) W: y- e) q7 U5 m+ R   STDMETHODIMP_(ULONG) Release();
8 s5 l, J8 W8 N# A- Y
. d. E! ^+ e% L+ s
  P- v0 L: |5 M! e// implementation
private:
" H; i: e/ ~: [4 t+ X  R        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
: V$ w+ s- X, ~3 X! ?) H8 Q

  x" B+ o& V, L1 Uprivate:
        CUPnPImplWinServ& m_instance;
0 E( _& ~* E  {: T* L. C. u        LONG m_lRefCount;
2 s4 g! X5 U9 B0 z5 R};
3 N( q8 U0 v  W; n$ ~

// Service Callback
  I5 @: s# x) D, |+ B) r2 h- Uclass CServiceCallback
' b" d) B& g8 h9 S3 e9 L9 C6 q        : public IUPnPServiceCallback
: J; l0 J( P: R; R* ?{
8 ^3 P8 \7 [# Y, I" S5 ppublic:
" Z7 U8 a2 k! p6 U- c0 w        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
' l( }& E, M( }( |   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 q6 v7 f: T2 B   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
( m, p% \8 P7 G' d
" C5 X! [1 E- A' b! t" Y  T8 v) w
// implementation
/ x+ |: P; r0 Z4 `4 L" s5 u& kprivate:
+ S+ D/ _$ {4 p' f! ]. |0 X% v        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
* k0 y6 i/ M( j# P3 ~* J( `" y        CUPnPImplWinServ& m_instance;
: {- @! O$ r3 k+ J        LONG m_lRefCount;
};
6 {" [2 O. H! c7 }0 o: X
% O1 O% E) z& R: a, g6 H7 {9 C
8 @5 x! _4 V! Y3 _/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
, w, ]9 P2 T& m' Q9 S1 RCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.