UPnP

天云上人

/*uPnP.h*/

1. 
   9 f2 M  g. e7 R# ^/ K6 b- ^+ z
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   9 o4 L0 P0 W+ d0 A
7. 
   
8. class   MyUPnP
   + x4 W1 P+ v# G4 a" _
9. {
   . `4 }  H6 }* [& C5 o0 Y2 m
10. public:
    
11. typedef   enum{
    $ Z7 @' J: A- _8 I
12. UNAT_OK, //   Successfull
    5 g8 ?% R% f* f
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    & A5 z7 Y+ g9 e; Z8 m) l  n4 F
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
    ' G- c  s) b( O8 k1 d8 o
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    8 e) h. ?0 n! }4 j# F& l9 Y/ s/ G' Y
22. }   UPNPNAT_PROTOCOL;
    ( X/ V- @: c* _* S6 K$ f" d/ `
23. 
    
24. typedef   struct{
    8 \' w3 T$ F0 i
25. WORD   internalPort; //   Port   mapping   internal   port
    % Q  [* V7 Q5 f5 B! b  a
26. WORD   externalPort; //   Port   mapping   external   port
    ) r- S2 o% o  B( ^! U
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    # H# I% {! {1 l* R
32. ~MyUPnP();
    6 e/ r2 W8 C0 A, P7 h& Y
33. 
    $ R7 y2 J! W& z9 |
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    0 f% Q1 P4 m0 g) e* r/ p
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    6 P  k$ o3 C* r
40. WORD GetLocalIP();
    - b6 L4 N4 T0 r$ X1 e8 e
41. bool IsLANIP(WORD   nIP);
    ( T, `, A2 M6 ?4 I0 U5 j6 W' t
42. 
    
43. protected:
    , e6 t, Q- X9 n" W/ h
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    / k/ N7 v8 T9 ?; j1 r
48.       const   CString&   descri,   const   CString&   type);
    ) H. `; S/ r& h' w% h
49. bool   deletePortmap(int   eport,   const   CString&   type);
    + ]1 J1 b/ E. }3 P
50. 
    3 }' Q/ f4 ^/ K8 C) O; u7 ~! l; [
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    2 N/ _& D" O$ f3 m7 @4 j
52. 
    ! R  t3 [0 ~7 P2 T  n0 f
53. bool Search(int   version=1);
    ! U+ _* ?2 c3 G8 p  V
54. bool GetDescription();
    * U: o5 ?- m( x8 @
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    " ~3 n& F( Q( N3 D. @
57. 
    ( v9 ?! o0 y' i. G. P' @
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ; H: K) L. @9 n
59. bool InternalSearch(int   version);
    8 j3 K4 G  o  r& a9 W' G3 g
60. CString m_devicename;
    8 Q' d$ F5 s7 e& l" X3 Z
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    : ]) J' ?0 C; t5 x4 u% Y$ a3 f+ _
64. CString m_controlurl;
    
65. CString m_friendlyname;
    & E" `; J7 k2 G! X
66. CString m_modelname;
    3 g; n5 O" I3 M* O4 s
67. int m_version;
    ) ?1 D& [9 w" o* [. s/ z) L
68. 
    * x4 Q0 J) ^3 N- w
69. private:
    2 _* s" g% R* W$ S: B& s! w7 b
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    * m1 L9 }( R# m. b
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    * e4 d. d& T6 u1 n$ t# A
74. WORD m_uLocalIP;
    + D8 j# o; D- g7 s# t
75. 
    
76. bool isSearched;
    
77. };
    : |# y0 x  D; G; Q
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   & C0 a+ ^& `9 B* F
2. #include   "stdafx.h "
   8 G. o0 M- A: k- y! T& v
3. 
   
4. #include   "upnp.h "
     s' i4 J; {6 {# y1 h$ ?0 q
5. 
   7 k* f2 ]. j& i* c
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   % {, w" {$ w8 y2 b" {% H# J. c
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    & H  N$ ~8 f% X6 V! f( v8 Y. ]3 a9 l
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    + U- V) b- b& j9 a3 K1 x# i8 {; l( J
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    1 r" J& x$ X+ Q$ H* r$ C) e* ?
15. 
    
16. const   CString   getString(int   i)
    4 x* b6 A/ T, o: C' ]" b3 O  I0 `
