UPnP

天云上人

/*uPnP.h*/

1. 
     @7 A  k% s# a0 o
2. #ifndef   MYUPNP_H_
   " ~4 k1 B+ e4 k& j4 |
3. 
   . x' g& G7 R! B. I7 c
4. #pragma   once
   # H# r; n9 f7 ]# D; ^, f6 ^
5. 
   + f4 _! L3 @0 c: b6 S
6. typedef   unsigned   long   ulong;
   / o2 J/ g; G- |
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    8 \! S  d9 y, _: p" G, R
11. typedef   enum{
    - {& h' s+ p- ]+ w
12. UNAT_OK, //   Successfull
    / i* v# c& |0 M+ w
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    " F- s+ U7 w3 i: K) B7 @
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    3 P3 _' T5 I- _) N. {% e
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
    7 M# F/ r( v: M
17. }   UPNPNAT_RETURN;
    7 U, g  X! U; Y6 Z8 q  j# W
18. 
    6 X1 z; H- E$ v+ H, C2 J$ ~
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    4 g4 i" _% G* I# J
22. }   UPNPNAT_PROTOCOL;
    
23. 
    3 u% B9 d6 G; M+ X# U+ N) V
24. typedef   struct{
    , v% C& j! C" d7 b) m
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    6 o5 A5 `. @! U9 }% a. M, E
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    - j9 y  O0 k2 f$ P
32. ~MyUPnP();
    / O$ W! E$ K9 z$ u6 M
33. 
    ) p# r+ u3 y6 y* ^
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 u; `$ L5 O* E
36. void   clearNATPortMapping();
    , I0 M; I( C1 |& }3 _4 Z
37. 
    
38. CString GetLastError();
    7 X$ s5 ^7 w) ]/ \7 o7 n5 t
39. CString GetLocalIPStr();
    0 X) i) b% q! I8 s2 H
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    2 a0 T) ?& I6 d5 j4 g2 {
42. 
    ; @0 w& [# ]: f7 i; ~! x& S
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    % I8 |; N- ]! ?) ]' S* H9 w
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
      V# g0 k& ~7 I! T
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    + Z2 j' q6 Z; x3 b9 D  K7 z
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    8 W7 w" n8 U: ^+ z) l
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    & `4 {; u5 f/ O0 ?9 E7 R
57. 
    % Y; }9 [; L) B* {# c
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    : Z7 X$ r# p$ x9 A4 B( ]. W# \
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
      J0 f# C1 \1 C5 R
64. CString m_controlurl;
    $ ~) E, {' a0 g$ [8 _9 B8 f
65. CString m_friendlyname;
      c1 z; ?, L+ }; ^) B
66. CString m_modelname;
    
67. int m_version;
    ! h7 c! W5 [9 F* l6 \; b3 w" L3 H
68. 
    
69. private:
    7 Q- D0 O/ J  j) t* U% I4 H
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    : s# ?6 K) }2 H9 N
72. CString m_slocalIP;
    * _+ @+ i* a; u0 U2 ~3 M
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    $ G- V7 _" e! L2 {5 U3 H; t  j
76. bool isSearched;
    1 x6 [2 r# p1 {' G
77. };
    4 o9 D" _# I. y' w
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   1 @0 q* D: g7 K; V# y
5. 
   4 q' T: j: I: I% _
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   * ~) E; z! L: H6 [
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   . [0 y, {. I& }: p4 u" W( t
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    3 m' l! v! I! M6 b* {5 _9 \% P
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    2 M6 _5 \' Z8 g3 g( l
15. 
    . q8 @4 B0 J$ i. a$ x
16. const   CString   getString(int   i)
    - n# ^3 Z' K3 g' _. c5 T& S/ s6 m; v
17. {
    7 s+ g' V+ w4 r$ H9 ~
18. CString   s;
    
19. 
    0 v% c2 f" I5 k8 p9 }
20. s.Format(_T( "%d "),   i);
    9 R$ }$ g, ?, u' u8 U
21. 
    6 q1 E* k! S9 d5 d9 E
22. return   s;
    
23. }
    : S' T& M' L0 [8 _6 r
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    7 x; \0 z! J/ ?5 g3 s
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ( O. m! `  k  G) P8 Q6 Y, w
28. }
    ; J- [' {9 F+ h
29. 
    * V, K0 d1 a7 e5 a' v" X  @& b
30. const   CString   GetArgString(const   CString&   name,   int   value)
    6 ^4 V* K7 U3 `+ x) i
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    . E* N" F( |% G) F5 C& \- D3 R
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    2 \% L5 b* K% G: Z8 S
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    ; p- h' G* `3 Z% j1 G
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    1 y7 w) C  L. J0 q* H' B2 T9 m
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    8 {2 J. _6 h: ?4 q. |. a
47. sockaddr.sin_port   =   htons(port);
    3 c* S, a) J' P
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ( X7 _- Q) K: j* X5 Z* v3 W4 L
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    8 O2 y$ J# }4 H4 y) u
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    ! A6 o" Z, Y9 A  e+ {
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ! A- `9 k  I6 g$ Y$ `7 Y
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    $ o6 v  K  Z+ Z8 |  Q8 K
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    $ B# B. J7 A5 }4 o, e
59. if   (!rlen)   return   false;
    9 @8 \6 C) c+ \5 R( ~
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    " V- M# N6 @$ ?& {. U/ H  Y/ M' i
63. return   true;
    
64. }
    4 \0 k7 N$ h" X, z+ L
