UPnP

天云上人

/*uPnP.h*/

1. 
   + J: I' ~: U8 w/ R! i' D. U" P
2. #ifndef   MYUPNP_H_
   
3. 
   / o& Q0 i4 {* q& u& a% r0 v
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
     D$ P8 M8 \6 i" I  @9 N: c; x; ~2 I5 f
10. public:
    
11. typedef   enum{
    
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
    ' \: P) [3 o& z& k3 \% X
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    5 O8 Z- T' R2 X
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    - i; o2 j! _4 m" t! K* ^
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
      _6 h1 r0 F5 p% Z0 x/ X3 U
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    # g4 @2 ]- k; \' H; C, l
22. }   UPNPNAT_PROTOCOL;
    + p; s4 Y: j+ l; q) a
23. 
    8 C* ^6 ]9 {; v" ?4 w
24. typedef   struct{
    # D1 I) f& f; {5 D5 K# Q/ E
25. WORD   internalPort; //   Port   mapping   internal   port
    2 N# v; G% }/ j- v7 j! @3 y( Z2 q
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;
    " O) k0 y- y2 K
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    / c, }/ A5 P0 q* C
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    & Y" C' _" s$ I" J* s
36. void   clearNATPortMapping();
    9 y$ P+ L( y6 E9 ^" p
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    % }9 Y9 |) f" m* m: O; R
40. WORD GetLocalIP();
    + x5 ~/ E, i; i) u
41. bool IsLANIP(WORD   nIP);
    7 c6 W' a1 ?- I# j& ]. O) T
42. 
    9 H- S0 |" _' U& K
