UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   ( |" I/ `# P  T! t- \
3. 
   ( a) a+ s& R# a; f; |. T) _
4. #pragma   once
   
5. 
   * [5 _) o7 k  A4 l0 G& m
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   - F, J3 H- \, A- ^4 t
10. public:
    1 X- T5 h* T, U; G  Y
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
    
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
    
17. }   UPNPNAT_RETURN;
    9 H+ O8 G; l2 ^0 H
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    4 @) @) ]: H2 W
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    2 E) N7 E  m) o% y/ e
24. typedef   struct{
    , Z1 O0 w0 C# w0 k4 M
25. WORD   internalPort; //   Port   mapping   internal   port
    5 K: R- `( L. E# ~( P+ B2 `
26. WORD   externalPort; //   Port   mapping   external   port
    % z. J8 ?# _8 d0 O/ m0 b
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    : U4 t1 ^+ y; ?- ?
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    4 }' S0 E7 @# q+ u
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    ; n' M8 ^: e5 k0 j8 C
33. 
    4 V: k6 m# H" u( }! b5 `& l
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    3 R3 j8 ]. m3 Z2 F6 x# J7 Z) m
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    4 b: G0 w0 g9 `4 ^, W/ Q
36. void   clearNATPortMapping();
    2 m) g& V7 B% J3 }5 D# r
37. 
    
38. CString GetLastError();
    6 v$ m  C1 b  D! |
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    1 v% p8 n0 \8 d  u
41. bool IsLANIP(WORD   nIP);
    % l8 a* H  D0 q2 Q( o  u" p: D* ~* _
42. 
    7 E0 |# U1 b/ \6 H. G; D
43. protected:
    % V) `0 T; j8 V4 _3 d" k; O
44. void InitLocalIP();
    ! I0 Y* t  R/ G  D9 P. ]1 c
45. void SetLastError(CString   error);
    
