UPnP

天云上人

/*uPnP.h*/

1. 
   - P: `0 s4 d! ^- M) p, {# y5 D
2. #ifndef   MYUPNP_H_
   # N9 m. x3 ]; r7 X3 O; S. l% D
3. 
   ' w/ |! h  C# G* t2 c2 t4 {
4. #pragma   once
   1 H- @1 O" |; e- l9 F, O- J+ X
5. 
   
6. typedef   unsigned   long   ulong;
   / W+ F9 Q  y4 p3 g! L- ^
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    0 d+ h# z# Z0 q1 J2 Y4 Z
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ; b0 J# L5 z0 p9 V& ?) J
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
      J1 `# {$ l# B$ E
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    ' c" K" t. Y5 p- g6 x6 X$ s
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    # ?( P8 s, Z/ Q  A
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ) Z9 S4 r6 `# b( _+ K. S* c( O
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    ( t( p' y# d4 j% m% S9 X9 _3 i
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    . l5 O5 D, I: f2 M) I, i
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    . o# W( w7 o& t4 {5 ^$ x4 v1 s# J5 _
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    5 n5 ]9 u- l- a4 ]
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    $ ^. p7 O7 ]  p9 v- m  k9 h
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    5 F+ A7 c( }7 |3 d/ y
42. 
    % ^4 n5 e4 a0 x' l, H( k+ g
43. protected:
    
44. void InitLocalIP();
    3 W9 C. d: z, v6 M
45. void SetLastError(CString   error);
    4 K3 @/ @# w) ^' x; K) |  C3 W
46. 
    % X0 P! m; j/ }/ R
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    4 o: [- {2 U0 }) C# @, O
48.       const   CString&   descri,   const   CString&   type);
    ( E$ B2 K7 D7 W+ H/ K2 E1 @
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    & M' n) ^1 V* s5 r# ~0 ?
54. bool GetDescription();
    % O* U' \7 k7 F% O
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    8 p8 _! u+ d; D* [6 l. s7 Z1 K+ x
57. 
    ! m! j+ g" @3 {( Z( r
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    7 N8 `* X4 Z2 D1 r3 m5 q+ [5 V
59. bool InternalSearch(int   version);
      r. V; V# z0 V
60. CString m_devicename;
    , w, g; s. }$ U1 F5 Q! \' f7 g' w
61. CString m_name;
    / u/ q$ ]* b# F
62. CString m_description;
    
63. CString m_baseurl;
    $ Z/ L, p8 Z9 O% o( P/ g, ^
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    2 g) k/ Q  q2 k9 O% M+ |
67. int m_version;
    # S' b  r3 K2 e2 l; n% U
68. 
    , J6 L$ k  d& C) i, Q) k
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    & @% k% e7 p4 j; [7 m& v# f
73. CString m_slastError;
    7 k( F( z5 }5 ]  ~/ a" u
74. WORD m_uLocalIP;
    ; `6 o9 V# L/ Y! o$ q, [. ?
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   * g: H5 J2 C) g1 w. Q
2. #include   "stdafx.h "
   
3. 
   3 B- y) V, U: w
4. #include   "upnp.h "
   
5. 
   # v" q# \! V) b- |9 [
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   ; x- G; W- U% w' _
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   3 L1 X" ]2 Z$ ^. u3 U
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   , ^5 @6 h4 ~9 |9 Z: ~
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ( J- ]; J2 u! _+ `; U! ~+ H+ F
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    9 `2 N$ o9 O: S
15. 
    2 }1 s! U: Q3 ^& {" @& K, ?
16. const   CString   getString(int   i)
    " l1 [& J& e) b% Y+ L& r  a
17. {
    - v# D  a5 s' P6 c0 a  p
18. CString   s;
    
19. 
    9 x# g+ h$ d6 p% `, `0 x# ]( q
20. s.Format(_T( "%d "),   i);
    $ Y6 p; t2 B+ z+ `! l
21. 
    
22. return   s;
    0 ?- P; g) i0 P& S) Z
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ' \6 P2 S/ I2 ?1 [0 C
28. }
    
29. 
    ( y. S  b0 K" V/ ~& Z
30. const   CString   GetArgString(const   CString&   name,   int   value)
    - R! G8 G6 [  p" T: u3 @; u
31. {
    " f% {+ i% A9 R% s, D( m
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ' p% x) s9 c, h+ X) R
33. }
    - Z; b1 s* o* I5 L9 h/ R8 W3 m: B
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    % L% |% ]6 ^) S5 T1 I+ u
37. char   buffer[10240];
    2 v! c3 E* m/ R3 o0 d7 H1 |