17. {
    7 G1 q6 N6 T! N7 M& ^* p! k+ t
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    ! d/ l7 M( \4 x# u. [# e0 \
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    8 P: q/ z  L" S" r8 B* Y* B* y7 _
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    + j0 e" o8 k' p, ]% v. l
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    : Y, g0 ]- _; A) U
37. char   buffer[10240];
    8 q$ G# O1 r+ q8 w1 I$ B
38. 
    
39. const   CStringA   sa(request);
    ( a, _- B2 b& h' @- J$ G1 Q( I' ]8 n
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    $ z. m$ c7 m+ h7 t0 N
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    $ t& |  r4 q0 G% V& Z$ V
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    * {" }. ]6 l( P
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
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);
    + A# T4 O8 M# }' U7 G
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    * W+ w- Z& u- E
55. Sleep(100);
    : Y) I. {% i% }5 X$ q6 ~1 M3 C
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    + [6 _. l4 s" ^+ `! a; n5 z
60. 
    7 U! r9 I5 z2 m  `
61. response   =   CString(CStringA(buffer,   rlen));
    4 c1 H' X8 m. Q- b4 f7 |
62. 
    2 ^9 ~1 f6 I2 }. y; ?. _
63. return   true;
    5 q( n8 X) v8 z( C
64. }
    . b$ Z$ D8 k- V# Y
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    - Z( V" M# ~1 D; b$ I. [
68. char   buffer[10240];
    + b/ p9 k) w1 F7 x  I3 C$ D# Z+ e
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    4 j- e* E5 z- u7 L9 t
74. struct   sockaddr_in   sockaddr;
    . u% J' \" e. R5 [" K# P" Z3 K" M5 [' J
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    $ @9 @2 W5 }/ ?: p  G9 @
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& R) }: P1 e# G; ~. C
82. 
    3 n& ?+ L' @3 `* S2 Z7 q4 v
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    ' G8 t$ p  D# f9 L$ s. ^  F
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    3 F0 d/ H% i7 @
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    ! F4 D5 J2 W1 f1 v  F+ e9 \5 q
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    + i. I' M+ a0 ^4 x! r
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
      P7 _9 q: U8 {9 Z
98. 
      J  V; R0 ^2 S; z1 C' c, [0 s
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     / k* ^7 _, u; p
103. CString   property;
     # v1 H$ y* Q. n6 _5 E9 b0 Q
104. 
     ) V9 C% t2 c, R1 S' s! E
105. int   posStart   =   all.Find(startTag);
     ) I7 h1 }/ H! O6 u1 H
106. if   (posStart <0)   return   CString();
     ' L9 J" V% ~& o: B$ J
107. 
     ; {7 ^% k9 ^' K2 I- R9 r0 {* ^
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     6 r  X! s7 r. P7 i
110. 
     # A6 N7 b5 V' b8 @( w6 A: T8 H# D3 _
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     5 ?& m1 m$ s8 f8 |
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     3 j& F4 m% ^, s: R1 ]) F; E, U
116. {
     : k1 f) E( {. v0 X+ P, \+ q, T4 I
117. m_uLocalIP   =   0;
     ) K$ u& I; P* A3 E1 f4 q
118. isSearched   =   false;
     
119. }
     ' Q# T) U' y* x0 ~
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     - O7 e5 j+ f+ ?( T1 o
123. UPNPNAT_MAPPING   search;
     6 f* z, g1 P2 I* I: w; G3 s$ D
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ) |& Z$ A2 w; _: ?6 O  g7 T
127. RemoveNATPortMapping(search,   false);
     
128. }
     ! c6 N" P$ e) ~  A! U
129. 
     
130. m_Mappings.RemoveAll();
     8 u7 U& H! _, S: P& q
131. }
     3 |2 x4 w$ Q% b% v* E
132. 
     * A# F! ?7 O# T! p' v" K
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     ) ~/ l8 A9 W$ g, ]+ u* S
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     ( \8 f( y9 X, C- F+ p  a1 c
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     + Z* F- x  t0 U& l0 k
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ' G& O3 G) }) W0 s9 v
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     + g/ b* B+ d2 ]# G+ f
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " _( y. t' J3 t$ T3 \7 ^
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     . W" k! |( C, Z+ G7 K
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 "),
     . V" y4 X  K3 [, [5 V5 F$ ?  @( X9 v6 z
157. 6,   m_name);
     # m( X" N+ B) o! b" ?/ {, S
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     7 i8 ^: ^; k& S9 K3 ?$ m' d) r8 B
159. }
     ) ^: S. ^! u. _0 j! l& p" e/ v