46. 
    ) w8 w. s/ b! w
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    $ W7 b! [0 `1 e: l* ]( H" T4 h: t
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    / J# \6 c& c" C* x; w, F4 B: x7 ~" E
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ! ^; y$ b/ C4 G8 u2 J( d
52. 
    
53. bool Search(int   version=1);
    9 U# T: }- w% S$ v. Z+ f/ g
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    + i( A" K3 ?( l, R: T
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ' I  c: n3 k7 Q3 F+ i" W
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    % j3 s  P# ~8 _8 L
60. CString m_devicename;
    
61. CString m_name;
    1 M" w. T# C, f3 H  D2 A; W# ^# t
62. CString m_description;
    
63. CString m_baseurl;
    ( A1 _+ ?% i2 a& h" n/ u- g7 ^, B* f
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    / V6 K! T# n- v: A
68. 
    
69. private:
    2 J: q: T& G1 s/ _  R; [
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    2 A2 F0 o$ S' H
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    1 T& R7 A* W+ h  f$ D
75. 
    9 i3 z1 K1 j" F( m) X+ a
76. bool isSearched;
    ( T; Y2 o9 \0 I# {1 }; g, b
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   / ?& L: K% i0 Y4 Q% h  R
2. #include   "stdafx.h "
   
3. 
   ! x- X+ S2 T1 J2 j( H
4. #include   "upnp.h "
   
5. 
   7 e7 a4 ?+ U; w$ j, g9 I
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   - B( A% U6 ~8 u3 D2 V
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ j7 g6 t" t) k+ G) ^
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   " G% J  `! Y4 {2 A
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    / M4 N8 z* G# c  y6 P9 t" F+ T
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    $ `0 j% i9 [8 G) A
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ) _. e* b8 @9 u/ ^- f* O* x
15. 
    
16. const   CString   getString(int   i)
    
17. {
    & F9 ?6 B( N' d+ q$ U  [8 G
18. CString   s;
    
19. 
    2 `: n- l2 t% j% s1 h
20. s.Format(_T( "%d "),   i);
    ( s8 }1 I+ W8 m# B% {& X" B( F
21. 
    2 `: Y2 B: W; h# i
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ; W3 F1 u8 T3 ]# ?4 L
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    , `9 C- S" n( M; U1 i( e3 O
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    ! e& q& @1 i0 n5 T) C
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    0 D/ g- B  E/ o3 m7 J* R- v
33. }
      d( P1 E# ]3 D- @/ m4 ]
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    " g$ k8 Z- T5 g9 F' y
36. {
    7 u; X/ j( L! }$ L3 H2 \5 u
37. char   buffer[10240];
    . D' x% u- i3 M
38. 
    # a, {5 h- T) C5 k/ @- z
39. const   CStringA   sa(request);
    9 Y5 y% B/ |8 G1 j) o& [
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    4 R0 I1 G" u  f
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    1 t: @1 x/ `) {/ P% ]; m: N2 M' S
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    0 K* U4 N: k9 p7 E
47. sockaddr.sin_port   =   htons(port);
    ; N0 F3 E, K. m
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    : F2 @8 m* I; q# y: v
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    / a- D+ c, r& ?: M( e
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 `$ r2 A, q' B, W
53. Sleep(20);
    , q! g! U; X2 z$ V! \  H
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    - N& u$ s! Z5 d9 A4 a  r
59. if   (!rlen)   return   false;
    5 e6 Z6 Y' H* ?
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    6 l* K& m0 S' m. C
62. 
    
63. return   true;
    : A7 G7 m" }6 }  ?1 `& c
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ' L& a+ m+ `6 f5 Z
67. {
    # G* T" D5 ^4 d. Q" f7 _# s
68. char   buffer[10240];
    1 V: l! S3 \( R1 K+ p/ h
69. 
    
70. const   CStringA   sa(request);
    2 ?2 y# ?, P8 M8 O
71. int   length   =   sa.GetLength();
    4 `  k5 b% P4 X. I
72. strcpy(buffer,   (const   char*)sa);
    ! S& X+ x( Y: }5 f
73. 
    + X* u/ `: `, K& u
74. struct   sockaddr_in   sockaddr;
    1 @4 o0 w4 Q; z# A& m( }4 Y
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    : r# h! F) q; M& l, \4 |) ?$ j' {5 k
76. sockaddr.sin_family   =   AF_INET;
    : I0 v7 L3 y' |1 c% g
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    1 s3 V0 j1 h7 w) W$ ~
79. 
    1 N& O, ^5 n% o8 k# k+ J7 a8 s3 w
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 O: p- Q, E, Y4 G5 r" @! ?
81. }
    7 E0 @0 f7 L) `0 q7 ^
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    ' {: H- A! l. z  d  L- x  F
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    : j. [, t! B) Z$ I" J
88. 
    
89. result   =   response;
    % f. W6 Q6 N' E& }. H) k+ g% a
90. result.Delete(0,   pos);
    1 H8 x" E5 n" A/ Q3 E' T( n; p
91. 
    " m* L4 [& B9 e
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    * `  b, |- N$ `9 J2 K/ F2 h
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    / ^4 R+ x! v) D' c2 }1 l" m
96. return   true;
    6 ^: T% ]( W1 ]+ w5 ?! @9 i
97. }
    
98. 
    : h$ z; e: `8 n! J
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    8 \6 |" Z1 }* `2 m, _% E
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     8 `, U# m: B' L" x4 o( l
103. CString   property;
     
104. 
     - Z8 l9 C% Y3 i1 f( n1 a+ f* E! g
105. int   posStart   =   all.Find(startTag);
     . X5 T5 E% i) e# h
106. if   (posStart <0)   return   CString();
     4 v1 y- k* f+ Q& ]
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     * G5 N5 v- |: K+ d9 K
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     6 c2 k9 B* `# q" l; s
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     8 D& q) h6 L) y- a1 M7 A4 v3 B# w+ l
125. while(pos){
     ' e  c) G% z( u5 f3 _& H2 b, c  Z
126. search   =   m_Mappings.GetNext(pos);
     # L$ I8 Z8 [! G( p1 _
127. RemoveNATPortMapping(search,   false);
     
128. }
     + q, e0 P" t" g; A
129. 
     , ?) N, p8 j1 S' L
130. m_Mappings.RemoveAll();
     
131. }
     & F' J8 L5 U  a6 P) u' j
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     # p. ?( u, w* m) F' T
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     0 Q5 [8 m, k  _+ m. E
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     2 D' _3 Z& B) X& s! j6 i
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     0 f2 ^5 w: A; P# T) s2 \% i- Y4 l
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     * k, e$ V- i) |. R( l; h' ?( P7 a
152. if   (!(i%100))   {
     ) O# h; _$ d0 }& q# c7 w
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;
     - B8 O8 g' S% ~8 H9 T
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);
     5 H# l- V9 d4 T$ N
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     $ N9 b- {; e! P4 ]' W7 Q
159. }
       ]5 U, n. A, W# P1 x3 z
160. }
     
