UPnP

天云上人

/*uPnP.h*/

1. 
   6 h  {+ \# t6 q. M2 Z
2. #ifndef   MYUPNP_H_
   # O6 X# K# I# t& F: O4 m
3. 
   - b4 Z) k" n* z, A$ L. l+ C
4. #pragma   once
   & |% }) w0 N8 x
5. 
   
6. typedef   unsigned   long   ulong;
   6 G* f0 W7 n. E( g$ ?
7. 
   5 _& d9 {$ t; a& C+ @
8. class   MyUPnP
   
9. {
   & O1 w# P1 }5 i
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    7 G5 ?. I$ @3 z+ y
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
    / X/ W9 L- y' J) Z6 H
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    & ?9 u% V* }; j
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    1 a2 G: O. O- T# h) L8 ^* n2 I) F
17. }   UPNPNAT_RETURN;
    ! t# e( U6 P1 w/ m, r3 w2 m
18. 
    2 I# @) g& u. y' X' u5 r5 a. A8 ?+ u
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    8 A$ |+ T$ Q4 L) V3 N
21. UNAT_UDP //   UDP   Protocol
      U3 ?) X3 M$ y
22. }   UPNPNAT_PROTOCOL;
    7 U7 F9 Z4 y# Z
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    0 ?3 v) m1 f0 O8 y9 n/ e) z
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    - A+ p2 K% x8 u* ~' j# B0 G
28. CString   description; //   Port   mapping   description
    $ V* D+ y4 E; i6 M2 o: _9 U9 W) c
29. }   UPNPNAT_MAPPING;
    8 l4 f/ j3 e& a* l" _
30. 
    9 d9 f7 |* n4 E4 s
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    : j8 `* Z( W7 [6 l7 b' u
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
      R3 g/ G1 H' c
36. void   clearNATPortMapping();
    
37. 
    7 G1 j6 I- ^, u* v1 v: T/ M, s" k" E  q
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    / G9 l% Y- k7 C) T
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    5 H9 M# m9 e7 ^' o
42. 
    + T9 D1 x% e; L4 ?' q7 z" F
43. protected:
    / D7 Q9 r1 a% J, H. w) J( v
44. void InitLocalIP();
    ' h6 H& b" E# L8 I
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);
    , h1 S( [* c/ I& {
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    & X  r% L4 e, A/ N! v2 A3 V& s: W1 Y
53. bool Search(int   version=1);
    
54. bool GetDescription();
    9 ]8 ]$ @$ ^" n
55. CString GetProperty(const   CString&   name,   CString&   response);
    ( R, V. P" K2 z8 O( W
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);
    
60. CString m_devicename;
    
61. CString m_name;
    ' c& N% v5 P  r5 ?
62. CString m_description;
    
63. CString m_baseurl;
    ; X# T$ I$ \- K' q6 S5 X
64. CString m_controlurl;
    , f) c$ \/ [  x% ^: v  Z# C1 T
65. CString m_friendlyname;
      h. Q1 w  Y$ f+ Z" ]% s' w7 `
66. CString m_modelname;
    
67. int m_version;
    3 E8 e1 I; I8 \# [- W8 A/ q
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    5 {! v  k0 l$ C
71. 
    5 }& M# J+ y6 E4 k% H" A
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ) F; X/ T5 C) g1 N5 {: y7 y# W
75. 
    # ~5 [/ G% x5 k# ]
76. bool isSearched;
    2 ?+ ?5 j" X( n- ?/ C* u# C1 r
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   + T; i. h: H8 Q4 ?" f
4. #include   "upnp.h "
     k9 y* d. `* M! x0 k, Z
5. 
   , [) e  w% Q+ D; b2 A+ _" a$ `
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ C, Y* K( T: z$ h7 M$ }
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' |! T- i; v; ^5 n8 p
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   8 K9 o! G5 P- }/ `+ q& x& G0 R6 _
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    " a/ H% e! X$ n* I, h" s9 A7 M' q
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    8 A) C' H/ l( i4 ]( `9 U
13. static   const   int UPNPPORT   =   1900;
    5 z4 L/ b2 s3 n# X( N) F
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    9 V0 Y. b6 D2 b  ~* I
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    5 H; D. Q! z( ]# o& o: e
22. return   s;
    + ]3 N. ]$ t2 s
23. }
    