65. 
    # w3 Y/ F' n; C5 u3 i
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    5 ]# s" F: y9 c$ a
67. {
    4 [7 O3 B: s' V, h
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ' {2 K9 @! m/ C( U# ], l
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    9 E1 Z7 L- v% d
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    " T4 F8 R  d* B
79. 
    3 n# D+ D+ ^. X4 e% d- \1 a( I
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    $ \  G9 q, w3 C2 K
82. 
    % D9 D; n" T* T& ]' s% f+ }/ T. i) l
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    . ^+ Y) `" h" z
84. {
    
85. int   pos   =   0;
    
86. 
    5 Q' c; P6 d& b( U2 r
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    - M  M( q! V, _9 B( L
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    # P& V5 d/ `$ u
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    - C2 k7 a1 Y4 v* W1 p( L& X( e
96. return   true;
    
97. }
    $ l$ b. Q0 r- _. S; Q6 L- l: B( g. ]
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    9 ]$ c" H& V% J% F2 ]% T& P) {- `
100. {
     2 W! `' I/ t* p8 t6 R, n
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     + d8 N; n0 i$ ^
106. if   (posStart <0)   return   CString();
     
107. 
     4 Q4 Q, Y! {1 n$ n9 q
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ) n0 z+ y3 a! _5 [6 r$ |2 `
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     # v9 {$ D. v* \, q9 u
115. :   m_version(1)
     
116. {
     2 ?9 D9 a% ^9 x3 O5 u
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     $ G  H6 f2 t% E2 A$ A
121. MyUPnP::~MyUPnP()
     " H' ~9 n/ G- B
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     - ]+ Y! I" f6 |: ]% r/ |/ O
126. search   =   m_Mappings.GetNext(pos);
     . n# `1 ?6 y" F% h. W
127. RemoveNATPortMapping(search,   false);
     7 o+ e6 B8 Z; V& Y( [$ x1 j6 B
128. }
     
129. 
     0 h2 p5 x# P. t& e0 x" u+ u
130. m_Mappings.RemoveAll();
     9 F+ c; A4 o0 I1 H- J# O& ^, Q$ y
131. }
     4 D" A8 g+ y4 U% S4 q5 y
132. 
     
133. 
     7 x3 ?% b$ Q1 J% I/ n3 ?9 S) ]
