UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   , u% V: U" O! T# J8 O! z8 J- o
3. 
   
4. #pragma   once
   
5. 
   8 c. @* @! N/ F" F0 `- m
6. typedef   unsigned   long   ulong;
   8 L3 s( K7 L5 R! |  a
7. 
   
8. class   MyUPnP
   : w( h( ]1 ?: C% w/ D
9. {
   
10. public:
    " {& k4 W- @1 j( @& {
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    8 E% G/ @- t! s) e
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
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;
    , S8 n/ t) `+ d& k& W
18. 
    # Q* h$ c8 y1 k8 F+ F# M
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    & ]- Z8 z) Y) A4 ~5 E0 k3 @' ]
21. UNAT_UDP //   UDP   Protocol
    7 e/ c1 u3 h' x/ R1 K- I
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    * z3 |( |4 [( u( P( R
25. WORD   internalPort; //   Port   mapping   internal   port
    . @( k& B! ], O1 O, {8 D
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;
    ' b6 @2 y, @3 h8 c- e; i
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    5 G) n5 u' X2 q# q3 e3 w. ]
36. void   clearNATPortMapping();
    6 i# f  j7 q; g% I
37. 
    * t5 B& B! r7 X+ N
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    ' O2 T" j% C2 l  a
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
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);
    . n1 s# ]4 U9 @. m% S
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    * h* Z! Q* L9 C; g5 {
55. CString GetProperty(const   CString&   name,   CString&   response);
      d5 b; D7 t' Q  E
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    # e( A5 y5 C  u: D' W6 l
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    ' k) j; h% n8 }8 V, M0 j$ H( P
62. CString m_description;
    * ]$ U5 @$ X, l) J1 F
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
      h( r3 B& L' J, c
68. 
    , \. x( G/ c$ ^7 _
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    ( a6 d1 W/ ~+ S9 g* L6 |
72. CString m_slocalIP;
      w2 @0 t" ^: v+ r; u3 r$ l
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    % k" b7 p$ T  G3 R
75. 
    8 V1 R# ?! S2 d: g! n  ]
76. bool isSearched;
    
77. };
    , I7 x; `) N4 w6 i8 B% _! m
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   8 O! L5 ^) M: k% m8 m! l
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ' m% G* t6 ?% }2 ]2 }6 O
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ) i2 k6 X) p2 o* c# n( U; v
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   1 K9 y$ {6 D) g4 _
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - e: h5 v- }. U* a2 O4 p
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    8 s3 c  K) v5 e0 [+ O
15. 
    * b9 P( M7 [* ^* n7 o- a
16. const   CString   getString(int   i)
    
17. {
    ' [0 g5 [' U! T! z4 F
18. CString   s;
    ) f. C: n$ O7 A1 x/ h, Y# Z( f( R: D
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    & l& p& ?2 T& v$ ?# A5 B
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ! n* e/ ~6 Y1 A4 l
26. {
    ; n1 m/ [% o$ l+ L; l
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    $ y, x) G( m/ Q3 L
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; t1 c* Q7 y' R  m1 p# f3 m0 U
31. {
    # T$ c6 R' C' y6 D* p5 h- {" c+ V
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ) n) o( }2 }& l$ I, k7 B
34. 
    & B9 W2 {& h" U* b( e
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    5 {; A- x$ _! a3 |
36. {
    
37. char   buffer[10240];
    
38. 
    7 ~2 e) }" Z; r  B
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    * @& X6 k, O1 g( S# q# f  L
41. strcpy(buffer,   (const   char*)sa);
    ; v& I% E5 w" E9 I. Y
42. 
    2 b6 t& ?# m( q4 r# d$ g0 j
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    + B# Y7 ^3 [& c7 ~+ T3 ^
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ( s4 d3 u5 h7 x8 l% B
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    3 S6 k1 a4 B6 Z# m
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    3 G4 J. b9 }/ D1 Y5 R7 ~
59. if   (!rlen)   return   false;
    5 Y4 Y; A9 @& y9 ~! i' e
60. 
    6 J1 L. a2 b; V5 B) t# v8 k+ ?" a
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ! t! g, D9 Q9 P9 y1 O) K
63. return   true;
    , z- Y2 M4 B' f
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    # q2 }1 Y2 A. N
67. {
    9 ^2 L8 v8 J3 q/ e0 e' R
68. char   buffer[10240];
    
69. 
    / c8 G8 N5 b2 \1 `" j. h" _
70. const   CStringA   sa(request);
    3 n& N5 R) }+ }7 t" u
71. int   length   =   sa.GetLength();
    0 x8 d) t1 F% N* u
72. strcpy(buffer,   (const   char*)sa);
    8 L5 {5 U  G. k, O% b9 _' @1 {
73. 
    , R, k+ A6 ^/ i4 M4 D2 W
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    7 z+ ^. I+ t: z% L6 g
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. }
    - L/ }9 B  P( t$ k6 h  L+ G; `0 B
82. 
    , k+ t; M' u$ A& a# y# @! \  u
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    2 A) s) U1 |2 {8 Y+ ]* L- M
85. int   pos   =   0;
    / i; r* ?1 R: \+ A3 z  B  j  `
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ) t! P# u+ A: J- q4 z3 @
88. 
    6 O, _6 i$ Q6 P: j
89. result   =   response;
    ! ^3 x* f6 i+ L2 g  r
90. result.Delete(0,   pos);
    
91. 
    6 P- |8 Y+ U( j6 J( D
92. pos   =   0;
    & m: `, h, f7 p- v1 `# V/ I
93. status.Tokenize(_T( "   "),   pos);
    7 }+ `, D; A( w. B
94. status   =   status.Tokenize(_T( "   "),   pos);
    6 \3 ~: N' |$ M& D6 G! \
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ! W/ o: U. B+ z/ ~; h& F4 w
97. }
    % g& @7 m2 D8 |: m/ q  H
98. 
    ) S. h/ P( Y' p  x
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    4 W) }6 V) R! S! X
100. {
     + n4 f+ y' _& A& a" i$ O
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     ' h9 D" H9 r; l* x; d
106. if   (posStart <0)   return   CString();
     1 i2 T' i6 M/ g8 ~  j( @. e6 R" k1 z# C
107. 
     $ j' K& v# y5 z! c! @2 u' Y
108. int   posEnd   =   all.Find(endTag,   posStart);
     & i2 t# B8 N. H* M
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     6 _0 |: V& n0 Q9 f
113. 
     ! p4 S2 Q0 Y4 f! S
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     % C% g+ z( e( I+ V0 j' {' B6 f
117. m_uLocalIP   =   0;
     3 d4 z; X3 |/ b: R
118. isSearched   =   false;
     9 T+ }- q; d. `0 p" L. U  j
119. }
     # g$ ~7 E+ f/ J+ N: s$ L  @, R3 b1 i
120. 
     $ p" F# C8 n, N0 X' x# z1 h
121. MyUPnP::~MyUPnP()
     
122. {
       o, Y9 c0 C' M  B' r, y
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     3 Y- {4 L2 i  a
126. search   =   m_Mappings.GetNext(pos);
     6 h3 s, O. H7 P8 U" i/ S2 q
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     ! b8 b( h. i9 {# c% F* N8 L
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     - M# A' `3 o  O% A, o& s7 ?7 x! {
135. {
     
136. if(version <=0)version   =   1;
     / j% J* p4 I4 M, n" }/ D) u
137. m_version   =   version;
     
138. 
     * q3 u0 u+ r6 s6 a- S
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     7 q3 N8 m, h' W1 J6 u% Z8 q
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     ) n5 e3 ~2 y- x$ g+ o" q
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     * n7 v; I" v0 R$ O) l4 U' h
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; _2 o8 |+ e: ~, H1 P
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++)   {
     ) Q( ]% j8 ~$ S' C( i/ [
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     , c1 c" i9 w3 S
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);
     # V0 ?/ j: t1 l5 \
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     1 M$ F: s2 ?9 ]8 A
161. 
     
