UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   # [' e0 e% F4 v( T. v  w
4. #pragma   once
   & l8 `: P+ q/ \. p4 }
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   ) l" L8 ^" {! q( U
9. {
     b  y: e. _7 Z- Z2 f+ w5 P& N
10. public:
    3 a; ~8 @( \1 i; @, s) e* J# I
11. typedef   enum{
    5 V4 k7 G1 A" c( Z% E% }, G1 T! ?$ h
12. UNAT_OK, //   Successfull
    
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
    0 \) ~3 C+ F. F
17. }   UPNPNAT_RETURN;
    * `( k5 {: }. P# e
18. 
    . L9 Z# w! q; l6 Z
19. typedef   enum{
    . b2 n: B1 ^# Q6 n
20. UNAT_TCP, //   TCP   Protocol
    # J! Y2 j' ^4 ~5 ?" U
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    5 z3 X' I+ [2 |; `0 w4 d$ ?
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    0 d3 u8 y6 V) q/ Y9 m5 v: R
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
    / Y- w, U# V/ R& o5 s
29. }   UPNPNAT_MAPPING;
    / r3 f+ U7 B+ @2 C( X! B- r
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    ! h! e: [: k* |5 X3 ]/ q1 }. |! S
33. 
    5 v4 p) y: D0 E1 x- Z1 E1 j
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    : W# N0 C/ z" |
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    . Z7 n- n/ \& V5 R% B
38. CString GetLastError();
    0 [# i- u* M1 h1 p9 {# ~7 L/ B6 ^: D. E
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    1 A; R/ T5 g# L/ K: H* S
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);
    1 F0 K( I- Z, E: e! ~. M
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ; }1 g6 c4 g. Y; A+ [7 b/ V
52. 
    
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. 
    . g& V. c  X+ b2 b
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    8 c& |9 A" J$ e+ _( s  A4 s
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    ! M: k% ~& x9 c( P0 g- T+ Q4 H
62. CString m_description;
    
63. CString m_baseurl;
    3 `: D. h: N; N' ~
64. CString m_controlurl;
    8 R$ o* L  \- d
65. CString m_friendlyname;
    2 F6 n% m, e. R/ d, F) D% c2 _
66. CString m_modelname;
    
67. int m_version;
    / X% D# @& m. r% ?) ?) A$ x  u
68. 
    4 F5 l8 _4 V" t% f; O* D. I
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    : F0 e/ l% Y! X- C% }; z# f
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    , A+ R/ R. u( w
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   5 x, H7 X) k8 D$ v- k
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   - f7 G5 R9 M! Y# u: y) a
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' d# H7 k. m. U3 n7 @: U
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ! F( N4 [% Z3 G: G' O% h1 x' F0 K
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    # t& U0 H7 N9 O  e( q, _
13. static   const   int UPNPPORT   =   1900;
    * c0 I9 n' j0 e) a6 Z
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
      G$ u  I* S: _0 Z( I* G
15. 
    ; `- V& ^- F8 L
16. const   CString   getString(int   i)
    8 ?! ]& |/ z% G$ b! q
17. {
    ( `7 M( e% g: [! i5 H; t
18. CString   s;
    % ~* z' c- M" M( M2 |/ w
19. 
    * D7 H8 Y2 h* W% C6 k" ?2 d% h
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    / T0 S5 v5 Q9 z/ S2 H- M
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    8 U  c4 U5 e5 l. v: J" T. |7 M
29. 
    & i+ m% V5 q* R$ q. O
30. const   CString   GetArgString(const   CString&   name,   int   value)
    # p! Q% }7 Q+ Y, ~6 p( q5 w
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ' L2 i* Y. z, g/ \0 y6 s% W% t2 N
33. }
    2 Q- t& `3 F; |4 E6 R; f
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    5 T6 X. @- P+ a& h( E7 J
38. 
    
39. const   CStringA   sa(request);
    , G) Y% o0 t' w. h
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ! E% p5 q* u9 e% N( r! p# T/ h
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    9 |9 j9 `3 h" i$ i; |6 g$ X5 U. h: f
44. struct   sockaddr_in   sockaddr;
    0 T$ }6 m4 [1 r2 n' h& n
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    - m7 ?$ d7 A- C# M- t5 Y
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    . Z& p7 J' x- n
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    5 q) ^1 Y% {* s
54. int   n   =   send(s,   buffer,   length,   0);
    5 j7 l& N. o2 }) D# J
55. Sleep(100);
    , o% G& o/ W/ }' O) y
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    . L/ u1 ~* ^* s& _) }1 I1 a4 ^: z
57. closesocket(s);
      V' Y8 M" b% G
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    - _( o: o3 k9 I) C+ G! e
59. if   (!rlen)   return   false;
    ; A! v; z! @) b1 v/ _
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
      v3 F. o. V) C$ `! v9 v1 ?0 l+ s
62. 
    
63. return   true;
    
64. }
    
65. 
    ; |$ k0 b- l  H8 _8 U& F; D9 q' B
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ) \& S: v+ n. G
67. {
    * @' Z. V; A7 u5 a5 f
68. char   buffer[10240];
    ' t) |; e7 {5 h
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    * Z' V4 |8 N! `9 n0 K9 r
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    , c* D+ p7 e  g/ @2 E
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! {9 f% j! A! _! V8 E
79. 
    , }+ o6 t& A8 l3 ~$ z! \+ f
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    & X8 c: v2 z; P/ E$ z6 m
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    5 g$ T$ H+ P! Z: b4 H, t$ N
84. {
    
85. int   pos   =   0;
    : |% J5 z( W; W4 Y8 F$ P
86. 
    6 s- j4 z& N$ H3 g$ _$ c' X9 s. k
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    7 x( ]& u& X7 u8 N6 ~' X
90. result.Delete(0,   pos);
    ! r- J3 d* ^/ q) M6 h
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ; y1 X' p% b" L; l; l- b
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    0 h! y& d0 p, X# g+ p
97. }
    . `0 S3 Z5 Q3 u$ I1 y- ~
98. 
    8 m/ {5 @2 E9 f) J4 d0 a  ~
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     + o/ m; a/ d4 k2 C7 J
101. CString   startTag   =   ' < '   +   name   +   '> ';
     2 o$ [5 z8 z3 n) W8 [2 M
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     $ {! z. m! I& O0 h9 w8 `
106. if   (posStart <0)   return   CString();
     
107. 
     * l& j! s5 g8 ?+ S" o$ b
108. int   posEnd   =   all.Find(endTag,   posStart);
     . a' h2 y6 d- m/ F+ g  V, L
109. if   (posStart> =posEnd)   return   CString();
     % Q7 h& {$ `; k4 _$ w# H% a
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     6 z% A1 }& x3 X; e: V/ v
112. }
     9 z& O# y# [3 l$ G0 u
113. 
     " A' ^+ J0 d: e0 H4 ~& }- k, J6 g$ i. y
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     5 F5 W% w; w3 F( T
116. {
     * l# f2 L2 z/ W5 E
117. m_uLocalIP   =   0;
     6 B; C$ S2 s4 ~! E, @
118. isSearched   =   false;
     ) Q. G* g  w. T: {  i6 q; S, a! a
119. }
     
120. 
     ) s0 A4 ~: i) h! W1 K
121. MyUPnP::~MyUPnP()
     1 f2 i& m- X7 M. R2 y% x) H4 Q/ A
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     ( W' z' e, w( N' I) Q7 k/ g
127. RemoveNATPortMapping(search,   false);
     ; s& X+ J8 _2 H6 B1 @% C
128. }
     0 C' K. Y8 H/ P( {
129. 
     
130. m_Mappings.RemoveAll();
     " J2 m3 e* v# Y
131. }
     3 s% c$ J- P9 I1 q" i$ y/ @8 f6 P. M: y
132. 
     3 g0 J  g  M/ a9 E
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     $ @0 n9 c% Q, k+ E, P
135. {
     
136. if(version <=0)version   =   1;
     4 m, ~7 \# Y! G) G: {0 C
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     7 O! X; |6 O1 h* D2 Q* P5 n
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     * g" \: s7 ]+ X  F
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     # p& E& m% L* I( o% Y& P
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     ( n4 X4 M) X+ `2 ?5 p
151. for   (int   i=0;   rlen <=0   &&   i <500;   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);
     1 c9 F3 Q) I% J) m
155. CString   request;
     & k* e) t/ T* d& v# W1 V( _2 o
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$ u& a7 v  w! R& z0 }
157. 6,   m_name);
     1 @8 Z3 a2 k$ Z( j
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     9 w1 Q7 d# E% p" Y2 w$ K1 c
160. }
     8 N& o  _# w9 }1 R