134. bool   MyUPnP::InternalSearch(int   version)
       K2 h6 t* G! n* s- h* C) {
135. {
     7 @, I; H) K3 \
136. if(version <=0)version   =   1;
     - I, m# s2 U' u* `# G/ c
137. m_version   =   version;
     - j6 W" u* d! }: P5 U
138. 
     + @1 X# B2 r4 n
139. #define   NUMBEROFDEVICES 2
     9 m7 M. _4 r1 `
140. CString   devices[][2]   =   {
     , }0 p, O: N6 B$ X
141. {UPNPPORTMAP1,   _T( "service ")},
     5 l6 D% C6 O6 P& y- ]" \
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     * _9 P" ~5 l% e5 G4 ]
145. 
     2 V  o( ~$ g. {3 B1 {. Y
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ' i0 b3 @! {" k) G
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     % g: D$ {3 x% U! n: v: ]& k
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     1 g4 o5 \0 Y1 w) r/ \4 c9 O
152. if   (!(i%100))   {
     / Y* c0 e4 `2 u& f. Z1 H
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     7 `$ }1 a+ e, v/ L. G, c+ @
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 "),
     * c+ x' f, p- W; T
157. 6,   m_name);
     " `/ l1 x# e" [% m
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     ' T9 n! I" u6 [0 k% R
163. 
     
164. char   buffer[10240];
     ; e' V+ B* d3 J2 \+ V5 y1 @
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     8 d/ N7 f+ w# J% ]5 n) E
167. closesocket(s);
     
168. 
     ; T& a* j5 y" r$ z* {
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
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);
     % w- _! D7 W$ V- X. N8 \9 t: ~
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     3 z( V, m4 n7 s
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     + O% P5 f! K: {
181. if   (name   ==   _T( "LOCATION: "))   {
     ( @# O8 U& g; a/ x1 \2 b# u' D& p6 \
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     ' a4 K( h9 ?% r3 H; \$ D
193. return   false;
     ) _' ^, Q7 G, ]
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
' ^+ k' {3 l9 j" ^) \* C
; i& T& u8 T& Z
///////////////////////////////////////////
" W; Q" b7 G: M$ b9 ^# m; E6 \2 [//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
5 h0 y, K) i$ j% B' l

#pragma once
7 C5 a1 D- {( _( S) ?#include <exception>
) B) d2 K4 S$ z" H# w+ a$ t7 W

  enum TRISTATE{
. `1 p2 W% g$ c. _/ U        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
- l! k6 e# N5 b3 E: z};

* j' a  ^; y; d& V8 v' N  L' T! v
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
$ m5 v% v9 x& |+ d- P3 J* C        UPNP_IMPL_NONE /*last*/
+ P6 T. X* a! J3 N( I};
4 K+ W8 X3 U0 J5 I  u" f, x& e

# s: a' G' a8 K$ |+ T8 a
8 p3 Y- C7 z, v( H$ r2 l# t' K
class CUPnPImpl
{
public:
5 z- {" Y( a% j$ A) \( r        CUPnPImpl();
: E7 a3 Q3 X: a; u& c  K4 b        virtual ~CUPnPImpl();
- w# ?4 `# Y+ E$ b* f        struct UPnPError : std::exception {};
# H5 w- a1 ?4 u4 w        enum {
( j3 n+ n( e: E4 i4 X+ V                UPNP_OK,
9 G" N2 D8 f* ^" P                UPNP_FAILED,
6 V$ x4 U; B+ E4 q# L' ?" \                UPNP_TIMEOUT
        };

7 f" T5 p. ?, y( H+ l1 g
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
$ u- S2 n: G5 }# q2 h7 u* f- ^/ s  J1 W1 Z        virtual bool        CheckAndRefresh() = 0;
) J  h3 U7 j# c/ [        virtual void        StopAsyncFind() = 0;
4 E/ Q1 _0 g! |3 f, @+ t9 V        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
, b- O3 h# S' I4 Z       
$ \& V! O+ {; \3 h: L; ]        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
1 W9 W- `6 v+ f3 j% X" }/ y
" L" c2 a$ O5 B1 F: }- n# u
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
' M+ m! x  [: `        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
# U! U8 ^2 _: w5 h- U        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 f( ^" m* t; P) O2 D$ }

6 A3 C$ p* y& m// Implementation
protected:
- e5 {6 v( A8 \& e3 P6 q        volatile TRISTATE        m_bUPnPPortsForwarded;
  l& ]. v4 C$ z: R& g* {        void                                SendResultMessage();
        uint16                                m_nUDPPort;
: w% n* _, n# x0 F        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

, x" D* w5 e9 D* _  z
+ M9 _4 |. |+ z" ?% R5 @" f' zprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


$ u" L( N/ V) a) c/ I, w};


// Dummy Implementation to be used when no other implementation is available
' c* a; r( |4 j9 X! N, K7 M6 h: nclass CUPnPImplNone: public CUPnPImpl
{
. l# T% W( [7 a, D) [4 B! p: fpublic:
- o) m. h+ I8 E: A  X" X        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
" _3 c$ e9 h- \" i1 O, W/ k        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
) O3 x; m' H( _8 A0 i. K% v        virtual void        DeletePorts()                                                                                        { }
6 V) J$ [' ]3 q; @2 D4 i- G        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
# {. {! o# p5 x4 v  o/ W};


/////////////////////////////////////
0 J5 G/ l9 w1 c6 a- ~. F//下面是使用windows操作系统自带的UPNP功能的子类
2 ^2 V+ U1 X7 E
- a0 q, a6 ]+ O
% Q' s. J  N6 n- D#pragma once
$ t8 _* W8 ]) _# o  b#pragma warning( disable: 4355 )

- b' F- |& n- k+ w
#include "UPnPImpl.h"
+ M2 q8 m, W) \% |+ V) {% q0 i7 ^#include <upnp.h>
$ Y+ q& r6 e( ~7 A. R1 @#include <iphlpapi.h>
#include <comdef.h>
+ l5 I( p" ]. p$ Y4 `#include <winsvc.h>

) Y  h4 z, s; {5 B# }
$ s1 K+ ?$ E6 m2 T& f; w6 k#include <vector>
#include <exception>
+ N3 F* X& M$ F#include <functional>
( Z. Y( l! y3 k7 ]4 x3 f( H
) t0 N& ~) l) q; g5 i

0 O3 C& p1 }- b. @! J/ t' X) u
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;
# [6 I6 l+ ?& k# m+ Z- Z6 t% Utypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
' j: _1 q: `: o4 W7 l2 Utypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
( z7 M3 c) Z9 w9 B; Gtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
: @: D( G! ~  u$ G! h  PDWORD pdwBestIfIndex
/ H' y" M/ \& |# j);


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

* ^! f. U, H, v  |) Z0 z; j
typedef DWORD (WINAPI* TGetIfEntry) (
) w3 d" p! n4 x. d7 C# g5 I4 g$ n  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
: S$ i; ?* M7 [8 @' F

class CUPnPImplWinServ: public CUPnPImpl
( [1 A4 C7 o* _& ~{
4 Z! o* b) }4 _8 L        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
3 O4 b; x/ L( [: |  U& a        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


: W, ^$ v. \" s. X, b        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
0 F- z0 l  o* l5 f. }" v. X* {5 @. w        virtual void        StopAsyncFind();
0 Z+ k- o; \: ~4 J        virtual void        DeletePorts();
        virtual bool        IsReady();
' \' Z9 G( J5 U. }) m        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
; {( T) r1 B% V. o8 }$ c2 `- K

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
  K+ ^, x6 Z4 e5 _% s) _        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
" s1 t" F+ }  e" X$ \/ G
( _& N+ U4 K5 U9 ^! I
protected:
, p+ b0 T! r! _4 H        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
5 ^& ~/ ?; [6 p. u0 H, @        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
$ h5 }0 B' e/ }( c: C, D+ `0 l        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
" ^/ ]7 k9 ]" B- u0 A        void        Init();

1 ]! A' C* u0 u" J( J
9 F; a7 a1 s, P0 P7 G+ c% V        inline bool IsAsyncFindRunning()
        {
% G6 C2 Q6 k, N( Y) l                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
2 M7 `& ~) r7 y. Q; Y3 L4 O5 e                        m_bAsyncFindRunning = false;
% c: @" n- I4 @                }
1 F2 t4 |) z  d( i1 D3 Q                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
) l) T( e  o6 J3 }2 U$ S6 x                }
                return m_bAsyncFindRunning;
* s$ D" \, y9 F* C4 A        }


        TRISTATE                        m_bUPnPDeviceConnected;

8 |7 [9 S6 y6 ]) h
// Implementation
        // API functions
4 u) c7 L# Y/ O. J        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
- P: w) }, q' P2 f. ]        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
6 e" J( G  w' i/ X+ w* o) ^        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
1 S9 J$ B  O/ e) p8 E( f" f        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
' s! u5 r$ @0 X1 J! o0 q+ E

        TGetBestInterface                m_pfGetBestInterface;
  m3 K5 p' c9 j; V4 W- \        TGetIpAddrTable                        m_pfGetIpAddrTable;
9 f) l$ Q4 W- \1 A5 B2 N        TGetIfEntry                                m_pfGetIfEntry;
# C4 L# P! |7 h( n7 x

' c0 s# c3 y" B! u        static FinderPointer CreateFinderInstance();
+ |+ [+ M- P* b; i) f        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
2 [+ u  e7 w4 H; R2 W" Y                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
- D4 x. e+ M& x; J. M8 c                result_type operator()(argument_type device) const
# o" N6 ~) h. Q% G( ~4 N                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

2 j/ F% p7 ?' f* k4 P4 U
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
7 k+ o' d9 F, F. t! x) c5 L& m        };
       
% i  {6 b% S. ~, N: |        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
1 H0 }& c4 \3 O0 G7 f        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
( }1 u- Y/ ]' o  t9 L        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ a; K+ z: j- A5 [. E' \        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
6 `% P  b! Z* Z

1 n: z$ r  Q$ o3 b        // Utility functions
7 o  r) t) J2 }- f2 Q        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
% q2 ^) _1 Q& c, o2 O        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);
* _. Q3 i7 C8 B: R2 d+ }
, n& O1 k/ C' ?. c7 `. \
! ~( ]1 _4 N' Z, J// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
" u! l  w7 ?) X- l0 U+ a$ ^- W% J# j9 O        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
  B5 G4 P) |' v, W        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