43. protected:
    ( A; m/ X7 L4 W) w* h( W
44. void InitLocalIP();
    1 b1 l$ Y# K. C% \# D3 T
45. void SetLastError(CString   error);
    . G  _! s5 B. U1 R1 {. d3 ]
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);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    0 o8 K3 `% U1 {' C+ D0 r7 O- ]
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    - ]9 J: L$ @3 r7 T
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    * }8 B$ Y7 n$ V; X) A7 x0 x* H
57. 
    % v  M- R! E8 I
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    , ?8 B8 y; _, W! H& d: u
61. CString m_name;
    0 q4 x7 i" |& H4 s& c3 |
62. CString m_description;
    * J$ X& X; ~" _5 u8 c
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    0 n' G, ]& A' h# v0 C9 T
72. CString m_slocalIP;
    0 d2 q# X/ L5 g' p0 x* j+ `( ~
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ' C6 m+ j% q6 e5 p/ x& @, L) }
75. 
    
76. bool isSearched;
    
77. };
    " H1 \! x  Z& E" P( P
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   % A  E  h& ]& T- p+ _! k! o# H
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   3 }0 Z# p2 W+ y  x% Y0 ]( U
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   4 d6 W8 X/ E% |, ~! D* e
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    6 H' v$ _" f  _0 U
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
      Q/ w+ p# a1 F8 j3 ^
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ( z1 j. r9 z0 z( O, S" ^6 X
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    $ u. l# G& k5 O' V3 ^: P; c
22. return   s;
    $ A% e) v# @1 O' H% O" I0 y9 W- T
23. }
    : W: c/ a( I0 B+ h; d( a3 G0 i* `
24. 
    6 }* T5 u* S4 `8 w$ z) ?* g# S
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ! m+ d) C1 [4 [
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    0 ^: H6 ]" k8 r
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    & v2 b9 H7 |+ l& z0 F3 N
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ! P9 f% q0 I% O/ O5 p( F# v
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    ; u' j+ n8 Z9 s
40. int   length   =   sa.GetLength();
    5 g0 M- o! i: B2 {
41. strcpy(buffer,   (const   char*)sa);
    & t7 i, p( f6 t+ X  |1 L' k, o( T+ S
42. 
    3 `2 f- q9 B( T9 j* n5 \; l
43. uint32   ip   =   inet_addr(CStringA(addr));
    ; o* z$ t1 \1 Q: t* m
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    & }2 a1 b' Q% W" g- W# Q, Y" ?8 r: M
46. sockaddr.sin_family   =   AF_INET;
    # p* I7 ^; @8 K1 K  {5 @
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;
    * X- i8 o( f6 R' x5 i# w0 P
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; P! Z" @3 d% _3 ~. U7 r
53. Sleep(20);
    : {! y; @! w& k8 B; ^& a
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ D( Z1 u: m1 j/ x
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ! }6 u. L2 L7 J0 D- o& E
60. 
    ; i9 ^1 w& X0 K% j8 R: W$ {/ }
61. response   =   CString(CStringA(buffer,   rlen));
    0 J- O5 w4 H3 ^. C/ \# k2 w" Q
62. 
    
63. return   true;
    7 t2 }& A( H- P4 `) u
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    ( y5 y% |* G$ X9 U4 Y
68. char   buffer[10240];
    ' d  N+ I$ E0 ~' L+ h- [/ u/ u
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    ( s) K; ~* j4 {
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    % q/ X$ W& e$ H* C3 f+ ^' h
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' A9 e. s2 j, u! A
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    / ]  _$ s+ [% r2 G! x( u" y6 K: z
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 t6 i/ `# x9 T/ w# Y  v
84. {
    6 T4 k3 ^7 \4 l
85. int   pos   =   0;
    
86. 
    . {  e; e' k! Z) ^3 A
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    : k& o4 T" n3 _3 ]/ j
88. 
    
89. result   =   response;
    0 s; \* g. x( r
90. result.Delete(0,   pos);
      g9 i, L3 b! s# c/ s7 y
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    3 V; t4 T% D% |- w: i+ k
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    & y3 [8 N; A" a5 _( M7 i7 A! l
96. return   true;
    . {( u# R2 {. K( X
97. }
    
98. 
    2 Y1 P& r8 @0 [; e
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     : j& R3 z' i/ x+ m
101. CString   startTag   =   ' < '   +   name   +   '> ';
     : Q& {' ?2 T7 `5 M
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     / n/ w: x+ z/ \$ C
105. int   posStart   =   all.Find(startTag);
     . ?# Y: S/ O( i* w- G9 ?9 g
106. if   (posStart <0)   return   CString();
       v3 X! o1 x& n! y0 I! h& M* D
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     % y* s# O8 o8 P# H! [
110. 
     2 o: {2 R3 R6 \& l- u
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     6 }% b; B& }$ {" R" n4 s; z
112. }
     ' u, D8 {$ c7 A& f2 Y4 [8 V
113. 
     2 Q. v0 k) A# h
114. MyUPnP::MyUPnP()
     . [, F0 V% w8 K; D$ |4 r
115. :   m_version(1)
     + O3 P6 H0 W* j, N0 m
116. {
     
117. m_uLocalIP   =   0;
     ) t. ]2 t5 u2 o& S( s- _
118. isSearched   =   false;
     . J& X5 _% K1 p' }
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     ( U- z1 S$ I+ E0 n# B
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 r9 G, P; }/ S- M
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     & z% {* t0 W1 M' k" Z* F9 e& B8 [
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     ! W7 |% A! B* F8 a
137. m_version   =   version;
     5 }3 t& j4 w  [
138. 
     + X4 s# H6 T) l
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     5 H4 j, p& @0 Q
141. {UPNPPORTMAP1,   _T( "service ")},
     5 m/ n/ q, r# Y  F  I
142. {UPNPPORTMAP0,   _T( "service ")},
     ( W7 W# b( I3 v" X% o( J
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) I0 P+ c" P% z7 E
144. };
     
145. 
     ; P' u" H$ i* C$ n9 f! E4 r4 c' n
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     " Y# |0 k$ Z# r; i' L# j
147. u_long   lv   =   1;
     + K$ a5 v% v- A2 e5 e3 s  d3 V  O
148. ioctlsocket(s,   FIONBIO,   &lv);
     7 l/ Y; v+ _/ \; @& O4 S& f
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     - ?1 u$ r2 w2 f) n6 ?  }/ s6 h( _
152. if   (!(i%100))   {
     6 R, T: d- i4 p. |6 F. d
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);
     ! J0 Q+ F# V9 s4 G
155. CString   request;
     6 ?, o# x" n$ C! h* B1 O5 k
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 "),
     5 b8 u/ |/ m/ R" [$ g
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     5 y5 t0 g8 y3 b: T7 k1 m% n, `! u
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     # f1 W+ z; l1 `# {, H
164. char   buffer[10240];
     . D; c! [; {/ y) d( ]
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     # M3 w% i. L$ R$ e, `* `8 ^# _* Z+ s
166. if   (rlen   <=   0)   continue;
     % d+ i! f+ c9 L- S) L% r
167. closesocket(s);
     " j2 h* s* y  k+ A7 l
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     * G- C; j+ b1 r. i5 s' e; c# k/ L
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     ; x. J  F' c% j( X& M
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     2 Y2 s* d) y, r' P4 f5 ~
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     8 ^0 {" s! y% p$ f3 ^
176. for   (int   pos   =   0;;)   {
       ^, K* A/ p. v5 @$ c5 C! s
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     # `4 k: N. i* h$ {% i
178. if   (line.IsEmpty())   return   false;
     , S4 B; b/ ]0 ]7 Q- y- U
179. CString   name   =   line.Mid(0,   9);
     3 P. l0 b* n) p! H5 K
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     - B5 r, v5 M$ C5 ~# z; k1 {! |
185. return   GetDescription();
     
186. }
     : ~. g4 s% R, R+ r) I& O! ^+ b
187. }
     