24. 
    4 A3 s1 @5 Y  `* O. t+ ]* V3 y) U
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    : h/ K/ B  c) o
26. {
    * z5 A3 M  m; z" s
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    " e- ~$ ]9 R. ]$ Z3 c7 g
28. }
    1 E, y& i, a! V) ~2 t9 J
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; n$ \( T" {3 v# l
31. {
    # S3 B  D; n. O- M6 b% `
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    * s9 b6 E' r) z
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    : ]" n# _. \9 |* q1 R1 I
38. 
    
39. const   CStringA   sa(request);
    ( ~: M* {3 d. @$ ^* e4 k) S
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    $ [" ~2 ]" ~$ `3 V: \0 p
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    , G: Y/ F9 q! T' r4 d. V) o
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ ]! r8 B7 q$ r9 |$ `
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    2 g$ \8 R8 q) |" ~+ G: P
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    # H* W; i& P/ _" ]
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + K& u5 e& O, @  Z
50. u_long   lv   =   1;
    . ?8 n2 o# l0 X
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    " h8 L" P+ ~" ]: e6 x
55. Sleep(100);
    , J: r# i, [: Q9 W& ^2 C
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    0 s" Q* f  S/ C9 \) t" d
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    " c, u' d! U* W+ `, |" z* u
61. response   =   CString(CStringA(buffer,   rlen));
    1 T4 D. G% i; e5 e- q
62. 
    1 P! }  N( F) M$ E1 |2 H# c) L! Q& I
63. return   true;
    2 y% _! _, d% r7 ^1 I9 y
64. }
    . E2 S$ \. R0 m) M) J1 K