162. Sleep(10);
     * @+ I# [# o" N7 h$ G# u1 p' D! M
163. 
     ( V+ L& h3 E8 X- S( [. M
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ( }/ v! x% i  f0 g  ]4 s) H
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + Y% B) `. V, U' R! I
170. CString   result;
     5 Y  P( [: U- v
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ' Y6 p, x( D2 M4 U! y0 K( [: v- C- @" I
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);
     
178. if   (line.IsEmpty())   return   false;
     , S( a5 _5 O6 `5 v9 q- g+ P
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     ( w! ~; ]4 P- u3 w
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     " n  J+ J. U- s
183. m_description   =   line;
     
184. m_description.Trim();
     * H6 d: d# E5 ]# D& G
185. return   GetDescription();
       a! W) F. |( ~. P2 i
186. }
     
187. }
     $ @& W& _* B" {1 A2 C
188. }
     
189. }
     . W( p) ~% F' o. i
190. }
     + H" h; Z, u$ h# C
191. closesocket(s);
     2 |, v$ A5 n! e  o( u# ?
192. 
     
193. return   false;
     " o5 d; T9 J0 b4 Q* v
194. }
     

复制代码

天云上人

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


; c/ m0 w# R3 ^0 L; J3 O6 ?///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
" b+ J+ p0 V! W