38. 
    , @# z. V7 I7 r" z
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ) O6 i8 C, w+ g) l! i
42. 
      c3 L# M! \: Z+ c4 A7 g
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    6 h% w3 R/ J2 a& m- J9 A. n3 k% j
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;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    6 D7 D$ `8 G" v1 l
50. u_long   lv   =   1;
    ; z8 _1 e  S" i: N/ g  ~
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    7 O9 V) T/ h# J6 @
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ! w6 b0 ]; u6 @
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    5 I4 k! K2 Y. G4 ~9 g
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    0 P$ i1 Z# F: w2 b- c
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ( i, Q7 h9 X. o( p1 i/ E; V5 a6 b, R
67. {
    ' H. u7 {4 `5 |5 ]5 k7 T
68. char   buffer[10240];
    # M4 e! Z: a3 q+ b: [% P
69. 
    ; @9 `0 g6 T: P: V3 Z; _
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    % M$ [0 |1 T( X
74. struct   sockaddr_in   sockaddr;
    6 c3 x: O& E6 R/ i: o
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    # s% i; V# G9 T* m3 w: M$ @
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    6 J$ A$ _% u/ u4 n4 ]' X
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    * e1 q5 Y4 V2 M; G% _  k( y3 y
81. }
    ! K/ t/ b& E% Z/ Z
82. 
    2 B' b8 a3 j% V' A# |+ r
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    . `  j  M& H, h9 \
85. int   pos   =   0;
    
86. 
    ! I  i0 i  I' w" C6 \) r
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    % U& I0 K5 w! s. }
89. result   =   response;
    2 ?( ^  A- a5 o2 m* i# z
90. result.Delete(0,   pos);
    
91. 
      n* O' b: N/ v& o* |& O1 s0 i
92. pos   =   0;
    ! u; g: i. p7 `5 G
93. status.Tokenize(_T( "   "),   pos);
    ! r+ u# d+ u5 L$ i) [# g1 J
94. status   =   status.Tokenize(_T( "   "),   pos);
    / ?# ^/ y( M/ i9 I
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    # N) l1 {+ W- F$ O) f) p
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ( n  F: C& I1 ]4 x0 n, L  M
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( N8 H0 J  V  D& p7 |0 ^2 q
103. CString   property;
     
104. 
     ! |( N# j% Y& z" z
105. int   posStart   =   all.Find(startTag);
     % S' q, O, [. o# I* q
106. if   (posStart <0)   return   CString();
     * f. b& k" `4 L/ L8 V, S
107. 
     ' i5 ^* x: v$ J! I: N/ ^
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
       r3 _* \6 g$ x; C  m1 t
110. 
     7 ]' K/ v" k6 K2 f) \# M3 p
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     # s2 H3 y% e6 v/ ^4 R8 L
113. 
     
114. MyUPnP::MyUPnP()
     8 ~- A: ?! f3 S1 t3 B- n1 J/ _  X
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     6 x9 u5 \4 Y; b. _# h4 h
118. isSearched   =   false;
     
119. }
     " k4 B" Z+ U1 J# g0 s* b
120. 
     