65. 
    ! K, ?& _6 h( P$ _' a
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    : ^# C1 \/ Q" V# H. j4 Q
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ; ~0 [8 o1 p( v
72. strcpy(buffer,   (const   char*)sa);
    ( p3 a$ H7 S2 W
73. 
    ' l+ m) k3 H7 w, h7 ?
74. struct   sockaddr_in   sockaddr;
    . Q. j1 J" j7 l$ d. E
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    , ^; y6 Y# B/ y( P, a% m/ ?% z
76. sockaddr.sin_family   =   AF_INET;
    ( L7 D& f/ `$ o
77. sockaddr.sin_port   =   htons(port);
    ! E, Q8 K  u! F3 t
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( s" C: d1 |; |3 \
81. }
    ! X( T- s* R! b; M% M
82. 
    - O- o. l3 _/ C: _
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    6 v/ C5 K0 G# q8 E& D
84. {
    
85. int   pos   =   0;
    9 \8 n( y$ W$ W, Q
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    4 _4 U! g; E8 W8 s( v% P
88. 
    
89. result   =   response;
    , B2 }0 h+ s( c4 z
90. result.Delete(0,   pos);
    
91. 
    . s9 C/ H: O# V' S  B. _; W* a
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    " J4 `0 U/ o% d* w* Q2 r& a
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      x& O( P" V' F) y& F
96. return   true;
    * E: c& x( `$ y4 W& A
97. }
    1 w" B5 v" D2 @: E* ^+ c& {( R& j
98. 
    5 h/ \. f3 a0 b6 W/ E0 t6 P
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    " p  y, m1 }1 R% y4 X) k/ u
100. {
     * O/ V. a( L2 d3 U- r: X
101. CString   startTag   =   ' < '   +   name   +   '> ';
     0 Z: `2 @1 J8 t. O1 T+ ^
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     . L' C: y3 U, n7 S0 Y
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     ! C5 W) [# i; ]# u& u1 j8 e
109. if   (posStart> =posEnd)   return   CString();
     % ]; e  y3 k+ w  H, ^7 ^# w
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     4 ?6 c- U) E. J7 q
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     0 b5 a. e2 g0 g5 M
117. m_uLocalIP   =   0;
     + Z6 u2 g9 Q9 P7 a
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     8 C, o. z) `2 ^+ h6 D; {
122. {
     2 f+ B4 v$ y! L
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ; P' ?0 b' J# n% U/ ]
125. while(pos){
     ) ~' |" m* F* ^5 \" N5 r
126. search   =   m_Mappings.GetNext(pos);
     ! w0 U% C, c9 y6 q7 u( G' t7 }/ c
127. RemoveNATPortMapping(search,   false);
     * M9 X; Z$ W# q+ v( }
128. }
     0 V0 f. K% W% r8 u
129. 
     & ~# i9 X3 L" y7 @
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     4 P' o  G4 m4 Q/ o" R& b9 v2 O
135. {
     7 W: i9 b+ M8 {  C
136. if(version <=0)version   =   1;
     % P, R* V/ h( T" Q. H9 Y( p5 N3 N
137. m_version   =   version;
     
138. 
     - _. X# \1 u& f
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     + }: J$ h& l+ e0 k' }5 M
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     - e' h+ X5 y# v, h$ g2 d1 ]
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     1 }: ^7 }! g: _( f' r0 b8 M" o4 u
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " T: v' o6 v  |0 ?
152. if   (!(i%100))   {
     - N! G% I& Y1 @2 W
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ' f  G; x3 h; q6 [& A9 v1 X
159. }
     / m1 K) h7 o  Y" P. ^
160. }
     6 v& P6 O- j1 V" H) W+ f
161. 
     , H5 q. V( p9 F
162. Sleep(10);
     
163. 
     4 B* N# D- J9 D1 X1 i2 z
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     , C. C% Y. |4 Q5 N
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     1 y# X/ `- e/ G2 E. T# ]; I
171. if   (!parseHTTPResponse(response,   result))   return   false;
     8 m; e6 b6 H0 B! X9 t
172. 
     1 ^; [) _! r* }( T) m9 V
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     # }. `8 y3 ^# k$ Y* o
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     / M7 C4 y3 H& Y0 b: {4 |
175. if   (result.Find(m_name)   > =   0)   {
       D" u' [: Z+ m  k  j9 ~3 P
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     4 v$ c& t, n8 d# b  K1 J
178. if   (line.IsEmpty())   return   false;
     - S8 w! g5 [) Q( z
179. CString   name   =   line.Mid(0,   9);
     % u  [# v8 G. W' B  v/ F& [, y- `: N7 P
180. name.MakeUpper();
     5 H$ a, m/ f: V3 S6 B/ d. K1 z  N6 T
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     / G! K% n0 S' Q4 r1 v
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     " p" d& O( n3 M
186. }
     & k4 c/ l; Z0 |  P
187. }
     
188. }
     
189. }
     $ M) E1 l& u. r! L# K6 G
190. }
     3 l4 c0 e6 k# U! j3 X$ ^
191. closesocket(s);
     & y5 o6 j, e2 I
192. 
     
193. return   false;
     ; y' J. ^$ e% w2 x, S: {
194. }
     ; ^: J& O3 l2 V$ j- t

复制代码

天云上人

以下有关upnp的接口来自emule，
/ Q3 w2 k" b3 C

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
' N5 K. Q3 d$ j( k) V  z) e

#pragma once
  j3 G8 ]  ~% N  U' }#include <exception>


  enum TRISTATE{
( N2 ^) c& ^: c* A+ `6 Z$ W        TRIS_FALSE,
        TRIS_UNKNOWN,
' u0 B" D9 N) [( D        TRIS_TRUE
! J# ]$ }- V3 k% S1 c7 c};

/ z/ e: f, q* _
% w7 o' [& t  R& T) G' Zenum UPNP_IMPLEMENTATION{
) W8 M) V2 E7 y. E6 |- }        UPNP_IMPL_WINDOWSERVICE = 0,
( E' T) c4 S% \+ c        UPNP_IMPL_MINIUPNPLIB,
  M# \, E& A4 S% a; g* P/ n        UPNP_IMPL_NONE /*last*/
' F: |& I( ]+ u};


1 E) t2 |; o8 U* Z5 y
. c( V" C+ F4 {9 n/ |+ T% \
class CUPnPImpl
{
  j& r$ A' h$ j  C3 N, A/ _+ _public:
        CUPnPImpl();
1 W! U0 w- y. n7 c7 V* {        virtual ~CUPnPImpl();
5 U( d6 }* f1 Y& l) e# j% v* E; e        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
7 X. i% K, p) h                UPNP_FAILED,
                UPNP_TIMEOUT
        };
& t8 T# }+ ]0 o' t9 G  N6 O" I# c5 f
7 @; P+ M' y1 L0 b& i& b, j
7 V3 k* z3 {" y! a% j* ?        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
/ Z& V$ `0 [, ?1 R, h' y        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
2 z% T% ]! B% C$ x; [6 ~

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ M" `5 A3 w( u8 M        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


  G% A: v  A# o. ^% l# k! @// Implementation
) {1 q! a7 P) U/ B3 x1 ]. ?protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
7 g& F3 \' m. n        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
" @6 r) L/ b& B4 F/ B# F( G        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

! ^5 D4 b9 t! p# N2 H4 h
/ \' K2 x/ u/ a6 Kprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


0 B; g* M  c9 a! Y3 C};