161. 
     - b( Q0 P" W, i! Z' ^+ ^6 X
162. Sleep(10);
     $ M! P, S0 ]( Q) k, {% f
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     " O2 Q$ `9 G; X6 j  G* t
167. closesocket(s);
     1 s' ^7 }" l& j; t) x
168. 
     ! {: }% a' P; F/ A: x$ f$ Q9 g
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( `4 c" X% n4 a  |, l
170. CString   result;
     ; `# }  {( N9 M: k+ d
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     & t% h+ j$ `+ y1 K9 K
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ) a, t" N: Q' A( [4 `0 c
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     7 P( d3 D/ }, ~, @4 F
178. if   (line.IsEmpty())   return   false;
     ' C. o7 q6 q5 ]4 ^7 g
179. CString   name   =   line.Mid(0,   9);
     : k0 Y# Q5 @8 r+ k
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     5 x2 t* M" [6 Z& R# m
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     5 j% I4 H# P5 g& i3 D9 I
185. return   GetDescription();
     
186. }
     " I4 c- e5 e1 R9 K3 [5 s6 c9 @* A
187. }
     
188. }
     
189. }
     
190. }
     3 m9 Q$ o" U; r- s
191. closesocket(s);
     % Q5 d% `! |8 ]; t( @1 V# a
192. 
     0 h. E" K0 I% Q  P  c
193. return   false;
     
194. }
     , m0 {5 [' b* X) `

复制代码

天云上人

以下有关upnp的接口来自emule，
. Y0 f5 t: b; `7 o, x

* I! z0 }* V) v( H2 c3 a' d///////////////////////////////////////////
! R7 n7 j( l7 L8 U- Q4 U; H//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

2 G4 S# U, t' T) |5 F7 Z
#pragma once
' i- r  S1 H# x0 c#include <exception>

# `% w! z* n% X. N- b2 x1 Z7 a
  enum TRISTATE{
$ v  U$ C! o$ [7 w) g        TRIS_FALSE,
3 {4 u7 Z; C/ E+ n7 a4 M        TRIS_UNKNOWN,
1 I8 p; }4 J1 Q8 r/ h; ]& E        TRIS_TRUE
};
4 t4 W* f3 L( N$ d

' m9 }2 V, u3 U0 T3 aenum UPNP_IMPLEMENTATION{
; m) s2 Z3 o: H' T: s4 _/ L        UPNP_IMPL_WINDOWSERVICE = 0,
2 n, D- ~& z: E* x: I3 a        UPNP_IMPL_MINIUPNPLIB,
) R' d, R. [0 \  Z" r0 t        UPNP_IMPL_NONE /*last*/
4 `& G) N+ i  B# r};
1 w  x$ w; J/ y1 W, H. C
3 h, p9 {. M' m

7 F8 x- V7 ?+ }+ G7 X
+ q9 B1 D% y* }8 }9 g) M2 bclass CUPnPImpl
{
5 m8 G, I, r* }0 s% A5 e& L$ Apublic:
* x8 b- r/ b5 J# `        CUPnPImpl();
# C7 x( Y7 P7 K        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
" k1 X* H: H! L. c6 m5 k6 J1 _( d                UPNP_OK,
! X- ]3 \+ M, j+ e$ y- M% V. O                UPNP_FAILED,
6 u7 W, f- \6 f$ z: y                UPNP_TIMEOUT
        };
0 D0 {9 O6 U$ q3 F. E
* L5 K. x$ W4 |) [! _
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
( c# }  K) c3 n: M        virtual bool        CheckAndRefresh() = 0;
* Y5 _3 i0 F& O/ h        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
. j2 o' L, x$ X% Z$ X& Z" b) m        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
6 e7 _* }6 ]1 w; g# z! M
$ `8 U7 q' K/ l/ p6 X8 u: \
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
5 a% I/ i; f/ n: Y5 X: s        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
! S8 F- ~+ w5 W. G; `$ \- y# ?protected:
. w4 o  q! e, I7 q& o8 m# Z  N( `% h        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
( n5 s# j; I" B1 s2 d0 Z' Y! \        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


1 V* e+ W( }! h6 f% g2 eprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

; N# a4 p, M* |0 h
};
' C2 q' [/ R; N- J

- c  S& I, }6 c* F) Y// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
% D/ F5 L4 h6 E' R- |& {+ c+ k{
7 R  V9 n9 V; D$ Bpublic:
  X7 r; N1 `$ o( _2 _) g, i# ~        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
7 d) C( f0 ^% M9 t        virtual bool        CheckAndRefresh()                                                                                { return false; }
$ j' M. g7 h, l2 S5 l        virtual void        StopAsyncFind()                                                                                        { }
. h5 q  J+ ~  N" Q) C+ u        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
6 T$ d( I2 k8 z% y# x2 O# I+ V        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
, ~9 U* ~3 W' }2 i};

: b) R& b4 i: E/ |$ z' E
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
#pragma warning( disable: 4355 )
9 X* C* p+ f  \# b- m/ G% ~5 J2 i

#include "UPnPImpl.h"
& F: _: k" q: G. b: f#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
% Y) R. g7 c. i/ U$ n#include <winsvc.h>
" j% G- T2 k$ z* B

#include <vector>
#include <exception>
#include <functional>

# s" A$ o1 o; a1 C! o$ A$ l
! v' _3 e* z) E" p7 s

* \4 p7 S7 @' m- R, n/ Utypedef _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;
/ U9 R9 q) Q8 wtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
. K! t1 g1 K9 N2 H& p; B6 vtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
4 Q2 a: H: a  }( M# E5 Q0 ~typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
( `" k5 G! ]8 ]5 F8 z8 C4 l2 [9 k# L  Atypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
) Y+ C: y; U4 ?% S0 }; A