121. MyUPnP::~MyUPnP()
     ) L! Z4 y+ @" B8 |" G) n/ d! x! A
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     * t4 @" b; |* K/ k) n# G7 ~+ Y
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     1 }7 F# Y$ R, v0 @/ `- v
128. }
     , H1 V) ]  \* l& `1 a/ V5 V# ~# s
129. 
     , m: L; T3 a" N0 \0 m
130. m_Mappings.RemoveAll();
     3 y& j* b* ^6 C9 c8 ~+ ^
131. }
     : x0 }  r7 y. ?# B# M
132. 
     
133. 
     ) M8 [( _8 L, o/ t2 o
134. bool   MyUPnP::InternalSearch(int   version)
     + q) O' {3 ~0 P- \; X& r9 M
135. {
     # y5 J3 o4 k7 i
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     0 e/ P9 ^8 _, h' E
139. #define   NUMBEROFDEVICES 2
     7 o; ]/ g  W) Z6 T0 G0 s
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     1 F# o( k' l: h
142. {UPNPPORTMAP0,   _T( "service ")},
     $ L' m& O- O* g( z
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     1 }' ?9 s" U3 B
145. 
     2 f5 k$ t' v! Y+ a! e
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
       A5 |' c9 w' c5 O  ~8 n9 w+ o. K
147. u_long   lv   =   1;
     ( o9 s' g) i# ]; p& n
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     " g0 G5 \* h% o) a7 X
150. int   rlen   =   0;
     + S2 g- C% r$ w4 c% g7 {& k9 g4 s1 J
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);
     
155. CString   request;
     # W% u" Y0 I5 h; o2 ?& |- {
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 "),
     : e; S$ r, X3 H) D; P( c$ R/ X9 R
157. 6,   m_name);
     ( A* m1 r1 P: P
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ! }% l8 e& l' S( a1 N
159. }
     2 R" X9 o1 c! a# `
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     7 @3 N1 F9 N+ Z; }" d
164. char   buffer[10240];
     4 W1 o" T& ?6 U+ M! |- u1 k
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     # D0 s4 l6 X" N. A7 }6 g0 [
167. closesocket(s);
     8 n. ~/ Z: J2 `+ J4 l
168. 
     , t9 c7 V9 w9 H: S# J
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
       Z, l% p4 T/ C; U$ l- I0 t
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     $ D' i# ]5 V9 l( H" g9 F# `0 ?
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);
     4 _! n( E1 h; J* b3 P
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     " s, z, |; ^1 x3 e
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     ; w& b/ R( u" o7 P
183. m_description   =   line;
     , K5 b" v* N* e
184. m_description.Trim();
     
185. return   GetDescription();
     % o( L* W# o- J- }0 E
186. }
     
187. }
     
188. }
     ) v8 m7 r& p/ V6 ]+ T: ]0 `
189. }
     
190. }
     $ T0 c* p6 E$ P
191. closesocket(s);
     , R. W2 [2 \4 b6 M: c1 F
192. 
     " S( T7 k' B9 _3 l
193. return   false;
     
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
3 t, l# F2 k! \7 ~
4 Y# @8 I" d; Z
0 G; X. n  y  D: T#pragma once
#include <exception>
0 e3 z! D% p, [! \5 o

  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
  k4 @. {9 o+ d! t6 T- w/ r# C        TRIS_TRUE
};
* M1 {4 R0 Z* C2 o  P- Y

  O$ x4 D, B' x  p* y# n# lenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
2 K4 F. m: O1 I" c4 E4 i. f& b        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
1 J$ y# K% ~$ a, N. P, F& {


5 }# P; u) s% W4 u+ g1 ^! R" _
+ v9 x( ]4 ^" zclass CUPnPImpl
{
public:
8 R/ A& i) z* z        CUPnPImpl();
        virtual ~CUPnPImpl();
1 d3 g2 I* d! e9 {5 E0 N' h- I        struct UPnPError : std::exception {};
+ d/ Y' y: M7 N# E        enum {
$ `$ v& J. U$ ?- {                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
        };

6 x/ W, G6 Y5 y0 k! Z7 y  @
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
% ?" N3 J- \4 e        virtual void        StopAsyncFind() = 0;
8 K( ^% h5 w9 `        virtual void        DeletePorts() = 0;
: S* N6 O5 O. c( Q        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
" s" H4 u5 Q6 F       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

6 A/ t7 }# S# K/ S! L, w2 q
& A: V% F5 T, M7 d7 M        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 c6 W8 b  H. K3 ]- }9 W& z) G        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

' O" W  x) @! i) S
// Implementation
1 J3 Q0 V1 ?9 U/ Fprotected:
- K6 _+ G1 C7 _* ^7 z8 l        volatile TRISTATE        m_bUPnPPortsForwarded;
; q! g8 V3 R/ t        void                                SendResultMessage();
3 o6 f& o8 j. W. v) L        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
/ A' T& V4 l. j6 B        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


& [. k# U! T" Iprivate:
1 D# P! A9 D5 D        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
' g+ \) M* y4 K2 o5 f1 X1 J
# F7 u, ^# E0 J3 k' i" G) V# T/ b
};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
) K1 z0 G3 w4 f6 V4 K{
- F  B) r$ V- E; q% Jpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
& R7 {3 [3 A, h# G9 N  ^        virtual bool        CheckAndRefresh()                                                                                { return false; }
) w  x5 h  f) x, p        virtual void        StopAsyncFind()                                                                                        { }
( V3 V/ {; o8 E" h! e, l9 G$ L        virtual void        DeletePorts()                                                                                        { }
0 b. X2 L( y- D( D0 j        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
) z, m: ?' s! y/ F
, M, Z- Q2 o( m9 m
/////////////////////////////////////
3 N) ~* ^' ^0 R5 t- B//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
! y' I4 C0 _+ `( @# W- k. u; @- R" x4 k#pragma warning( disable: 4355 )

  _2 }( f5 H0 @* k: |
#include "UPnPImpl.h"
% X, R5 G' m) D3 |& v#include <upnp.h>
#include <iphlpapi.h>
4 k5 i# V  Z5 T3 X9 G#include <comdef.h>
#include <winsvc.h>


" f2 x, _; H9 F* W#include <vector>
#include <exception>
' y9 i  k, F$ L% C#include <functional>
2 F) O8 L+ C8 G! b
3 P9 P; Y8 D) f. f: n/ |
# x  e3 f' z! N  d- Q5 t1 y6 V
. ]2 q4 k+ G( `& \, J$ n
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
3 y/ \+ S2 f+ R- etypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
, M, ~9 e% J0 x5 `6 ptypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
3 o  p4 A- s% Ntypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 @" O8 m" H9 M. @* Q1 Ptypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
7 C$ k0 }) _3 D3 E( }, ~5 a/ ~5 Ytypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