9 d& x5 o+ M# i# K$ e7 z
// Dummy Implementation to be used when no other implementation is available
, B% @3 ?0 a  E9 ~" [class CUPnPImplNone: public CUPnPImpl
{
public:
# {  `* ~( s! O; q5 z' n* A% d6 w        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
- B) c: t6 A" J        virtual void        DeletePorts()                                                                                        { }
* r" z, t6 e* {        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
) ?2 u2 M. h) B  t( X

/////////////////////////////////////
  }& [: I+ j7 V# W//下面是使用windows操作系统自带的UPNP功能的子类
# t" ?" i+ @6 Z

#pragma once
#pragma warning( disable: 4355 )

- e! x. ^' `+ N( f4 ]
4 N7 l' m9 r2 Q& x8 m#include "UPnPImpl.h"
5 f0 l! ^' O# Y* m) R#include <upnp.h>
#include <iphlpapi.h>
7 N6 b8 p) q/ b#include <comdef.h>
#include <winsvc.h>

7 S% U, e- o/ F3 Q7 s
) s6 a2 [4 S. Q* {7 X) c" H7 G#include <vector>
9 J+ @5 p& ~. |8 C#include <exception>
( _7 G8 V1 O9 L* k#include <functional>



6 V+ R4 l1 v3 C! L& ?
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
/ |9 c2 R( |4 d+ |3 Ztypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
2 r3 _# w' @) `9 {( G6 ^typedef _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;
4 q9 H/ z/ S/ _typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
8 q- K# |& M7 h3 c& Y  {

6 [& e# q5 X* {: _" z" \5 k# etypedef DWORD (WINAPI* TGetBestInterface) (
) O) E2 G0 a# {" B& Y  IPAddr dwDestAddr,
$ f( u7 E  A' C' _; @  PDWORD pdwBestIfIndex
);


/ y! u" v# Q* Ztypedef DWORD (WINAPI* TGetIpAddrTable) (
/ V0 X5 H# b% p, d  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
2 l) e& B% v, s+ ?7 ?  PMIB_IFROW pIfRow
  S0 \5 {7 j5 g' \+ J0 L);
+ J' R; d: T& Z4 [3 n1 p6 v

6 x! J3 a6 t( i! I- O8 s+ WCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
" \( x! q& H- q# l1 d) @

class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
) A5 P* s0 T6 |& w// Construction
* K' s6 e( {# ]" Zpublic:
2 c& J, l2 E3 B9 _        virtual ~CUPnPImplWinServ();
" S1 ?' q; J) t! D        CUPnPImplWinServ();
% L+ Z4 Y$ e  R, K
& q- m4 G! }9 V+ V/ ^
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
! L4 U# j( a: m2 {        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


, h( A2 s& }0 _( I# G* P5 A/ r        // 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; };

' g0 k" j: F2 Z" d- x1 S* o
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
; B1 G5 ^% t- [& ?  Z8 a8 _$ ?        bool        OnSearchComplete();
% Y( B0 Z) K" o: T        void        Init();
9 O# [, f; d3 t; J( S& v  c& W

        inline bool IsAsyncFindRunning()
        {
3 `# q. H. w$ q4 R" ~# o                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
+ C- S1 E# R9 L( w4 Z( [                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
: j  ?2 c+ w2 ~, i4 M0 a) u2 s                        m_bAsyncFindRunning = false;
                }
                MSG msg;
! E/ n! k5 H* X                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
6 I. @+ H1 L  E  }( T# o9 K+ w                        TranslateMessage( &msg );
9 O6 _7 \) m0 n! g# p/ N# D                        DispatchMessage( &msg );
) p7 b/ D; p& ?4 M                }
                return m_bAsyncFindRunning;
+ o: L9 E& B6 h9 @' N1 k6 F" A        }


( J( k" [$ Z" ?( K- \        TRISTATE                        m_bUPnPDeviceConnected;
* i) L* T2 a3 i9 H/ J5 G  @; S

& ]0 `, ~' u1 ~  z// Implementation
        // API functions
) w, c6 n9 ~" I% }& n6 O+ s# W        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
0 r7 {! l1 H  x: e2 q( @+ u" _        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
0 @6 z/ r# `2 S/ `5 t# D6 X9 h        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
+ n. |9 R. l7 |2 E& ^        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
) M! \' P- `2 \$ O% q

        TGetBestInterface                m_pfGetBestInterface;
' L2 X5 g3 |( b; Z        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
6 O  l4 P% b) \
, z7 i* _: B: U/ k; v+ }0 ]5 m
        static FinderPointer CreateFinderInstance();
1 U' D& g, i+ d6 M, h9 x        struct FindDevice : std::unary_function< DevicePointer, bool >
# s- n- L- z! |        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
6 q& @1 R; E  |- l                result_type operator()(argument_type device) const
0 i. J4 M8 r; G& T- _7 f) H                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