188. }
     
189. }
     & z6 G7 g- g8 }. E5 e; a6 D2 R
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     & L/ G( U) e( Q2 [' q
194. }
     

复制代码

天云上人

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


9 Y# q0 ]* g  L& T///////////////////////////////////////////
/ X: g) S" ~- Z% P: S4 V3 B8 V5 {//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
#include <exception>

# n% T& ]: W2 A( z/ _+ T3 L9 g
  enum TRISTATE{
2 a+ A- a& @6 [- j7 n        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
- i1 D3 _4 y3 j6 g" q* A};
6 l# C# u! i* H: p2 E$ a3 x

8 g$ G' a* y" u( Y1 qenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
3 L6 }0 {, ?+ b0 s% r( c- P};

, k6 p6 v& n% y

; {4 e# r1 @& m2 v9 @
class CUPnPImpl
+ U. b( V: M% ^* _{
public:
6 ]; b/ b5 u: k) T8 v        CUPnPImpl();
$ h) b- y+ y4 p+ v8 Y* t        virtual ~CUPnPImpl();
3 v. A# T( _( x* `7 ?        struct UPnPError : std::exception {};
        enum {
- l) s, T3 s+ c+ t$ j                UPNP_OK,
                UPNP_FAILED,
  Q3 [# H6 p$ }. U9 t- M9 D                UPNP_TIMEOUT
        };
/ n( i8 \/ ]+ x) m1 R
0 Y( T2 M* }# {# P0 o
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ m6 y3 I  F) r1 c        virtual bool        CheckAndRefresh() = 0;
6 L3 d4 k/ L' t0 i9 d7 B" D: `5 }        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
0 r2 H, t; i5 |        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
: a- }; K) }# ~, |, S       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
$ {, g) e$ Z2 c7 u# w; @
; V0 ]" }9 ~4 r
+ C2 ]2 ^, C0 [4 u        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
' \3 p: D) y; d% a: Y        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 S2 j7 \1 W9 d6 F  F+ w6 Q* ~4 a# o! w        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

! ?3 p3 H; \/ W' D
// Implementation
protected:
, @+ w4 F* t, j$ m" ?        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
1 i' h7 R" U1 ~0 c( J        uint16                                m_nUDPPort;
1 d$ d1 V% l! `        uint16                                m_nTCPPort;
. s# L! A6 v! V8 ^# c        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
) [5 V* c' _8 v

private:
3 B8 H$ U7 b% _. Y, i) ]( c  ]8 J        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


: S/ e0 X+ `3 w/ z};

2 J, w" Y; v( v1 N: E5 D. R
3 V! [- J: [1 ]2 s$ [// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
" C' E, `  H+ ~% O{
4 ]. |4 V5 E( l) g3 S: ]: U  cpublic:
5 X6 W* u) C3 v6 |        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
; F+ n$ [/ Q) m+ Q, f        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
6 O0 Y' q. @8 s: Z! p$ _8 B        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

! ?1 f9 l. k& ^  }/ V6 z
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
& o/ g3 c4 Y" ~+ _8 M