' s9 R4 Q* s. H3 f. w0 a2 ftypedef DWORD (WINAPI* TGetBestInterface) (
) X% d  L: r: r& I8 Q2 o  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
0 P7 O' S8 L' W$ y1 j) W& P);
3 y7 _6 E) C& C
4 m! j+ M3 z! [+ p" Z' P8 K  S
typedef DWORD (WINAPI* TGetIpAddrTable) (
' R& U  C5 v1 {$ L  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
* ~' m. b7 {1 H4 c, A  A, @);

' b. S' h2 J. z! _# \
" i& d2 y1 w1 g2 h1 J7 h& _typedef DWORD (WINAPI* TGetIfEntry) (
0 N1 M1 G. Q9 f4 A* I6 {  PMIB_IFROW pIfRow
# Y: N0 p& R9 Q" U! T, @);
& z! @( X" R6 p2 v4 @' ]! x# d
% @8 r- c1 M4 V5 g
CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
- M7 u) g$ `5 P/ Y% o

class CUPnPImplWinServ: public CUPnPImpl
3 K6 [3 g- k  c: [" H{
' z4 J4 f) F0 M        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
! S+ s" o7 e; a- Z/ ?// Construction
: Y* Z  W" u( J' Opublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

2 i- B6 `' o; b6 z) A& r
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
0 G4 H, D0 Z7 L" A2 e8 C& @        virtual void        StopAsyncFind();
6 f* d: ?; L3 e3 {" }2 |: i( O        virtual void        DeletePorts();
7 J. ~  y0 O6 r% i# u0 X6 B& H4 ]        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
( G" e8 M: P$ }5 q# R" X- `1 o

        // 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
; r; C7 Y* E# x4 Y        virtual bool        CheckAndRefresh()                                                                                { return false; };

% n$ a& N$ d# k% W8 `- G
0 b2 z8 `) k; J# N7 tprotected:
& a, z1 W# P, X7 Y9 c3 T        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
' [; m. K$ _2 b' X9 F        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
. @* {5 Q% p' G* L        void        RemoveDevice(CComBSTR bsUDN);
! r, i( M7 h; g8 a* M; M) l- _        bool        OnSearchComplete();
        void        Init();

8 J( U  q& y7 i9 {' x. o
2 k3 i% Z6 f7 Q5 m" s8 h        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
" e2 |' S6 \7 J$ r2 A$ N& h' i                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
( h. Q9 ]* p. _1 c                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
# C; Z* c  n) j7 `+ d/ a                {
( t/ S+ a. F( s; M' Z/ i                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
" T' Y4 x7 b$ K5 a5 V- i0 T6 @                return m_bAsyncFindRunning;
; n) J4 ]! E2 F1 l) M) X        }


        TRISTATE                        m_bUPnPDeviceConnected;
+ r) k1 ?- h/ f

// Implementation
        // API functions
1 c0 e! a8 W* C6 |: q) S7 @        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
: r/ d1 m5 F6 E" ]7 @        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*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

1 w0 g0 Z  R( L" t
- B( f6 w  t& L7 c2 u        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

% k4 d7 U4 q- X# E7 P
  R8 G! d. l! g  Y1 ]" e        static FinderPointer CreateFinderInstance();
& H+ K3 J) ~, ?; R        struct FindDevice : std::unary_function< DevicePointer, bool >
; Z( K4 E2 w" z1 z8 S' g8 ^        {
$ t/ g1 g% G( W: x/ Z                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
2 \2 M% F2 A: {- `4 f, o                {
                        CComBSTR deviceName;
7 ~" @# ?4 v4 M+ ]- ]& P                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
; U% o/ D/ T# A

                        if ( FAILED( hr ) )
" y( ]- f& c# M5 j; {* H5 v( M                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
7 p' B( E$ H# A8 C4 L                }
                CComBSTR m_udn;
- d) Q7 O; C4 O3 s. X1 u1 a        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
( n. n" f* h4 \4 M. B3 g        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
; q% E8 r- f9 `        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 O- w. E: ]9 z, p5 _5 g                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
' c' i5 V. w& S$ |" B6 Q; A

/ P2 {  a) D# B3 h# S# n6 ^        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
. S  V* m/ I; z5 w# I2 N( ?7 J        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);

+ ?  H* _2 ^1 b: E) M
" j; `$ x( b* J' x// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
) p$ g' x, u3 k; k. ]. f) V        std::vector< DevicePointer >  m_pDevices;
( Z! m, z* t& P# Y6 [  T        std::vector< ServicePointer > m_pServices;
, M. {3 e. G0 t        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
; u3 O# F, k8 l) y# H% P  o
0 [2 d. u( L$ A+ {# w: e- o- B* r+ g
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
. l9 V' h+ B. `' E        bool        m_bPortIsFree;
4 h  O4 w+ {/ h* t        CString m_sLocalIP;
$ r. h! p5 B. i( ]7 h1 t        CString m_sExternalIP;
2 c: Y% c  P1 \% l5 S9 V        bool        m_bADSL;                // Is the device ADSL?
% j: ^; n7 A  L. Z4 ~8 }        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
% S1 ~/ }8 C- c+ I# j4 Y+ ?        bool        m_bAsyncFindRunning;
& s! |5 R2 p! X! _3 l        HMODULE m_hADVAPI32_DLL;
( b& }; P: }$ H; W3 o( B1 W* r        HMODULE        m_hIPHLPAPI_DLL;
' z4 i; }% _/ M% V# t0 K        bool        m_bSecondTry;
5 o  [& L; t; M2 y3 N        bool        m_bServiceStartedByEmule;
8 `+ ]" U0 S+ m1 c4 |5 G3 }        bool        m_bDisableWANIPSetup;
1 g% ~0 E& }5 P& P- s# o        bool        m_bDisableWANPPPSetup;
6 ]' ]' u! b7 i% R* W: H& e

0 |) C3 _* b7 F# F( r5 Y};
6 @6 T/ G" F7 R0 u: B1 D$ t

, S9 M8 }7 E, k& K# k// DeviceFinder Callback
4 w9 M4 X+ u# b5 |/ @0 ~0 a, O0 C) ]class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
) a* l. y# f1 _{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
" q" Y* S5 ?; u: U7 A( {        { m_lRefCount = 0; }

! g$ S+ y, v: e; z; }0 n
/ A- o1 a) w) K' Y2 q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
; e( q/ w5 ^" B. @. l/ i   STDMETHODIMP_(ULONG) AddRef();
& j6 D' g7 a3 C# l   STDMETHODIMP_(ULONG) Release();

& ~0 U2 v( h/ Y" U" i
; ]- k3 _0 v9 X8 [: j. z// implementation
5 i  d4 `; ?- v  a+ a; v6 U% P: S1 gprivate:
  W6 n  E2 P5 k1 K. I* i  Z. e8 W        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
- g: d4 c' Q( q
( i' k! D- m# h0 \1 X  B% F9 k: v8 K
private:
        CUPnPImplWinServ& m_instance;
2 J- A% r% E" g( r' X- T        LONG m_lRefCount;
};


// Service Callback
class CServiceCallback
7 O+ E* t' S5 q% g* m% J        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
+ W9 [" f7 B& U; [, f: R2 J6 v        { m_lRefCount = 0; }
; N2 I" K/ |: u   
; B8 i3 Y: E7 {( Y: w+ s2 W   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& v% u& i2 k, b& N( _3 m   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

/ g0 w% o- ]) v9 J! W, S
// implementation
private:
6 Q- c+ W* J- W. {8 Y& `: D( Y: d        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
; B! j, \5 \* N; Y8 N5 J* V
  m) ^5 ^7 K, }
2 E+ r  X; ?  g! z- H( Vprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


/////////////////////////////////////////////////
7 V0 F! ]5 h  d' l' y8 ]

: X3 F2 S, B" V使用时只需要使用抽象类的接口。
5 n! G, [- X' W0 w/ h" yCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
3 A7 i! P8 M1 p& [CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.