160. }
     
161. 
     + D" u0 V( h0 p* d$ G5 N1 Q
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     4 v/ p3 {+ S# ?- K  R4 W
168. 
     % k, p3 Y  C% Y% r. Q' }9 D. m
169. CString   response   =   CString(CStringA(buffer,   rlen));
     + Z6 [: X3 V2 m! f( x% F
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     * ^/ x7 _* i6 e( A
172. 
     : t5 ~) C0 ^8 R+ q, |$ ]
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% X% R1 B  v5 b6 T. k2 |2 q- {+ ~
175. if   (result.Find(m_name)   > =   0)   {
     + D" Q1 ?% A# y
176. for   (int   pos   =   0;;)   {
     ' H5 V3 S. I7 S% E, R, u
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     0 j3 B; B: R- x0 i+ n' @! T
178. if   (line.IsEmpty())   return   false;
     & f7 k) P% T: o3 K8 X; g
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     + S9 V% d  w9 n& g& y5 w
184. m_description.Trim();
     # M8 N* ]! b0 O- I
185. return   GetDescription();
     0 O1 ^2 [9 q8 u
186. }
     
187. }
     
188. }
     
189. }
     * U5 J& m4 B. K% l# t5 t
190. }
     5 u9 b' c$ E. J7 Y
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     4 S% t) q: U) p& k' o5 Q  d+ P

复制代码

天云上人

以下有关upnp的接口来自emule，
  y/ o) V8 @8 L, q
. Y+ w# f" d$ a6 |, @5 T; W
///////////////////////////////////////////
% a% |7 y2 N! p2 g* K//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

6 Q  d  R% A. h5 C+ d2 j, V, F5 Y1 l
0 T  @1 Z# {4 x  k6 w2 D#pragma once
#include <exception>

6 u  k9 V: r% B; \5 u
  enum TRISTATE{
- z4 y; r  M2 }  b  F& @, d1 x        TRIS_FALSE,
# L. L- u* K# ^8 x  J# }/ }  P$ x        TRIS_UNKNOWN,
        TRIS_TRUE
( w& c5 T/ p6 v% N7 n9 n) V5 Q" G};


enum UPNP_IMPLEMENTATION{
# v; Z5 V' ~% F: K        UPNP_IMPL_WINDOWSERVICE = 0,
* V0 o2 h! z8 r# Z+ p3 ^        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
5 y1 U1 C9 ^3 u) d& {! @};

5 p  d( a% s- _

+ x, J8 _: Y" ^. o
class CUPnPImpl
{
public:
4 D! e! x6 i+ ]' Z& Y/ j! Q        CUPnPImpl();
        virtual ~CUPnPImpl();
0 k% a: S0 U# V/ Q- j        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
9 [% L9 ~% ^0 t# F* {                UPNP_FAILED,
                UPNP_TIMEOUT
! ~! ^. }3 a/ O: k9 j" T        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
  X" s. K% [6 ~$ V% Z; h        virtual bool        CheckAndRefresh() = 0;
: A! a9 W& ?0 u# [        virtual void        StopAsyncFind() = 0;
, S, V( O5 u+ N4 A        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
9 ]" X" r/ p- U% E+ G/ N1 V        virtual int                GetImplementationID() = 0;
, ]8 M/ B8 ~0 N: @9 C* M' v       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