#pragma once
" t. t5 b8 y; q8 b% U9 u  B) S1 K1 r3 [#pragma warning( disable: 4355 )
3 r4 l# ]0 P! L( x/ e. j: H6 p6 q
, J& t& M7 [6 I) V0 w; D
#include "UPnPImpl.h"
5 V7 Q5 r: c1 R. R; [#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
9 l/ O/ E3 z; j#include <winsvc.h>
+ A( o! }/ s! E$ Z" K, F4 ^
3 t+ n* N$ I4 U7 }- k% e
/ L0 a3 q# P2 j/ W: g#include <vector>
* L; `" z4 H. _. T, f#include <exception>
#include <functional>

% t, Q& }" o& a! u


8 r9 Y  @/ m; D1 U/ U: q5 R* i6 Jtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
5 o- G& q- i3 s* Htypedef _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;
# u9 s: t  r( q7 H: [6 M' Z. }typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

6 L; g5 }+ C% M, h  V; }
6 Y& ]  P% T% A8 }! M( Btypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


8 U, c1 ^; F; }- K* x) Gtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
6 z9 R4 K* r- |, v; x0 A  PULONG pdwSize,
$ v  l6 t- A: L, v* E; |4 t  BOOL bOrder
+ p& U$ O6 K- j& N; f! [);


9 J6 m4 Z, d+ k! Wtypedef DWORD (WINAPI* TGetIfEntry) (
% M1 h) d- X$ j% }  R  PMIB_IFROW pIfRow
# b' y% x" Z5 N. z5 Y9 v);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
! q/ d7 j; m3 j. s9 t, U& H. f
" G2 f" D0 C% w! S9 E- V* K1 C
class CUPnPImplWinServ: public CUPnPImpl
) v8 r& O0 I  w# d7 ~3 o{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
6 A0 m8 P$ @1 T8 V/ G7 l2 ]// Construction
public:
        virtual ~CUPnPImplWinServ();
  j7 ~7 |3 L( P- }& q6 o        CUPnPImplWinServ();
: _% C7 H- P8 O) i% o8 ?% R5 J

+ b9 O4 f+ ^2 p        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
! P: t+ c1 B9 R* f7 \" B$ T8 `, O, K        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
; C. u6 B: s7 {) p  W2 m2 A9 g) @

' [' i7 X2 y4 L. l        // 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
& T( ?3 t3 F# `8 T        virtual bool        CheckAndRefresh()                                                                                { return false; };
1 x$ a# G$ T7 G1 A

: ]/ o8 }' X1 Z0 e( A  K8 N3 ?8 Vprotected:
1 |( J' J6 U2 q$ c        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
9 _/ G; _4 B9 Z1 n* O        void        RemoveDevice(CComBSTR bsUDN);
0 x: ]1 Z2 z, ]% P/ f6 \        bool        OnSearchComplete();
: R9 r5 R" E: w0 Z$ B, E        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
2 ]0 k% n9 W( m, n8 S$ ]0 o9 v                }
% y4 {9 l/ N+ {' k+ _                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
% E9 B# @4 G. l                        DispatchMessage( &msg );
- A( U2 e) B9 l. y- R5 E8 _                }
$ P" [! O/ `' ]                return m_bAsyncFindRunning;
. x' t$ g/ x% X2 Y3 Z        }


" S0 a  f; d; Y3 T4 K2 n        TRISTATE                        m_bUPnPDeviceConnected;
; s, l6 z. A5 }
. {4 u  a1 R! H* T% M3 \* O0 o2 W
4 x3 x# L9 {: H9 I; q' E8 U// Implementation
" M% y" e9 }3 O& @* [: ?+ d        // API functions
; ~; n; I. Y8 l0 F- ?. r# G        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 y2 k5 D( Z) I$ ]; s* b) u        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
+ B: V& b( ^/ W9 s7 w        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
) M: s( r1 z: q  f+ u        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
! q! R# s: U& V3 ]        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
( ?) e, L' @" L        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 I+ |; }1 d  Y; N/ K+ ~6 f
  b! Y: D6 b# i  A
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

- u5 Z9 u0 o/ B' j. D
3 Z7 i* }+ t5 b4 j. E1 v        static FinderPointer CreateFinderInstance();
7 x0 D  b+ O/ J) o6 b& k$ i        struct FindDevice : std::unary_function< DevicePointer, bool >
4 q6 k  x, \' u        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
( t% y' R) G. O                {
$ h6 H! p8 ^+ k5 ]                        CComBSTR deviceName;
/ t& `9 r: y4 q6 c& C                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
% m; V* y7 l. X9 x  H9 b) k+ H
; C6 Y$ R+ N* K2 i! S
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


/ F" ]1 J3 p$ t7 {" [+ D5 x                        return wcscmp( deviceName.m_str, m_udn ) == 0;
6 e; L7 C* g& q" f2 w  Y4 Q2 ]% U                }
                CComBSTR m_udn;
: a7 C9 S6 Q! r: `6 C! F( ^+ D        };
( _% Q4 U& j. b4 t6 F       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
" L7 t2 V3 F- |        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
2 A7 i+ n% {9 ]2 _        void        CreatePortMappings(ServicePointer pService);
5 ]2 z1 ~" \" t0 C# G        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ o& |& z- X! n" r; w- h        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
( O9 K' A2 g2 V! Z$ T% m                LPCTSTR pszInArgString, CString& strResult);
0 n& y) Z  |3 Q( }5 P) E. m        void        StopUPnPService();
0 S8 A- f5 I$ H3 ]6 q7 ^+ t