161. 
     
162. Sleep(10);
     0 K6 l# ]2 X5 \, `  X
163. 
     % e2 k( ^$ z3 r: t7 z1 M" M+ C3 [
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     + H9 @- z$ m& N7 a2 [
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     9 |0 C* @# n9 B) ?: G# S
168. 
     - y. X' O0 @/ Q3 F' M; ?! {
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     3 d1 a4 z- T( e
172. 
     5 [; _& u& l8 A6 G, `; L
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     " b* d: P% d" ?" p0 `
176. for   (int   pos   =   0;;)   {
     " J! s* E" o0 B$ k1 T
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ) M% r8 a! y; a% }3 L3 Y8 T
178. if   (line.IsEmpty())   return   false;
     . I& t, h4 a0 [3 p9 m# S9 U" C
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     : a* s5 T  [4 ]# p* b0 P' f
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     ! e0 t+ e9 ?; U8 S$ U( X$ P
189. }
     
190. }
     
191. closesocket(s);
     9 Y3 \& i/ h" j2 {! h& x  F% W
192. 
     
193. return   false;
     ; k) n* _3 m; A6 \8 u
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
7 n1 U& _8 `0 e  ^

* |3 l9 c+ R" G///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
  @4 j1 B5 c4 c

! g" o. ?* b0 D$ N& D$ E& A#pragma once
#include <exception>


) R' @0 `! D* E/ v0 w: [  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
# p5 @0 ~$ S- ~; w. b/ W$ `        TRIS_TRUE
};


% I  _; U# Y8 [8 Q3 t& xenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
/ c; U$ x" e5 q; ^        UPNP_IMPL_NONE /*last*/
0 m) f, [0 ?4 P( ^2 \};

& V3 k$ B- g9 h: F/ D# Y0 a8 P# E


class CUPnPImpl
# u' _. c; K1 n/ \7 _5 f{
4 d. n& m/ O3 I, `4 hpublic:
4 x- V; T2 x+ R7 S        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
. Z7 X0 y+ H  A$ G! g$ n        enum {
9 {/ l$ e3 L0 w; g9 ]* e/ d9 i, u7 [                UPNP_OK,
. F$ P/ M4 b; b. w0 j6 o2 ?                UPNP_FAILED,
                UPNP_TIMEOUT
4 v6 d, J. H0 D$ O        };

. A7 W; Y  C) W& K) F: R" J
! J# \& q; X( Q) W% `        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
# g1 _6 q6 q6 h+ {* f" k' e        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
- r( t/ u# l' H9 B- a        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

  v7 i# ^. A& I- O
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
) F( Y5 Z  a3 X8 \# Y) H        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
+ d5 m- O( i. P; m$ P8 S9 C        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


* V/ m5 \9 D; _) N// Implementation
# E4 b5 ~4 x* e$ w: E: |7 t1 t2 bprotected:
( y$ F" k9 c8 ~, |/ `. ~$ A* e% w4 K6 R        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
/ A3 |! e+ K& N1 x: m        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
  Z" m4 L3 F; o- c4 V+ c1 y

/ n  j$ @2 Z( D, Y$ P; Gprivate:
        HWND        m_hResultMessageWindow;
& v+ G7 U2 q8 h, q% z0 n        UINT        m_nResultMessageID;
+ t2 _$ q( n1 U. R2 Z- t2 K0 D
% T- o, ^( |) x; U* M
, j6 S+ A' t. ]$ B};
8 r" _. n& W+ b) O  v- R, s. H  o
+ V- _* `& D/ o$ F/ ]
, q# D9 w3 r' K% v2 X3 y( {3 E// Dummy Implementation to be used when no other implementation is available
2 G  s; D( r6 o. Nclass CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
8 V% B# M% O3 B3 f        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 T% s0 G# B& t8 D  \        virtual void        StopAsyncFind()                                                                                        { }
7 D* T' v% v" K: o& w- o        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

" O( X1 K! b& v" h4 g  X
/////////////////////////////////////
, Z( M' U  i+ w* a  i0 O+ G6 J: X% t//下面是使用windows操作系统自带的UPNP功能的子类
/ @4 B9 c( }! l) r; T

: Y) |; w0 o" N7 k4 R$ {! `#pragma once
7 f8 b: h" r( B& m; t/ ]#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
4 f3 v, i2 U  ~2 p1 M6 Y& T#include <upnp.h>
: y% L& h' L7 L' J#include <iphlpapi.h>
% Q0 l2 |3 y4 ]+ \#include <comdef.h>
#include <winsvc.h>
% e4 ~  F  G* N  W( F% ]

#include <vector>
  b5 ~7 u+ |6 ~; D' I#include <exception>
#include <functional>
% N; Y# k$ L" z



$ o- J+ b0 X- w8 Z: R1 qtypedef _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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
; x7 d- z/ z+ _typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
0 e2 B$ _0 Q% Mtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
2 e% G: ]' [) I) B+ c7 E

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
3 Y, s% D/ ]7 b, G: r$ W);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

1 |5 k1 q+ d, S9 }% t
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

9 h6 t  K$ v( ~9 [
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
9 t0 v' v9 x+ a+ p7 T// Construction
3 O: v3 g! a1 q) B7 Zpublic:
        virtual ~CUPnPImplWinServ();
& \" y+ P) O% K8 m+ \3 I9 I* M5 n, k        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
- E! G6 r" Y: L3 B# w& Z) T        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
% V5 e! n; W. k0 P
% q* ]$ S1 _, _* l7 {
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
5 `5 V8 i$ L1 n8 z! ~        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
( w, S' @  U6 Z8 [0 M" `4 G( B. d        bool        OnSearchComplete();
        void        Init();

, w$ w& Y2 o8 e
$ D% D( n" ~5 u# ]* K# `. D; O        inline bool IsAsyncFindRunning()
" K' f4 u! J/ {8 g; y3 F        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
1 A) ]* D" ?+ F3 n                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
: c, y3 G! w  U- i. B                        m_bAsyncFindRunning = false;
1 j+ ~8 Y8 p; N- L5 o" E3 z" V' M& w  q                }
% r+ H3 a3 n( U. ]- F                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
7 g% ]! C# N6 c4 P                {
: I7 q" ^6 ^% K4 L1 f                        TranslateMessage( &msg );
/ K, |2 d' j. d- M# T! r* Y                        DispatchMessage( &msg );
                }
) x) \- k6 M$ Y( t% V0 F9 F# w                return m_bAsyncFindRunning;
$ o. A5 }9 z% c3 W7 I, R. S        }
; v! f) M& M' {6 i3 w

        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