+ a9 [; B1 n* x! }
: n* t5 V5 K( Q( k; q" ntypedef DWORD (WINAPI* TGetBestInterface) (
7 ^2 |5 w! @5 R& u! D, I  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
/ d8 M+ j! j! ^1 M) J# Y+ G
/ X2 ^) b/ F8 ]
  i. W9 ?3 B. P. Z& J7 x1 k# @typedef DWORD (WINAPI* TGetIpAddrTable) (
* G2 \6 C! }/ c  PMIB_IPADDRTABLE pIpAddrTable,
- A* `" i6 H; Z/ F8 x. g5 P  PULONG pdwSize,
  BOOL bOrder
);

( B5 u" A& G2 |8 S, B: C
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
$ Q! w( X  O9 h);


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

5 k0 V6 n: M% m- ~2 _+ f4 I) f
5 t, ?; ^' b8 V8 G" j- c! Sclass CUPnPImplWinServ: public CUPnPImpl
{
) N3 F/ b" i- R* w& F4 N- `% ^1 t        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
* T1 j* A) r% ?2 ]7 X7 V/ L/ apublic:
3 y. h% w/ `5 {& l        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

; o7 X& i5 V" D" h3 Z* z
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
- h; [! a/ R- F7 \        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
3 |% V: p& X: F! |

        // 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
: B+ [) G- ^4 \: R  h' U        virtual bool        CheckAndRefresh()                                                                                { return false; };
* \$ j9 |7 Z) p" F5 ^2 x2 _
* Y+ C1 |6 z* o( J% r& m) j
protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
, ^2 J% K2 z) K2 G& E, L        bool        OnSearchComplete();
( T6 [0 s* a: l. Z        void        Init();
4 U* D: d) x( t% W" n) q

        inline bool IsAsyncFindRunning()
& ~& b/ N; Y0 l. P        {
8 u. |( S+ _5 J$ o, z9 u                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
& a* \5 J( U8 n; n* l4 E                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
, c, [" l0 `$ h: L4 Z                        m_bAsyncFindRunning = false;
                }
  _" F. y1 O* K1 @- w                MSG msg;
