UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   : |8 X, X, E- K9 t/ R
7. 
   ' [( J1 o- G! ~0 N; K
8. class   MyUPnP
   ; W$ N( y7 z  X/ F- P8 s
9. {
   - m+ p8 t2 Y4 d' J
10. public:
    # y9 y( M9 e8 o( `
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ; y8 w& y; y" c9 z. x
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ) p3 i4 h, l& ?  C, i, i+ U
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    , `$ ]/ q; a, X! r: \
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ |) `( u0 S1 c) Q  e
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    3 W/ W0 L' ?. i5 b: M( K
20. UNAT_TCP, //   TCP   Protocol
    . g" I, @+ g7 ?+ R8 ~( W5 H& P+ O
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    6 I+ I: w8 Y" ?/ I
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
      k8 L) {, r9 J, a& }; @) N2 [
26. WORD   externalPort; //   Port   mapping   external   port
    " h5 O; z2 |+ R" @$ U. A& ]& p
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ( ?' @5 Y; `% T# o4 r8 D
28. CString   description; //   Port   mapping   description
    # D8 a' r3 d- i  _
29. }   UPNPNAT_MAPPING;
    : E% Y( X8 M4 Y& B# k
30. 
    * k% L* c0 D+ t$ [- U" X) R) ^
31. MyUPnP();
    9 q5 L. b4 E, J8 F) A
32. ~MyUPnP();
    
33. 
    % @8 T  C2 U' g7 u3 {
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ! g, h6 G& i, a# K3 z1 m% U; T5 |
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ; I8 R6 X' e2 k+ t; G0 ^/ R* k$ `
37. 
    & S8 y) B9 O0 o/ G" }1 a
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    / C4 \, v% A/ r) D+ E
40. WORD GetLocalIP();
    8 N: W* e: n# ?0 x
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    
44. void InitLocalIP();
    9 P4 O6 [& L. `7 m$ ~' k9 u
45. void SetLastError(CString   error);
    
46. 
    ( l: m* T8 F) `* @/ f- m" N
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();   }
    
52. 
    
53. bool Search(int   version=1);
    8 s7 [6 @% `0 A" B
54. bool GetDescription();
    . e$ D& A) }- R- T
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    + p4 y/ m$ w  ^5 U
59. bool InternalSearch(int   version);
    # P3 b1 K9 ^+ h& ~; S4 X
60. CString m_devicename;
    / i. c/ p3 a4 K8 T5 Z( Z4 s  N
61. CString m_name;
    ( A7 O% Q9 F( w* i' i* Y/ v8 z7 K
62. CString m_description;
    
63. CString m_baseurl;
    $ J* h9 [7 x+ ^
64. CString m_controlurl;
      c. m% D3 [. ^) n  C
65. CString m_friendlyname;
    
66. CString m_modelname;
    * ^$ t" [7 E, q
67. int m_version;
    + ~+ M( r+ ^* A
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    , b/ _" T8 }" g
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    3 S! h# v. @1 E  z. b/ M
74. WORD m_uLocalIP;
    
75. 
    % |9 u0 q" V- Z. |
76. bool isSearched;
    
77. };
    5 p: \) p$ |, B( \) B
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   , Q$ v) _# g. V
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   & U/ _  m$ b6 D; v9 q& p% j
5. 
   2 o9 n; m# B: Z+ o& ]
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   , J% G% C6 X( ~" Q: q
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   - e, g0 s. M  J  M& H
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ' t  y; R( V5 S* Y: c
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    . V, W% A: D' F! U! b. S
22. return   s;
    
23. }
    
24. 
    " B+ |( ~) L! v+ T' e9 J5 d0 u
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    8 s8 G$ w* u" L3 c7 C
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)
    & [' u1 z$ K6 g  O$ F# X- B: Y
36. {
    
37. char   buffer[10240];
    
38. 
    ' d8 P  R, w& e
39. const   CStringA   sa(request);
    # H" D8 H7 l% k7 A! e
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ; O! O" L% o4 e2 Y# B3 n
42. 
    " ^* r6 E' n" n9 o1 Z4 @" a
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    7 Q# U+ Q; p% N- r
46. sockaddr.sin_family   =   AF_INET;
    ) u1 ^$ L4 _9 g' v  |7 ~1 ]9 ?
47. sockaddr.sin_port   =   htons(port);
    6 O) {" n7 h7 ~2 Z7 V
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    / w1 W) B: C$ t
50. u_long   lv   =   1;
    - P3 w1 ]8 u5 v
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    2 h- [  N) m  x
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ! U+ d! ]9 N/ h9 [! X5 h5 F
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    % Y5 _# C2 X" m( V4 Y
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    % I0 |* A1 Z$ J1 z/ d; g+ }
62. 
    ! P3 S5 W' K- K- Z) V/ {
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    6 }( \5 n1 n+ r" w  p2 g
69. 
    + m6 u7 o# V% z! M5 J- f
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    " w* x, z: P2 P) e
74. struct   sockaddr_in   sockaddr;
      v  C9 ^$ S4 e# G' g8 O& S5 C
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 x. \9 R( J4 w( q& L: g& z! O4 j( F
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    6 N$ h: G$ Y3 M8 L% F3 f5 M$ ?5 b
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 N4 G* @. D4 ~8 z" L
81. }
    
82. 
    ) E" _# I" Z; |! D' v' d
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    : t' F% p9 @1 V* Y" G1 L' u1 c. {8 l9 B
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    . \7 ~; D# `3 T9 X0 S8 O+ B8 b
92. pos   =   0;
    + X# D+ ]7 V. w& J. B% P0 w
93. status.Tokenize(_T( "   "),   pos);
    $ _: F: x9 }' ~: G  @  S% W) {