% \7 ?* I6 j7 t. o. I        // Utility functions
& a( J, E& X0 g& N  Q8 E6 r/ V: Q        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
! Q" k4 q/ I5 t        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
4 ]$ {: \, p. _; t: X  D0 }        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
0 Y5 P( c. I) o# s8 F        CString        GetLocalRoutableIP(ServicePointer pService);

+ y! s( y) z4 O( ?: n
, E+ T& w) i' u0 K; `0 t: M// Private members
' \0 |$ S( U, p# i$ Kprivate:
% r8 ^) A5 ]5 T        DWORD        m_tLastEvent;        // When the last event was received?
9 ?& _$ ^  Y5 M        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
+ e7 A: f% v6 I        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

3 n  s  K, G  v5 ~: @0 b
        LONG        m_nAsyncFindHandle;
1 v3 h4 b0 u6 p) c+ a  j+ W        bool        m_bCOM;
7 H4 y/ R1 r! s+ b$ H- l9 o- J* N        bool        m_bPortIsFree;
" ~* x/ p. I9 R& R% l* p        CString m_sLocalIP;
# b" D1 q' D) g" T7 }        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
1 s2 g2 K! Z" `, c, }1 ~5 t" \0 C6 M        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
4 m5 K, t0 t  @- U' T6 U        bool        m_bInited;
, m6 o! w: P- {- k* B/ C- i        bool        m_bAsyncFindRunning;
- q) A8 z6 H9 m* M' H% `( P9 I        HMODULE m_hADVAPI32_DLL;
( c* h4 n; W+ f' U: M" X' j9 J" r        HMODULE        m_hIPHLPAPI_DLL;
- J3 @  b" E3 t        bool        m_bSecondTry;
  @# }0 x; d& I. g' O        bool        m_bServiceStartedByEmule;
" \. h! Q4 s+ a+ ^0 V; J        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
3 H, C( c/ A: f( F; S- X
" u% o1 b$ q" z
$ f% ^' U; K! {};


// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 |, s1 \3 b# O- }3 E                : m_instance( instance )
        { m_lRefCount = 0; }
& X5 r7 K( x0 M3 A# E

% J1 s7 N8 G, o- f- u   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  e2 e  \, E, ]5 W   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


# A5 D+ k, S* m( ]9 |// implementation
  q$ t3 b6 G; D' tprivate:
: N& ~6 e: W& y% x1 V8 N4 K        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


' q) V: I5 t$ D' ?private:
        CUPnPImplWinServ& m_instance;
% o- {$ {( c; [& A' L; o* z        LONG m_lRefCount;
};


// Service Callback
class CServiceCallback
: ^4 u* X' L3 ^; F        : public IUPnPServiceCallback
: G2 S4 k# @! M- O4 i+ T1 |{
public:
! y) E( j9 f" Y7 m, {6 u        CServiceCallback(CUPnPImplWinServ& instance)
  p) o* h* s, s6 a# k& t5 p) c                : m_instance( instance )
, o( N' k' s/ l$ M. l0 V& I8 Z        { m_lRefCount = 0; }
& C: W6 r  v% d   
/ ^/ p5 L. m% T8 I$ v$ O   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
0 `8 i/ r1 j! a# L# o* m2 I   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


+ c% K! h% O* X( I' y  C// implementation
7 J6 B. V+ C) H- j0 y6 @7 w8 gprivate:
! Q1 O' D' d) i6 y7 R: n        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
$ B+ e4 Z6 m6 Y        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 Q0 i$ V  g% q9 L: g
private:
( v4 [* ^, y: A( d4 c" L        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

$ ^6 C$ l6 n/ v7 p& P5 `! t7 L; C* w
1 W+ H1 A  K5 b/////////////////////////////////////////////////

# P6 S% G4 f& u: R( d/ B( I
使用时只需要使用抽象类的接口。
3 F& p, M* U8 RCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
& Q# }$ n: }- {" ZCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
- i) F$ ~& m0 J5 CCUPnPImpl::StopAsyncFind停止设备查找.
1 u  b8 J7 `( h0 wCUPnPImpl::DeletePorts删除端口映射.