7 ]0 {! y7 v" R: I# v#pragma once
#include <exception>

* S* o7 c* Q% p9 D4 x, q; u" l. v# H
' g2 C: |; M, ?0 b4 t" }$ Q+ w  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
9 s! q8 V  v$ `        TRIS_TRUE
};
9 e* ]& `& A. ?$ Y1 V2 c
$ L3 e0 C3 `! q
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
" Q1 g) m* V: e5 O( s" O2 G        UPNP_IMPL_MINIUPNPLIB,
+ v7 m: ^, W: k) m; K/ j- j7 I0 U        UPNP_IMPL_NONE /*last*/
};
  R7 K% _4 Z; r* A6 w



class CUPnPImpl
{
5 a0 D5 z" G1 h" s, Wpublic:
        CUPnPImpl();
$ b2 b3 R1 n0 F! ^/ Z# m7 E' R        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
$ w1 Y4 g  _/ m" W9 V4 m5 L9 F                UPNP_FAILED,
1 F) |8 S4 b; M8 P                UPNP_TIMEOUT
        };

+ G, F# B/ P  V" w
3 T: c# P- b1 P, X        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
0 G3 d$ \+ R' e# M        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
0 e; l& |0 F7 [. e8 b        virtual int                GetImplementationID() = 0;
" v# p' s+ l7 Q" n6 E8 P$ I       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
* d8 j5 E# q% [" M, `
: `& i# J9 X# r5 }8 X$ f* |
7 K3 S  u# d2 }; d3 G        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
0 u' x. Z  Y7 |        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
3 w' w' @/ w+ \# x5 a        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
* t8 P% v  x$ v3 Z$ n4 N3 F4 M( _1 Y

// Implementation
% _: M( C$ y/ R! H1 Mprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
0 _' x6 W; e. R( O        void                                SendResultMessage();
$ E4 g) E2 t. I% y' b. z        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
- b& f/ B( X7 \$ _/ k/ F        bool                                m_bCheckAndRefresh;
: Z, j/ p0 `2 j/ E7 i; R

private:
        HWND        m_hResultMessageWindow;
2 Z9 p# g: ]9 i9 Y        UINT        m_nResultMessageID;
. y- S9 {+ M( y- l! m( g: I  a
3 ^1 N' S4 J2 ~2 g: ^
};
8 u, d; G: c, L# m9 \/ B( i
" t/ Y7 Q$ S4 ]( U' h
// Dummy Implementation to be used when no other implementation is available
9 X6 |8 P5 ~' m  Zclass CUPnPImplNone: public CUPnPImpl
{
6 x  x7 }8 _' z+ fpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
2 Q% F) I8 ^$ }& g+ i$ G1 S) L# k        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
. C. A0 G4 C' d. ~        virtual bool        IsReady()                                                                                                { return false; }
" ]7 v' j4 ]3 d- x; z* A        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
8 \, y8 W0 |" d8 P" w6 }