94. status   =   status.Tokenize(_T( "   "),   pos);
    ( B% x9 X7 T  a0 Z0 |* E
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    " r" U$ I; f$ I- Q& n
96. return   true;
    
97. }
    
98. 
    6 \! U8 i3 b( A
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    0 t- t* g$ \+ H& g6 h
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     9 }) a6 r. z' ^% [$ o+ M) D
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     : B, o. o0 m5 Z
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     1 A7 A5 v+ S, \2 G+ o4 k
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     + B) F2 u& q/ g5 f2 m: M1 Z
118. isSearched   =   false;
     
119. }
       _7 f3 d4 A$ Q. C5 \. _, R" G
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     9 {2 j& |5 T, x1 Q; I2 p' F% h
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     $ y  G# |' N+ a1 i
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     / G# L1 j. R5 T! X( G) h
133. 
     $ u0 A( d% Y- F! d) Y6 X6 L
134. bool   MyUPnP::InternalSearch(int   version)
     - K) x1 W6 J+ S, x6 A
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     9 Z: j# b$ t+ U/ J
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     * ?& M4 v0 e7 G9 d* W& d# ?' ~& _* \
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     3 ?# h) q( [6 Y+ `
144. };
     6 e, ~! D$ G# r  {9 F5 R% ?
145. 
     ; z! w' t/ G9 e8 a3 L. F$ v; M
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     + R: B) c2 M8 t9 |2 ?
149. 
     ' D4 F3 Z  n# X- n: Z7 Z, j  _& ~6 o
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     # }9 {# ~0 J6 v) \% n
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);
     
155. CString   request;
     3 @& I" ~4 {1 o) g& a1 n
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);
     9 [  {" u' Y( f% U& ^! e7 @
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     2 V$ \! s7 ]8 {
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     8 y9 e2 J6 n9 z5 D6 m  G
164. char   buffer[10240];
     3 s( p  V9 j8 k  a* m/ j
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     1 k4 q4 G+ `$ W! c
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     $ m! G, }- q4 @5 v. T, [0 `/ ~
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ' [  W$ o+ `: P1 d: g! O' E
170. CString   result;
     % u, p% j  X9 W$ C- `% F4 P. e
171. if   (!parseHTTPResponse(response,   result))   return   false;
     % L& {1 S2 x; f: ^" D" r
172. 
     
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)   {
     9 @& t+ K% O7 y* C) U: N
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     3 g2 `/ M2 }% v$ A
178. if   (line.IsEmpty())   return   false;
     & ^/ \' z  t& B2 e3 s
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     $ n# C) V3 Y* T: F6 }1 A
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     . Z. ?- U7 ^0 }, P3 w! r
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     : ~- b3 s5 X+ O! d# g
192. 
     ; \5 M4 i+ K. n: B" H+ ^0 I% {( i
193. return   false;
     
194. }
     $ Q2 i7 l9 v& T& S" p- \

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
" d# _1 G8 ]; }1 L/ }  a$ m8 b