. _- R; j; q5 f6 J        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
) ?' u% [9 b  ~) J( M        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
) I* p' v' a( X4 q0 g        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
0 c+ q1 D# v8 ]( g$ _+ j
7 w( X2 i* q1 D) @0 G8 s) x' @
3 u/ M! L! {0 M! ^* G. B) l// Implementation
+ Y$ I8 c' o/ c: ]protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
, ~! m" z" \" @4 c. K3 R  c        void                                SendResultMessage();
2 Q7 \, O2 ^) u; a        uint16                                m_nUDPPort;
4 q3 Z! L/ h6 i! O% i# R: `        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
4 ?* A+ ^! O. ~5 o        bool                                m_bCheckAndRefresh;
; A, R& C' U" D0 A; D
4 Y7 S, x( l3 {% \
private:
1 Y1 V/ d3 A, S6 @        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
% ^. e- \/ A+ r+ J

( D! J7 c: E, O9 O8 G# A# |};

9 W" r0 j7 @" K& b7 r9 f+ @: L
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
1 l- \' m" T5 [        virtual bool        CheckAndRefresh()                                                                                { return false; }
& T/ V. ^( ~& f4 V' T* ?        virtual void        StopAsyncFind()                                                                                        { }
& h! R8 n: ?! v        virtual void        DeletePorts()                                                                                        { }
" K7 t3 d5 ]4 B4 m: F1 A+ q! R# ?' I        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
2 ?7 C/ @3 q9 d# |, X};


& I; l' p5 g" i# t- d8 c- O7 i( H/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
& a+ s' Z1 h& I5 [+ Z* V+ J. B

#pragma once
#pragma warning( disable: 4355 )
% X/ [1 F. R$ c% Z: o; H! W
9 J2 F1 K+ i. M/ }
0 o' y' w: p" U9 u! b#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
  W0 s  u$ t3 M, y#include <winsvc.h>
1 E& s) w9 m( t% U$ |

#include <vector>
, m4 I1 j8 U5 [/ F1 U. W#include <exception>
#include <functional>

$ Z2 e( U+ F# J5 |1 X: {


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
' E3 o& U) g$ D. G4 Wtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
2 f" X# d* M6 o4 Ptypedef _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;
3 E, `+ X% L/ h$ h8 ?3 u+ d5 N5 f1 |typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
. U% J7 `) H* L. V' p8 I7 j  IPAddr dwDestAddr,
" D1 {! V# D, B( ^  PDWORD pdwBestIfIndex
1 }/ {7 N5 L& K! x);

, Q+ x+ `& [7 Z& y( [
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
& R0 H, x; @+ U* @. J  PULONG pdwSize,
9 \' _4 `1 {8 q2 v9 Z& z  BOOL bOrder
* @( S! E% m3 f3 V# f7 D6 B);
: A: m$ w. R1 f1 v; e

typedef DWORD (WINAPI* TGetIfEntry) (
( J& B  h  r( j6 H, I  PMIB_IFROW pIfRow
);

; U6 L& r' O/ q
CString translateUPnPResult(HRESULT hr);
  t" j0 l2 ^# L; ~9 ?HRESULT UPnPMessage(HRESULT hr);

* W  g4 y* E, V
class CUPnPImplWinServ: public CUPnPImpl
1 m4 @6 a4 B2 T* M0 E. y$ n{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
8 d; L" v3 }, G1 f// Construction
public:
+ w1 H8 Q, }8 E1 \. D8 I: U, L        virtual ~CUPnPImplWinServ();
* H( w0 }' J* G6 y7 t% Z        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
0 G! k' }/ K9 a. }        virtual void        DeletePorts();
        virtual bool        IsReady();
' ^# g2 t. \- h5 k1 A  U        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

+ r( C4 r8 C* t5 r. n% j
& m5 A" j0 B5 M+ w) |2 N* F        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
# |" z- ~6 L) ?* v7 L0 s5 P        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

$ G6 s) n) H0 p: U* O. @3 }
protected:
9 y9 ?6 L8 R  H% Q4 `4 z* ]% k! s( k$ I        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
4 M+ E( A, _# s        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
        {
9 u0 u% ?7 a' Q8 P                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
* ]: n% Z6 U* `+ Q0 g. n                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
8 x/ N& X7 ~1 V* d6 x+ k                        m_bAsyncFindRunning = false;
                }
) z) _) x3 m( s# ?                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
; a% g$ ~8 y5 E                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }


        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
$ s9 K, L; p2 A1 e: M3 R        // API functions
# y3 h  q1 G$ d6 R0 I( l! D- }1 \        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
8 [, M2 v7 M1 b3 O' H9 F        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
* ~, }3 E5 ]6 y0 V4 ^& W        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
! X1 c" K$ K# H8 k: x- D$ j
+ k1 t4 Q6 d6 L# Q
; H2 R: B& M' z) V- q        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

' a! S6 G) y. ^! j
) S4 K8 m- j( C; N0 R! _" b        static FinderPointer CreateFinderInstance();
3 M. Y/ a2 B# E) e        struct FindDevice : std::unary_function< DevicePointer, bool >
/ h, t* B8 t* Q, K" f" }, C        {
7 Z: c4 i) C/ S                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
$ k* A+ h9 Z9 @& N% y                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
* `/ U$ C- L2 Z* W& }& a/ m  D                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


, s& j, a# Y  Q9 S                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
* R/ [$ G7 E  r5 y" k$ p! v9 v
/ I, T& Y9 ]& T6 u) W+ x
6 Z6 I  ], z) }7 U& l& ?3 J" ]2 X                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
* L  q2 q' M# ]1 g+ W& I. `                CComBSTR m_udn;
        };