( H+ L7 i: l& j' n& c/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
' }. u) U- N8 x6 v# U& r9 U- f

: e0 b; `  q7 }# ~8 d$ y! f#pragma once
; k* D! U6 u3 M9 V#pragma warning( disable: 4355 )
% L& J9 k  }; C3 q* V

#include "UPnPImpl.h"
1 X% n. X1 _% c" F5 g( P  ]#include <upnp.h>
#include <iphlpapi.h>
% a; V# Y$ `1 j% q& U( O#include <comdef.h>
, ~$ r; N% {# v( A8 z#include <winsvc.h>
$ u) ?. @) F/ L/ c2 w5 P, U
0 ?8 s4 ~5 H( n! t
; y5 `; q  J, W0 ], h) v5 J8 _#include <vector>
9 Y& h6 @3 P4 X. b/ K+ N#include <exception>
#include <functional>


/ f2 D1 _- z' O

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
: l& l% A: ~1 {typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
0 I( f# J2 |) j- Ktypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
9 ^# |' w6 s7 f1 L, m. q7 A% Gtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
- M" a4 u  ]- k' D3 ]$ f; A, W2 Xtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
- [1 n2 w$ q) q6 a) Rtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


" e( ^: M/ c0 C+ gtypedef DWORD (WINAPI* TGetBestInterface) (
. e' S& v9 ~5 Z' X$ l; }' z  IPAddr dwDestAddr,
1 b3 ?5 T* C$ b- V4 ]  PDWORD pdwBestIfIndex
);

; J; V0 I! \* Z8 ^0 n6 c& E' \* c
typedef DWORD (WINAPI* TGetIpAddrTable) (
% r* {1 ?2 e. S$ Z  PMIB_IPADDRTABLE pIpAddrTable,
" a! e$ F% r8 C& F- n  PULONG pdwSize,
/ B5 X. ^0 a9 V+ l2 l: F  BOOL bOrder
5 I; m! w' u* |);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
! Z& p8 p- `3 r+ r+ O

CString translateUPnPResult(HRESULT hr);
' M- }, z- s) V' C4 R8 m, S1 f% SHRESULT UPnPMessage(HRESULT hr);
+ g$ F3 L, X9 B7 l" \+ M+ y
, K. p2 T  P, ?6 V* N
- Y+ h) h& b# i  X4 z: M  \class CUPnPImplWinServ: public CUPnPImpl
9 g. [; _6 [, W6 r+ T{
, R5 Y6 j7 z" s  H. E# R        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
& h! b* e; E! L! j  m& d! zpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
% S3 _0 _3 f3 V: c) i, g
; o8 N" H* L8 M* ]0 j+ \& e8 i
( _: m7 b2 h7 x( [        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
  I/ [& K! W1 z3 x$ I% q/ a        virtual void        StopAsyncFind();
% t& j8 K" ~& `$ W/ `9 e/ B        virtual void        DeletePorts();
, n# ~) j0 \: ]$ P4 z, j: d) ^        virtual bool        IsReady();
- u& ^0 C/ N4 s+ i3 r% A, s8 v5 E+ L        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

( b2 o% t1 b6 L  m9 T9 ?
8 k/ w% w1 n# _5 R" P8 H5 R2 ?        // 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
        virtual bool        CheckAndRefresh()                                                                                { return false; };


- s( w# E5 h" ^1 i/ D! J( aprotected:
5 Q1 g3 ^5 Z. z4 p7 Y" U        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();
& t9 A; H9 b9 @        void        Init();
# l3 d, v+ Y+ B
& T$ E3 R9 S$ ^6 y" x
$ |: b, r2 ?  y" B        inline bool IsAsyncFindRunning()
        {
+ K+ s1 {6 _/ Q5 O  i7 \- J  X6 H4 l                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
+ }( B6 l: T. B' s  k                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
2 ]7 L8 j, q5 p* y                MSG msg;
' l" @7 X. t- _% R$ z                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
: [( h0 ]/ y8 z; [7 `* J                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
( F: k/ n4 B% \& z) K                return m_bAsyncFindRunning;
        }
  h& {4 G0 t1 h2 o

        TRISTATE                        m_bUPnPDeviceConnected;
  g' \5 U5 i* t3 |8 {1 r

// Implementation
        // API functions
& V; {' @7 t& c' ^$ [        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
; L0 a5 C' Q0 @' V/ L        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
/ w3 M5 b- _8 {# I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
1 S2 G8 ^/ V) l' a        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
, _$ z# |0 {. m9 C) ^" j& o
+ D+ C) m4 u3 k) h
+ L3 J6 X3 E& [' Z& Z        TGetBestInterface                m_pfGetBestInterface;
6 f% Y: [3 Y, m) M: U" A: h        TGetIpAddrTable                        m_pfGetIpAddrTable;
6 \* s+ s3 B. O& G8 P& ?( A* R        TGetIfEntry                                m_pfGetIfEntry;

" P4 f9 i% K8 g- S$ s
+ G$ U  ]( u% B+ b% \* Y; {7 J        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
/ x. a6 v  R( N: g9 v0 o        {
0 e" D2 n& k( O$ m8 Z! D                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
; f% F2 M" m& f4 }$ \; L& E/ B                                return UPnPMessage( hr ), false;


& b. z( N: a& g. T3 k' O, t; e5 H                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
5 ?' e+ Z0 i& l4 E        };
/ ?1 G- U( c: Y$ k       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
* _/ m! T8 `# O        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
9 N7 M( W# i; u" i        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
# ~4 I- J" @% c4 G6 B( O        void        StopUPnPService();


  r9 B& _1 |& P7 C, F        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
0 Q. p6 Q8 |5 X9 S) I& L: c        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
. L' T+ V/ D$ Y& |' B8 \; v        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
. _$ x( }8 ?+ H0 |+ ~( C        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
2 L) w4 W  n, z  n        CString        GetLocalRoutableIP(ServicePointer pService);

( f$ p6 R% a3 {
$ c" F  M! l- T" H// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
5 M- o( q1 o1 T3 e/ A9 w7 j        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
" K# l( c, a: g: F) `        ServiceCallback                        m_pServiceCallback;
8 s1 J" p! G9 g
# E& j0 g" B5 e: }1 X, F$ v. V
        LONG        m_nAsyncFindHandle;
6 w8 o2 W: B) f9 F7 t$ ^        bool        m_bCOM;
. I4 v; Y" Q, C; G) {        bool        m_bPortIsFree;
7 s( P4 O) T! o0 t  o0 V        CString m_sLocalIP;
        CString m_sExternalIP;
6 p) W2 G$ z$ c/ Z        bool        m_bADSL;                // Is the device ADSL?
6 |' i! s  `( j; Q5 P# F  R# p        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
5 i$ ~3 @! c. r1 F        bool        m_bInited;
& j, {! b2 ^' B' I/ ]  E        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
+ M$ q/ b0 ^' h7 j+ P9 Q; u        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};

* G5 H4 s$ a6 B6 R/ K
// DeviceFinder Callback
class CDeviceFinderCallback
9 E8 i. o5 t! I) t  \( P; ]8 }        : public IUPnPDeviceFinderCallback
{
3 P+ i  E; ?; t; b) _0 x2 C- n  Npublic:
& t. t& h; ?. v) w% V        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


' g6 E9 I! Q0 ?2 n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
% @1 N7 x* v. ]& J0 S! C   STDMETHODIMP_(ULONG) AddRef();
* K3 Z; K- ^1 I& s6 a/ f0 b   STDMETHODIMP_(ULONG) Release();
" X0 d: W2 r3 i

// implementation
private:
) C$ H. F* y" P        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
+ [! m) X# G9 S- f6 Y1 {1 O+ g; J        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
$ }* R* ~( i3 g0 V3 I        HRESULT __stdcall SearchComplete(LONG nFindData);
  M0 f! ~* S7 }6 e1 C6 V
( ~+ B" _3 m2 B0 a
private:
& ^: o8 c: A1 o        CUPnPImplWinServ& m_instance;
! U+ F) c* |- {( ]. F% v        LONG m_lRefCount;
};
, V/ q) q9 _& v7 N; E! x9 a
& }- |- x4 E9 L0 z: W! [
// Service Callback
/ o4 Y: Q* w2 Fclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
9 e+ Z8 u$ u: C% g        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
  d0 t9 W0 ~* y5 Q$ j   
( H8 t9 L9 Q" U! D8 y" ]   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 I1 X' z6 R3 L7 c  n: ]* E   STDMETHODIMP_(ULONG) AddRef();
: N2 B# F6 K: I4 O: ^( ]% f   STDMETHODIMP_(ULONG) Release();


' T8 \6 ]  K+ j5 y% J// implementation
% u* F  B4 r) S1 Z1 c) y2 kprivate:
; ?) S) R0 w: L0 ]) T0 f1 w        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
1 k# M4 |! h5 `        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
# X7 r0 I6 w; c: c$ _5 y+ I
3 i; k, L+ {6 N0 R& P
' P( d8 h& `0 Wprivate:
; S% s( Z7 M: _: s) u; X; y! w        CUPnPImplWinServ& m_instance;
+ ?$ X6 g* ]7 O! u) K: w2 b        LONG m_lRefCount;
};

4 `& N" G& Q" t
+ w9 I+ z5 d% }; n2 `/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
' |: G. y7 l0 t5 fCUPnPImpl::StopAsyncFind停止设备查找.
* @* ]2 v9 r) e0 qCUPnPImpl::DeletePorts删除端口映射.