; v9 m2 W5 `! Q/ t6 O1 y: y  k8 }) _5 d
2 [* a) M  \6 R2 e* F+ m                        if ( FAILED( hr ) )
) K! ^" Y2 k; o4 F& l( i                                return UPnPMessage( hr ), false;


. G1 n% {9 J$ `/ N9 ~$ i                        return wcscmp( deviceName.m_str, m_udn ) == 0;
9 G- t* u  B9 r, K; i1 K3 y                }
% F( B% ^4 l- {1 v) h! J                CComBSTR m_udn;
& `4 B1 I( r' f3 o( X/ {, _        };
       
9 n" b$ q2 u7 P        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
) \( Q+ t6 f/ d" Z        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
" Z7 S( I$ d* y( J, {
; e; i5 q" c( \' C
        // Utility functions
$ k( N2 E' }  t) b9 G        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ x# e6 }5 _2 v        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
5 m, _& {3 E8 {        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

* N2 M$ s  d3 z7 E
// Private members
$ J0 D; G* Y. U  Z( c  e. M1 b. bprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
% i$ u# _* v' b- b( H        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
% ?8 w" X8 f0 h/ m; e0 F        ServiceCallback                        m_pServiceCallback;
$ t& O  n0 D% }* L
; l: T; u! F, Q9 u
4 p: F, O$ x( R# ]0 E        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
3 ?1 |3 }/ r6 y; D        bool        m_bPortIsFree;
$ M- H$ g0 ^2 w0 I( k5 ]( g        CString m_sLocalIP;
2 f9 z5 U9 }4 Z. G2 ]0 X        CString m_sExternalIP;
; }, x7 T* a5 b5 Q$ c* Z) v        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
! v' N& t# \2 _6 |. I3 v        bool        m_bAsyncFindRunning;
# F' p& p  ^3 Q% Y* O! y' Y: p        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
3 K* @* k; D  o        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
" a+ z- |4 E* o+ a8 a1 w        bool        m_bDisableWANIPSetup;
1 `$ Q4 O7 e, ^# J+ ^        bool        m_bDisableWANPPPSetup;
/ `3 g5 |% ^# _' [' D+ A# H! W

5 ?& h& }5 b! V0 ]};
9 u4 u3 B3 N3 D4 i

// DeviceFinder Callback
class CDeviceFinderCallback
  v6 d9 i# K' w9 b        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 {  D- m- M; d                : m_instance( instance )
2 Y& T6 ~- [6 T- Z: e- x- A        { m_lRefCount = 0; }

# g, |6 T+ r+ n+ Y6 V  ~; K% m0 _6 Y( Z
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 e* s6 j6 x( H" x: }+ e0 V0 l' e   STDMETHODIMP_(ULONG) Release();


// implementation
6 C& U% w: c- Rprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
0 `9 H8 ?  M. V& d- P- Q" q- Z* i        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
. j% L" R% t* `& e+ n4 \

9 G2 y7 W1 i5 q7 r, eprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 L; Z3 Y9 k( B};
1 W% }% t  p- x: p- n
; B8 m! D/ U' n7 u1 j
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
# H0 x# K" Y& A% w4 _5 ]1 Spublic:
3 r) ^) D1 c$ n+ y3 T        CServiceCallback(CUPnPImplWinServ& instance)
- B% u9 l9 _# r$ W& M& p                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
* _; K2 w/ v! e2 O2 }" Z   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
9 e  a6 x* g- ]- y" L$ d
) R7 D5 s8 Y2 O; n( |
// implementation
3 ?2 L3 O& F( ]% \5 @! Pprivate:
- A- L! ], S& Y        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
. I4 s8 j/ ~. @  p8 @        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

3 ^9 D( }5 E# ?# r" ?3 b
% B3 U2 p6 R( T6 iprivate:
3 b! K7 s. c' H        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
/ ^& ~2 S$ v/ t5 G
* g' g/ U6 v* X+ w- Q7 ?
/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
- q' i7 P+ S7 \2 ]$ E& wCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
- ^# O- r7 M+ ?+ ?; \) T; l+ V+ wCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.