/ Q7 B0 W" F0 P9 P. @7 |       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
" ~9 z6 c  S) e: U! o        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
! h4 \' a: k( I3 O        void        DeleteExistingPortMappings(ServicePointer pService);
* i+ \5 o/ ~# i$ t% t# N        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
5 h1 t! G& n2 h* B        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
, b! \) p3 a( J: s( {1 y                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
. ]/ T! X4 {  _1 }7 C) z
8 F3 D. F: u3 o( {9 s* C
( A! h' i3 |& {* z2 f( K2 G1 N        // Utility functions
$ P3 Q9 T9 e0 I; ~% `% x) w6 T        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
* S; u8 l7 l0 r        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
) a! z0 Q" {5 y: ?, ?        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
- j7 k' Y* _& K3 A        CString        GetLocalRoutableIP(ServicePointer pService);
1 Y$ U$ u  p7 _0 R. I2 N. x0 ?
. C9 L% V* i( Q
// Private members
5 k- ?4 t- O  y! E# B) X& Iprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
* R6 z* |" d. j& s4 T" G+ S        std::vector< ServicePointer > m_pServices;
% z! d9 P( @4 z: h/ ?        FinderPointer                        m_pDeviceFinder;
1 u9 p& f6 O6 U        DeviceFinderCallback        m_pDeviceFinderCallback;
% y4 l  Q  s' A% b, ]4 r        ServiceCallback                        m_pServiceCallback;
6 a0 [1 M4 F% G* m) V

9 L4 n% F" r% c& u$ @        LONG        m_nAsyncFindHandle;
' |  c. x  y1 i        bool        m_bCOM;
- @/ @( C; c6 O) |2 x        bool        m_bPortIsFree;
5 @2 V% s8 }2 w! k5 d        CString m_sLocalIP;
' `" D& l4 ?( L7 O        CString m_sExternalIP;
0 p  C" R5 Q$ \5 l6 R# R+ _        bool        m_bADSL;                // Is the device ADSL?
4 @' B6 ^0 O% Y9 A, G+ S        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
9 H8 k' O: D# e% P9 j; U        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
- |$ K! H$ q6 O) k; s* t        bool        m_bServiceStartedByEmule;
& Z; |: f1 \' C' y; {1 U3 @        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
- x" n# S: p6 y3 ~# @

; M  b# g1 W+ G1 E1 ]};
+ C7 `; z) q# e
% q0 ^8 q' p0 p# w# \
! A* r  E0 E( p2 e// DeviceFinder Callback
: ]6 y0 x  O" _% Mclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
7 _+ {  h8 x* Q& {public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
, }+ `8 p" V8 @8 ]2 N9 a* M- A* m& \

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
3 D* r( M& Z2 C% e& ~

// implementation
private:
4 F& K; A! r; p. f) t        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
- p1 O0 W$ E9 M5 A% k        HRESULT __stdcall SearchComplete(LONG nFindData);

! `' @* _! E+ t. I' v0 g
private:
3 }0 n  ~3 e- `1 O  j4 P        CUPnPImplWinServ& m_instance;
0 T/ p2 ~# a+ w, i! I! Q        LONG m_lRefCount;
/ O! Y9 n1 d/ a1 d* F+ s1 |};
4 a1 s  t# K' X! U

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
0 [5 g" ]& g" T8 R/ a( I# ipublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
$ i$ h  F2 B8 T0 a        { m_lRefCount = 0; }
8 j+ [  M) _% @7 u1 g6 Y; }   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
& k5 a$ v! c: p' o) V3 k# E   STDMETHODIMP_(ULONG) Release();
! E# Y" k& _8 X$ C% t7 b

- H. w; D4 v) o( T+ ]% s// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
3 }5 l( d6 J$ q& O
9 N1 E7 W9 \0 F4 g( I
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

4 z5 _* W% I4 V) w0 m7 \- m
0 {: s, v4 k% C/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
; E" z+ z* k6 ~) L7 K1 i0 l0 RCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' c* p' a* _8 c! R* [CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
5 \* k! _8 P0 P  z  l; y# ?0 ?$ ]. gCUPnPImpl::DeletePorts删除端口映射.