. d& x! @2 s7 O% i! |, U        // API functions
2 K3 U/ }. X, U  t8 e- G2 v( E/ y2 P        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
& r! `+ w2 c5 Z3 K        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
8 V8 b% h& T. f$ D; b        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
$ w( {2 G/ _" d3 W: _        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
& Y" `" a0 }1 ?! `8 Y! C, z$ K        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
; X( Z" U* A5 c+ M
. ^: o) x- H# x, b$ i/ z3 d; r
4 j2 y. J6 S( A! J        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
6 Q  Z4 z1 ~: N4 @8 `) H, e2 U        TGetIfEntry                                m_pfGetIfEntry;
. b) b5 T9 A* f
/ @- N# X0 _% _% m; x, v1 t- Q
; E  c0 T% u$ X+ w        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
4 ~! `# N" U- R3 ^; Z$ f        {
5 O3 c3 h0 t: T& E                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
1 y$ V6 ~  |" A- E6 h3 n5 r, d                {
) k8 t% j6 B% v$ S. x                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
' v" j: S5 r  b: n$ P& J
2 x3 I5 e3 V4 d
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
; |" D; r5 g) L, ]( B                CComBSTR m_udn;
0 T' P0 ]+ {. i* b( ]        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
3 M4 [2 f" j5 |" `        HRESULT        MapPort(const ServicePointer& service);
1 x* S; f! y" o% u/ S        void        DeleteExistingPortMappings(ServicePointer pService);
+ _/ W2 C/ ^# C3 G. f        void        CreatePortMappings(ServicePointer pService);
. Y# P# u; t  h& x% U" b        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
, C" `0 D" T' e  ?' i/ J        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
- C" b2 u% |" Z# _  y( B        void        StopUPnPService();

# j. C; |8 |  v5 q
4 T8 r7 y4 p8 u9 `; v        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
& ], ^  X8 N$ `' [

// Private members
, `8 X9 u6 W/ i7 ^$ N$ Eprivate:
% d- `+ I" }; A4 e+ f        DWORD        m_tLastEvent;        // When the last event was received?
# M: j: I/ A- ]( A7 `6 S        std::vector< DevicePointer >  m_pDevices;
3 [( q1 R/ P  ^, B        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
9 i' h9 m4 T5 q' _( `' m+ Z, y        ServiceCallback                        m_pServiceCallback;
, V! Y9 A  r* }9 h% M3 Q, l+ H
5 X! a% T, @3 ?! l, m
# [& m4 A3 L9 G: B& q! f7 F$ O        LONG        m_nAsyncFindHandle;
7 S* M: L1 A9 [: [: ~! l2 V) d        bool        m_bCOM;
+ T8 h2 H# I* Z9 H8 N/ X  A        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
. O/ [- J7 h' I7 p0 O# [; ^        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
' I( k, C0 g" c        bool        m_bInited;
% B/ S$ O$ T$ `- _        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
# e, ^5 [8 n  k( e        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
! C' t* E' v) t) w' G: K6 W        bool        m_bDisableWANPPPSetup;


};


- Q6 Z  R4 F) l. L0 {& O& E// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
/ ~8 y7 t; F' @& Tpublic:
8 b0 P+ p; J) F        CDeviceFinderCallback(CUPnPImplWinServ& instance)
7 t1 b9 a6 n! v: h4 ~7 _5 u                : m_instance( instance )
. }3 W$ W. a4 K% S7 K" s$ B  F- Q        { m_lRefCount = 0; }
; w- u! |8 @. A6 {

3 _7 F' W6 x$ K7 _   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
! T; K+ r. I+ E8 V7 H  v  F   STDMETHODIMP_(ULONG) Release();

& H7 m) n7 u  a1 `
: {1 I2 X' j! b4 l( V9 S8 ~// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
6 g/ |% g( K# _8 L* `$ E; ^+ `        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
* C, _. d& O2 v        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
& @0 D$ `7 b: x& v- |! D        LONG m_lRefCount;
};
5 T' U8 [3 `9 f
4 Q7 l5 ^* t8 w, o4 o; v' ^
$ p5 _* q$ `6 ?6 D! F0 u- m// Service Callback
7 N- w1 P4 \$ |3 k1 Uclass CServiceCallback
8 q  a. k6 S8 u" a; ]2 q        : public IUPnPServiceCallback
; m& I. P" U$ e- [2 M. o{
public:
3 r: m2 x1 Y4 z. a# D0 L' G/ {$ q& j        CServiceCallback(CUPnPImplWinServ& instance)
7 j' N# N! o# N( ^$ x+ \7 E/ t                : m_instance( instance )
        { m_lRefCount = 0; }
4 p/ q- T" L6 `" V- W   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
private:
+ X* k! L8 _, \3 U' J        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
, ~- g: D& ?7 p" I* M        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
- S7 V; ^9 Z& w( S, Y
& ]9 L" W) w8 J- ~  W- |/ d* {
4 D( H% L+ Q( @: E/ N1 i0 a8 J- J/ hprivate:
2 C+ g- k$ ]9 r5 ~1 u9 X        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

9 f" Z5 B) n5 L& A$ z, i
/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
, I: {9 [; C+ q, x9 a! n% iCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
/ @* i6 I0 [6 ^6 M* ]" a5 uCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.