#pragma once
#include <exception>
3 u4 Y( n: v, ^
: O6 u  L8 j3 S) k/ o. a
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
9 p$ R8 s! k9 k0 S- ~! Q- W+ V        TRIS_TRUE
! z$ R6 F9 I% }* {, Z};
* X$ T6 J8 r- k1 M2 y

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
2 m- C' _: z. i; o! p        UPNP_IMPL_MINIUPNPLIB,
; r- ]) Z4 x: B/ ~1 m  B        UPNP_IMPL_NONE /*last*/
};

- T$ W0 q2 q* ^* A# `9 l: t


class CUPnPImpl
# V; O# Q) A; x% w  r; _& L{
; m7 V) k5 U5 r5 vpublic:
        CUPnPImpl();
7 x2 M8 k1 k! z, c        virtual ~CUPnPImpl();
! k7 x: Z3 |& x' X        struct UPnPError : std::exception {};
9 u# @& u/ I3 D( B9 W  W        enum {
9 n3 {( S# j0 K+ m( D& U  Y                UPNP_OK,
                UPNP_FAILED,
2 e/ Z" u. G( m  o$ I1 J                UPNP_TIMEOUT
        };
' f+ f& Q, e* T

7 M& l. V8 ~2 u( a9 w        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
8 i6 r' \& j# r% `        virtual void        DeletePorts() = 0;
; d+ j' D. ?, F  g+ Q8 t        virtual bool        IsReady() = 0;
2 P1 D" H: }# }9 a        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
4 g0 r6 O  x" p

2 `  F" v( O. _7 w9 \( ~        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
- J/ E" t5 h7 F) b% `3 d        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
: W# {! ?+ w; ], Rprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
& u! f; v" z4 X; p9 a        void                                SendResultMessage();
( G' W' V- ~" V* ?        uint16                                m_nUDPPort;
& X4 i5 D% K- ]* _! i' l        uint16                                m_nTCPPort;
# ?, `9 B5 u, |% o: {# o- S6 t        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
: Y* t' J* f% h' ?5 H9 F
8 K# r/ S2 v5 q. U$ [, g' G
- x7 W) }- a7 o3 a: i5 C( yprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
$ x. A, z; u( [* q
1 W! T; B, `6 X1 U- G
};

# g3 n7 c' O9 W( a! F' q- e
// Dummy Implementation to be used when no other implementation is available
8 ^6 _4 w3 L% _/ dclass CUPnPImplNone: public CUPnPImpl
3 O5 j' H7 H6 t. I0 z  b{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
7 j( u9 h9 L6 g8 J. V5 i# Q0 L        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
0 X% L7 \6 X; q( @        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


; G$ z$ x6 K# I! P$ j0 a4 ~: a/////////////////////////////////////
/ B1 M  x( S/ q3 H8 r; v//下面是使用windows操作系统自带的UPNP功能的子类
0 B- j9 X6 u8 ]% E1 f9 N( a
$ j# Z% l$ v5 o) I$ g  U
. H7 |/ H2 S) K! Y- |! ?#pragma once
#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
5 [+ l' L2 m8 V: z' s#include <iphlpapi.h>
! L; t/ F( d- `" k+ m! s! ]#include <comdef.h>
2 [: [; {/ L3 R/ _6 V/ P#include <winsvc.h>


+ x% f4 b8 ^6 n: b% ?7 |#include <vector>
. p3 z! J% B/ \- L/ T- {+ g1 k#include <exception>
  D% D0 V7 C7 F! d3 M" O#include <functional>


9 g( e  h# z  m( r9 n
  @  P; J- N. S) Q: g
typedef _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;
8 R( o' P7 H! p% J, h+ v( k- ytypedef        _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;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; j* b4 S3 w5 `% x- v9 {
3 r% K+ E" o& P5 y+ H/ e0 H6 h% u
/ g# Z2 N: p9 n  L6 c, v. b" k( p6 mtypedef DWORD (WINAPI* TGetBestInterface) (
, M1 R* C6 B/ ^6 u5 U  IPAddr dwDestAddr,
4 F9 J* I2 u- X& Q4 |% G  PDWORD pdwBestIfIndex
);
4 A# p: j) B( \7 ^4 J) X

typedef DWORD (WINAPI* TGetIpAddrTable) (
* ]/ K0 B/ y* T  PMIB_IPADDRTABLE pIpAddrTable,
; P1 P! O$ M/ Z& K, n) M1 f  PULONG pdwSize,
  BOOL bOrder
);
! N2 Q2 w+ y& \* T5 J8 y- e

4 a6 }3 P2 U4 F& n$ X! @8 @typedef DWORD (WINAPI* TGetIfEntry) (
6 F9 N8 j( m# K7 x' }  PMIB_IFROW pIfRow
9 \8 p/ w! Y2 s( M9 U  b+ Q- P: F);


& A9 \, N7 y3 @2 w6 {! l0 _# V3 oCString translateUPnPResult(HRESULT hr);
" |4 M% v( S8 Q( T/ L2 v2 DHRESULT UPnPMessage(HRESULT hr);
- N: a- a3 G$ H: P' h  J
( x$ H; i( m% `) P
class CUPnPImplWinServ: public CUPnPImpl
  I4 \6 `8 E! r4 E{
        friend class CDeviceFinderCallback;
9 P. u) t- L" t5 k, E        friend class CServiceCallback;
) p; I' n& D  f; |+ O  O// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

* F4 b5 {% O+ P1 j. j
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
  O% C% g4 E; O. N! W        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
$ x$ h7 M9 o$ M( p8 m6 Z1 a        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
" A2 e0 e5 w- t( G% |6 ?; @

' N: q7 m) Q* M3 h$ H/ L  t+ a        // 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; };

5 @8 U- k; I7 m! Z: z
& p: j6 g! f9 Z, |- |) kprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
- U( I7 T. R; L: `- C        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
3 n9 X0 z; v7 v1 H+ M  i        bool        OnSearchComplete();
7 V3 ?* W4 O5 k6 U& [/ C% i        void        Init();


7 N5 M( {$ [/ e9 C1 g, ^, @        inline bool IsAsyncFindRunning()
/ z5 ?+ l8 Z9 {( U2 X) n4 Y        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
7 w9 y8 f0 i) X. k- x) Y& r# V                {
3 }: V1 Z0 v3 A% [  ^: c! m/ u                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
( r! s+ P4 ~- `+ S+ W2 e; \                        m_bAsyncFindRunning = false;
5 b" N5 I4 B& j7 p% |; ~0 E                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" _. }, v0 e/ n6 t                {
: j3 s" }, N  ]( r                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
8 \" f1 R, G6 j$ d+ d: y% {                return m_bAsyncFindRunning;
( R9 G0 d+ a) v" ~) ~        }

6 M" U2 h( T5 d0 e0 K, C
( `, W! R' D; C$ [        TRISTATE                        m_bUPnPDeviceConnected;

7 n- J7 u' m! W4 @3 c! w" y3 }
1 c1 u$ u& W! b// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
8 @. p' j" t% d- ~# w2 y; d        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
% h2 ]) d& h3 m) X+ b% V1 C; K        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
+ @) C& r8 {8 e: P  G        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

; ?/ t: L! B* m, T
1 M8 x+ r% |9 W        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
6 C; g$ |- K& c4 y        TGetIfEntry                                m_pfGetIfEntry;
  w' {1 h3 _4 v+ ?. e' \9 u* M

0 l' H# D* |6 t- t; n        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
" T# a' E$ E$ P+ ~5 i                        CComBSTR deviceName;
. W0 _- H1 s) i4 e+ T5 I, ~! G                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
0 [7 i2 V' ?# r3 m/ f2 c
4 B. Q" @* n; _/ n/ D, H# S" R
% @0 H- S  K7 |' H                        if ( FAILED( hr ) )
/ q1 C& p4 G  h; |8 u                                return UPnPMessage( hr ), false;

6 F4 q( K1 T0 \5 f  g7 T+ o4 G
0 r9 ~/ c% A+ R7 T                        return wcscmp( deviceName.m_str, m_udn ) == 0;
3 I! Q! ^2 @' P" T# D: z                }
% [; q9 ^# v- I4 }2 @! F  E8 N                CComBSTR m_udn;
- z: Y( P' Q! ]% V1 R5 w/ s        };
       
) |& T" T+ E/ v" ]        void        ProcessAsyncFind(CComBSTR bsSearchType);
+ ]5 T3 u  W7 T0 f2 ~        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
" W6 N  U2 `  O8 m* |( V        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);
8 Q' N: V6 B. y- S% {        void        StopUPnPService();

4 b! c! A" [( j, `# t4 \+ w
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
1 R, @$ w  G; [: u        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);
; D! X5 h- L1 x3 b
2 g. J! S3 W! B! w- K2 b
// Private members
private:
: j# L/ F- N8 D0 @1 V9 }        DWORD        m_tLastEvent;        // When the last event was received?
4 k. S0 @0 y4 H        std::vector< DevicePointer >  m_pDevices;
" E; G  m9 d: ]. {' T( b        std::vector< ServicePointer > m_pServices;
4 [6 F1 Y7 W5 z1 ~- A) N* Q% v2 A* t        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

# h5 g: i6 A. `, E' H; B! q" p
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
# [( b! A- ?( U2 Q& D0 j        CString m_sLocalIP;
        CString m_sExternalIP;
- C! f/ p3 K5 Y: G3 x8 t0 R        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
6 F" R; B# ^  g4 F7 @        bool        m_bInited;
4 O! e' _7 p! y8 i6 ]        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
( m/ j% }4 w2 `) W$ ~        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
1 B* ?  N) z6 x' o        bool        m_bDisableWANPPPSetup;


0 W* B( q0 j0 @( l' I0 C1 p};
+ ?  e8 @  K& ]% v! r0 E
; A6 P8 C& l! Z2 j0 x5 g2 G
// DeviceFinder Callback
/ N' s8 l  x- J: Z: Zclass CDeviceFinderCallback
# Q7 v" y/ d' C) B6 A) V9 |8 A        : public IUPnPDeviceFinderCallback
2 E0 G/ _  f9 W2 K8 n+ A{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
$ _( T& v# I/ Z' g                : m_instance( instance )
  d( n7 |( |) O        { m_lRefCount = 0; }
' Q, Z$ A- P) A0 y
# [+ x7 s+ s3 F6 [. J( w
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
% h: X! U! m- N7 x7 K% m. V

// implementation
% ~& w5 s! L5 g; ?% Sprivate:
* R: ?% X4 _8 h; b) p        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


private:
        CUPnPImplWinServ& m_instance;
5 m/ n; A; ~1 x- E2 P        LONG m_lRefCount;
};

6 ]" \6 f. s" u
// Service Callback
' f3 f* O4 u$ f' vclass CServiceCallback
) [1 @9 z( j5 B8 |, u4 C2 U0 j        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
( K- l$ S1 _0 c   STDMETHODIMP_(ULONG) Release();
! ^) G% N) g, M* M" p+ B8 Z0 n

2 v( v4 Z0 k: J  K$ j4 X9 L- Y// implementation
! E" e% u. S  Q2 [0 Y' a) Hprivate:
8 c: `9 n$ m; s' m        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
" e3 G2 z4 d6 w

; m8 P$ @9 k+ H( r+ Q* B% @private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
( {! N# ~/ t8 Q' ~$ \: G
5 ]8 ]" ~6 |+ U1 }$ n6 I
2 R% j* S7 ]. _( Q' J2 {, }  e5 U/////////////////////////////////////////////////

, u& v6 `4 F  A2 k
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
3 H6 o6 P  t1 j: wCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
$ k9 a6 s5 l4 G! Y9 u- HCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.