6 M* R) i. k# W( ~) Z8 d0 L, }$ G        LONG        m_nAsyncFindHandle;
" ]7 l6 v* o, B" R& f$ u) J        bool        m_bCOM;
! ~1 g) r4 s3 p* j0 v9 y        bool        m_bPortIsFree;
  z; a- M3 g  S! j+ S        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
! N! E$ U" \$ P8 I7 d        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
) M% r1 r2 C/ C$ P        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
. X5 M! l  L. _6 X( x( R        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
4 ?+ G% \9 |' Z
% q" T+ I, L7 a8 w* B2 p% n3 N/ G0 h
& F8 m4 |7 B$ F8 [9 i};


// DeviceFinder Callback
class CDeviceFinderCallback
- v6 y: z% }3 c6 q7 O        : public IUPnPDeviceFinderCallback
{
7 o) G. q1 o& {% ?. u9 Vpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
" B* R2 n0 F- U6 s) g" y
$ x( d. T6 a" g. a8 M
0 o/ x7 {" z- H. F  \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
/ u; K) w1 y3 Y8 a; @   STDMETHODIMP_(ULONG) Release();
8 D; l9 j1 O9 y& s: C
# ^0 s/ Q6 y$ J5 e5 |9 V/ F
! ]/ _2 F' N5 g+ {// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
: s' r2 r; m" E: w8 V* A2 s        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

" a% q  `9 T1 P- H6 C  R
private:
; r! j: X( F' f% T        CUPnPImplWinServ& m_instance;
3 X. k* n- r; _: u' R+ V' m        LONG m_lRefCount;
};
, F& U! a. E6 j, W$ Z

// Service Callback
# I4 n8 K' Y2 Zclass CServiceCallback
        : public IUPnPServiceCallback
- V4 b- ^+ b' }{
2 @3 J2 d2 [* M" Wpublic:
, y8 _) M; o) a: R7 [2 m' x        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
* O1 _% L) f; E$ k6 t$ @" y        { m_lRefCount = 0; }
   
% X2 w$ C# V$ C. `* _3 \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 C) C9 W& C6 o& p   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

# ~1 G+ q9 M/ [0 ]
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
% c/ ]9 H* m# T. R! t3 m$ F        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


5 N/ i) W/ C6 Q- nprivate:
        CUPnPImplWinServ& m_instance;
/ }; r9 A3 T* I: _        LONG m_lRefCount;
};

( P0 K5 _  P6 h4 d( }/ u& g
1 m7 e# U" o8 z& y* A, s: W/////////////////////////////////////////////////
" w* ~0 I/ `( _- f) f% |0 e
" c0 w3 |! r( g; W
使用时只需要使用抽象类的接口。
; `0 O0 D7 w7 DCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
* p0 K4 u) F0 M* p; |3 cCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
! \7 Y3 H; l0 [# tCUPnPImpl::StopAsyncFind停止设备查找.
! G0 F/ }" i% I% i$ @CUPnPImpl::DeletePorts删除端口映射.