: }, [2 }$ G5 X& [. o1 v4 M5 s# ^                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
% \" ^/ N1 m( M2 c7 N8 ^/ d- W                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
& `# J4 U" k( f; ^                return m_bAsyncFindRunning;
( `2 o2 o' ]# }6 ]( x        }


        TRISTATE                        m_bUPnPDeviceConnected;

( H) n6 U6 W! f7 J0 m
// Implementation
5 D- n) f$ o  x& G+ ?. H7 j4 _        // 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);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
+ z+ P( [! c* B0 v8 i        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
/ L, B; D) T0 n& K$ b2 h: p, T

( M2 r8 u  d  [$ k8 m6 D7 g2 j        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
5 M& P, f. Q% A0 @        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
6 H3 s- n8 p5 ?  R, i) y                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
3 T9 _( T6 z0 U( l9 K

                        if ( FAILED( hr ) )
& J# }$ @! J$ [9 g9 }3 N$ V                                return UPnPMessage( hr ), false;
) v7 |) M7 i/ f  B  |. C& ?
# c. N: N( J2 ^6 }1 x4 ^- y8 \
* H$ w9 L' T& D" B6 R                        return wcscmp( deviceName.m_str, m_udn ) == 0;
; \& X  X) A" m                }
8 F+ q& a3 ~! v- |! z" v# q! S9 F                CComBSTR m_udn;
        };
) ]* d* k6 U# S9 C6 K  @* f/ I6 F0 \       
. v5 V! h( S9 n/ p- w4 `        void        ProcessAsyncFind(CComBSTR bsSearchType);
, w5 T9 k3 n: N1 `- S! c6 q1 W        HRESULT        GetDeviceServices(DevicePointer pDevice);
! M6 @, d3 [+ Z        void        StartPortMapping();
2 K8 V( N" ^4 q2 z. I) O& G        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
. Y5 E: B5 d/ i        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
/ F' ]3 w7 _  q- x6 P! V                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
. O& E: G: h7 `2 a7 g* l

' k/ X3 K4 e+ K+ q2 U        // 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);
- H/ j1 @% ]2 m# d) ?% Y. P' z        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
( z, }: i, @7 j# D! }6 m* G3 I0 H        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
, y/ s. \. W% P% f  C        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
) ?3 o: c9 \1 A% T
3 z; H% ?( Q* C% }
// Private members
. \: T: d9 b6 }, p8 d4 x: Wprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
( E# X+ b( Y7 W( `$ U        std::vector< ServicePointer > m_pServices;
% U+ Y" S7 z7 W, V0 K( j        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
8 k7 W- G5 A& j, }  e

        LONG        m_nAsyncFindHandle;
% S2 E( e$ Q6 U, R        bool        m_bCOM;
        bool        m_bPortIsFree;
+ U6 F( t  e" L( q, z, _        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
- ~/ ^6 L) e  F& ]2 n' S9 |        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
" A+ E0 C$ g1 Q5 F* I  b        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
$ t9 ?" S6 E6 X  s: D0 f3 j        bool        m_bServiceStartedByEmule;
. }# K. w8 b! b* @! _        bool        m_bDisableWANIPSetup;
* l$ j6 E  E- ~! V        bool        m_bDisableWANPPPSetup;
1 |& _8 h9 m$ L' i0 @3 L" `7 m
7 a5 r, v& O; \8 D3 \6 ]8 N
};

/ w# R7 X) f% }/ w9 x8 `
// DeviceFinder Callback
class CDeviceFinderCallback
1 _9 a7 |# `$ y$ Z        : public IUPnPDeviceFinderCallback
0 a: r* l# P  _7 \; Q% Q8 u{
8 ?0 s$ s2 ?2 cpublic:
& Z' ]* G7 @) K        CDeviceFinderCallback(CUPnPImplWinServ& instance)
5 Q* K* ^4 S  H6 t2 R                : m_instance( instance )
5 j- x% B2 @3 Y0 H" a( I        { m_lRefCount = 0; }


7 t4 p0 }0 T7 N" g7 P4 a8 j   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
$ i+ B8 o$ z5 {   STDMETHODIMP_(ULONG) Release();
) h" E) k9 ]; q/ f/ ?% {

5 i6 f/ ~/ z' n: X7 [2 o1 m1 F// implementation
private:
1 Z0 b+ U0 _7 [" N( h1 |! p        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
; l3 P/ Y! k# a        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
4 `/ l2 U8 O1 l        HRESULT __stdcall SearchComplete(LONG nFindData);
$ I! p" m1 \0 B$ B. q
3 {$ L0 S& r* O) T' I. O/ V
private:
4 T7 V1 j. d* U) y3 ^        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
! {- o1 D' Q/ O" T& v, M};

+ v3 S- g% S, g; Z! J; U8 t9 J
2 N0 Q. Y2 h3 r9 O" `. W// Service Callback
+ b6 e$ n% X; Eclass CServiceCallback
8 j/ ]7 |  x' w  i5 @. C+ B        : public IUPnPServiceCallback
) K1 \% \3 k' S4 M- V8 w{
& A+ k( l( l+ W& L' }$ Z/ l  D7 j4 w5 Epublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
5 c" u% y6 r) H' K        { m_lRefCount = 0; }
! |$ [6 b1 g+ u" Y9 t6 O   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
& e% @  @) F7 p$ k3 Z5 P

// implementation
: S! j( v! s/ }  [) G6 |9 s; M$ Xprivate:
4 ^/ c2 ~. X9 j0 R: T        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
$ ?+ R9 U8 X, D; ?& w        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
, ?% d5 T! W2 G2 q        LONG m_lRefCount;
5 r4 z' v# K: B+ e& I* V; q3 d};

7 h( q! k" m7 _6 ^/ e* F
/////////////////////////////////////////////////


5 Y5 A5 N% S) S& E9 j3 w使用时只需要使用抽象类的接口。
, n  I* ~; L4 X' f$ Q) bCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
3 m0 b6 g; Q8 U2 A  Q1 jCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
% i1 ]- G- }: t6 yCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.