UPnP

天云上人

/*uPnP.h*/

1. 
   0 }4 u* D7 [4 K1 `: x
2. #ifndef   MYUPNP_H_
   
3. 
   + n& R* t6 b: J4 m1 \) p
4. #pragma   once
   4 B; ^1 u' f9 h2 f. o/ f; s* h
5. 
   ) y7 I( t. ~+ e4 U
6. typedef   unsigned   long   ulong;
   0 r$ U1 ^) q0 Y! T
7. 
   6 k9 u" I) l7 m* u! H0 y: F3 N/ f
8. class   MyUPnP
   & I% [% v' v( n. L3 @% X- f3 C) e
9. {
   ) y8 C8 f& e3 m9 Q3 G6 R: {
10. public:
    
11. typedef   enum{
    ! q- R8 y7 x3 \9 Q2 I2 `, }
12. UNAT_OK, //   Successfull
      b- @) ?4 S6 B3 a4 Y$ E
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    1 d6 I0 A0 E6 i' U6 }9 w; ~
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    5 U, ?7 r/ {; w4 B3 X2 i
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;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    9 k" `+ n- f. ]6 c4 O
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    7 e5 a; D5 }1 ?
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    8 q* y" A5 G+ L1 A3 l4 K
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    5 w# Z- _* H( w1 s
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    2 w$ F( g5 O+ N- n& Y
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    * x! Z; G4 T8 [( O0 z. S, B
42. 
    
43. protected:
    + i1 u( E' z' c1 I+ T
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    . D' x7 D% V) {% ?5 @
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    2 d- o! d/ Q. d4 t7 r& I5 H  q
48.       const   CString&   descri,   const   CString&   type);
    , ~; w: q# {8 W# A, x
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    6 M- R! m2 K2 {1 F# X3 L: p' H$ r
52. 
    
53. bool Search(int   version=1);
    ) f* G5 g( x/ e+ @5 D( }2 g
54. bool GetDescription();
    6 o7 S; i  [8 A& Y
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());}
    7 O/ ~* r1 |1 ?; h
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    # w2 q, G; l* O# m0 p# p2 x; T
61. CString m_name;
    0 r0 G) @* \$ x* U6 W1 M6 K
62. CString m_description;
    1 r4 J  J8 f4 X3 i+ @5 o$ ?
63. CString m_baseurl;
    
64. CString m_controlurl;
    ; B6 ]( d% N+ j4 A" r; ?
65. CString m_friendlyname;
    3 Z+ o" v/ l7 |
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    - g3 V6 k. I; A) ~
71. 
    2 @, X/ v5 S1 s. K) D) E* q/ ~
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    % v: F% {: H% L, p6 I
76. bool isSearched;
    : z" i. C1 z: [0 u/ y& j% @
77. };
    * @5 e# `7 _" s# ]0 }4 P
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   , i/ x# w% G3 C! u) x& L$ q" L
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   : ~  n: n* r8 a) F8 k
5. 
   ; c2 X+ r7 B! M1 ~9 R% N9 n! S& d  o
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   # S: {8 L. m" j( q
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ! V' H  }! U; Z6 b# U/ P$ j$ y
11. 
    & {5 _5 b$ F" w2 d1 h2 X7 d
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    + a4 n* L) O! A) w* G7 F- ^# F
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    ( F. E6 l. W( t; z, y
17. {
    ' d) j& W  L* \+ l
18. CString   s;
    ) {+ H7 t  i2 }1 ?# a$ {! m  e# n
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    9 N2 o2 N7 o# `
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ( y" P" j0 }# B- \; ~
26. {
    * t. S! g2 _0 ^: e/ _  s
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ; V& O$ R4 ~0 @7 F5 @
28. }
    / D" {2 h8 i5 m; S
29. 
    ! i( n2 s1 |! J  W; b) @
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    - c; {* Z9 u% p" b3 Q+ `
33. }
    : ?4 U- Q6 y8 W  A6 s7 j, r9 g
34. 
    ! ?+ I( ?  {, A* A
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    0 t2 l6 i! _. w$ G+ y! u
36. {
    8 ^' Z* L" k8 s- _- v1 M5 l! M. O
37. char   buffer[10240];
    
38. 
    - R: y- Q, Z7 P/ z5 B
39. const   CStringA   sa(request);
    6 l) o" ]  L7 r$ `% ~
40. int   length   =   sa.GetLength();
    . Q+ H1 o/ g, P) g, U
41. strcpy(buffer,   (const   char*)sa);
    , u' [1 m8 ~4 d9 g7 S
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    , A+ i5 Q+ ?8 R$ s4 |
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ) N/ h, o6 f0 n
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    5 a, N  A" O4 l  i+ P- Z( R
55. Sleep(100);
    # y! r7 g- F# S4 B3 y# \3 J! }, ~
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    2 i# q0 h: N4 p. p1 G
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    6 b" x* k* L$ `- F, |7 g8 K
61. response   =   CString(CStringA(buffer,   rlen));
    9 L' ~% e0 k: @) a. R- i$ ?
62. 
    1 u# Z; U4 c! ]% n4 D
63. return   true;
    ( ]! D4 I( B- O3 ~
64. }
    3 L$ K$ f6 f  ^% C" b
65. 
    % u1 k/ G0 S" G7 L( Q& D- Q0 B
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    5 ~# Z, t7 ~; u+ a: h1 H! s
69. 
    5 l7 q1 f6 i4 E/ `8 B' j
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    . U: @/ g1 j) p6 O+ T( s" b
72. strcpy(buffer,   (const   char*)sa);
    , x; x  p( }' M5 b; [0 u+ x1 m9 s
73. 
    ' o% B" w" T3 j7 m' w
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    % a9 g! R- ^9 L' w
76. sockaddr.sin_family   =   AF_INET;
    0 P2 z5 v) j4 y* X9 z3 {8 T1 N
77. sockaddr.sin_port   =   htons(port);
    ! r" s: ~" t. O& X7 R) ]) F
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    0 C3 x# h6 T6 ]3 I
82. 
    , Z# S! h; X8 _$ c8 t
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    # a  ^! l0 G# g* M* a: g, a
90. result.Delete(0,   pos);
    
91. 
    5 f4 Z! s" H" }  e* Z8 g( F, T
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    2 [/ L& B: z2 R, s* D2 {
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    , H6 Z/ J" A. _- z
96. return   true;
    , M, I1 ]; u. [1 V; u3 s
97. }
    ! F" w) X; ?7 p4 r
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ! e! p' P# C* r/ V. L
100. {
     ( V; M2 P5 c0 h9 c& Y
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     1 }7 z* l1 m7 ^6 g8 k
103. CString   property;
     
104. 
     8 t" _5 z9 E+ e0 a- T2 |* Y
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     ' T! v- O, r" c# ?& y* [
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     ) C2 k; ?: W9 v3 v, s. J8 X) F, P
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     $ u3 D+ c8 v3 Q( a  S. Y# D& u! V
112. }
     7 H- ]( v$ i* x& Q' d
113. 
     
114. MyUPnP::MyUPnP()
     9 J, j3 l; ^2 D" m+ K: p
115. :   m_version(1)
     % g+ \; N- A5 }7 b: t6 Z5 J
116. {
     , W9 R& h2 \7 X$ _
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     # Q3 B- Z: q3 ?  p9 L) F- H; f
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     7 i" K4 }) b6 e( H1 x# O
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ! u% T$ O  E$ y+ p! a
125. while(pos){
     , [6 ~# t6 c! x
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     . F; _% E+ j# g+ s: {8 |$ u4 Z
129. 
       p8 N! Q$ q& h$ c% @
130. m_Mappings.RemoveAll();
     
131. }
     . N4 j' p  l; Y7 c& c! J
132. 
     3 \" h/ \, y+ F$ ~. Q
133. 
     : p0 v# V5 E  b4 U) U) f4 t
134. bool   MyUPnP::InternalSearch(int   version)
     : \! R! o; }! ?( h+ a: F& |! H
135. {
     2 s) I+ u, b7 B
136. if(version <=0)version   =   1;
     % w0 k5 z7 Y' D" }) @
137. m_version   =   version;
     
138. 
     / }( M) K1 M0 j, K2 B+ ~4 i
139. #define   NUMBEROFDEVICES 2
     ( t1 ~/ ~5 u- p3 m
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     1 A" Q' q1 U, l: ]
142. {UPNPPORTMAP0,   _T( "service ")},
     1 ~1 D! x9 [$ R$ C/ n
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
       l* |2 R  ~; X" A
144. };
     3 V2 w) G& [( e5 q3 y2 l
145. 
     9 p2 z  b9 e) A9 ?
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
       y0 U& E6 E0 w: i- U
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     ( b6 F, a7 D$ a# a  R
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     7 T3 ?# v  ^2 Y1 b1 j
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ; c, A# v0 @( J4 p* K
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     6 F$ n! G  t" R% o6 f: b1 P( \
155. CString   request;
     ( F/ C5 T1 h/ ^3 r; T6 i
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 "),
     ( [3 `2 I! {" B. F+ r4 b
157. 6,   m_name);
     % a& _8 {1 o. _- [
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % {( y$ V' Q2 B5 c4 D5 u+ m
159. }
     ) Z, c0 ~3 P& m3 G% @4 z, v
160. }
     
161. 
     ) k/ A$ Y$ M; ]% Q
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     & K# K' l8 P$ P9 m8 o* O+ u3 O) H
166. if   (rlen   <=   0)   continue;
     " T9 B: e* v& q7 a/ ]
167. closesocket(s);
     
168. 
     $ f0 l6 ~3 T2 _
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ) U# C6 B" v& J
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)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     8 f% W$ U3 U3 ]6 D: q
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     . I7 U( s. y4 u
180. name.MakeUpper();
     8 R5 D  n# @2 D! L% t  o1 u$ D
181. if   (name   ==   _T( "LOCATION: "))   {
     . t, r# w. E/ z6 S2 i
182. line.Delete(0,   9);
     1 ^8 p" {/ J" w
183. m_description   =   line;
     8 w5 |, W$ c: F
184. m_description.Trim();
     ; d3 B* J1 M/ c
185. return   GetDescription();
     
186. }
     5 {4 V/ m& Q( ?) x( L9 h
187. }
     ) e0 Y; l4 J1 _8 ?5 s! v( K
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     ) S( R! L  v* E+ K
192. 
     
193. return   false;
     2 L* M" Z# v# u' [2 Q% ^- i
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
" Y- @) n' F$ X

( Y; N- \8 m1 r///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
% L( H# u. t9 F; i1 i#include <exception>
, j9 t& o$ ]8 ?& X
! z) x" F, N9 Z5 [; A; l9 X6 f" R
0 ^2 W$ V6 q" {1 R4 A  enum TRISTATE{
        TRIS_FALSE,
. h, C, A8 a/ f0 Q+ s        TRIS_UNKNOWN,
, V8 `. y5 d0 u. R, D+ {        TRIS_TRUE
};
9 e! x0 E. W+ u" L  z: S
2 r1 P2 H. y5 a: {* q  H* f
8 ]% a: B$ i8 [7 x% Uenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
4 j8 E9 @* l  C. k1 K: _* c6 X& j        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};




class CUPnPImpl
+ C2 s# |% n7 r7 O6 j. t{
public:
, {! _9 D2 r. W: f2 P        CUPnPImpl();
3 y9 c" A( z0 j4 j        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
- s5 ^: f* ~7 n2 o, D6 Y: D3 R                UPNP_OK,
1 P0 q) T! Y2 O* d9 M2 \* Y                UPNP_FAILED,
                UPNP_TIMEOUT
7 N! ^7 T9 ~) R: F& _5 N        };
0 @0 u& N" Y$ J- |

- [9 [+ c( X3 J( L$ D! M+ H        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
: O) c1 U) ^+ f8 ]        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
. c4 Q! ?% V7 Z* V$ |2 @! y/ M9 B        virtual int                GetImplementationID() = 0;
4 a9 u' y- B$ K3 @       
3 G. f" f, I' _        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
# H5 o1 l0 F/ m

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
7 d$ K  M4 [' L* v/ _        void                                SendResultMessage();
7 G: A/ H! X$ k        uint16                                m_nUDPPort;
1 o7 E+ Z/ a; Y0 f! Z, `* B' T) X        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
/ w$ D, C2 {! g
" k5 ~( y. d1 c' o6 G) e9 I; }
$ u& u7 `' a# j. b6 Y) c4 ?private:
& L! U! u! m' T1 r! d! r( v& a        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
4 K. O. k% [' f3 ]% c% O) V. Z
; d* h( f& a( y5 E$ L8 n  T
/ f$ x; H  ~7 N, ~6 h6 W};


// Dummy Implementation to be used when no other implementation is available
* o) m' o! }+ O: P+ Dclass CUPnPImplNone: public CUPnPImpl
- A) r# `( b+ H' y9 q; N3 n{
public:
$ O& K# E' [  ^) R; M2 a        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
. Z4 W+ b! {# U9 h7 L+ g5 s        virtual bool        CheckAndRefresh()                                                                                { return false; }
: ?4 r% [: A( {" U$ b6 L  t& V        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
' A& z) g; D/ V: u. B. R        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
9 D0 K/ z' @/ H4 X

/////////////////////////////////////
. h& B. R  T: H! q0 [//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
% K2 n: t' m  U- K: Z4 [7 }#pragma warning( disable: 4355 )

8 y, d6 c9 K( Y8 }/ W5 \
#include "UPnPImpl.h"
$ ^7 x* i6 U/ B+ ~#include <upnp.h>
. K) \% J. }; ?! y& \#include <iphlpapi.h>
#include <comdef.h>
0 q2 X5 A' \% z' j9 h2 R$ j#include <winsvc.h>
; U. t- }. J" h8 z! R) ^& F/ ~- l

+ c0 F# w% q! P2 p#include <vector>
#include <exception>
1 B; l1 ^: N* X& i% k; F#include <functional>
2 n: m& O9 y; T3 ^" R

/ h1 ?8 O, F& K! M1 S% K* m
  P$ b3 J" x  i
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;
typedef        _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;

& F/ U* x$ r$ z
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
5 y. {/ O9 ^! [1 m- O  PDWORD pdwBestIfIndex
7 [8 I7 U" ~1 O);
. `1 L+ S5 e. ^/ V8 v

typedef DWORD (WINAPI* TGetIpAddrTable) (
: q7 Q+ @$ F2 k8 |5 H9 l  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);
0 f: B& e/ m; \7 N  t2 g
8 S) A) n; C% T+ n% s: c5 ]! R
2 |' I9 W# U$ c. I; x9 z) ^typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
* _0 J1 Y: P( ~( n);
6 @7 S3 D) e' k& ]
( B8 A' ]1 i* t, k0 N
' u3 w: m# D( ?* R( y& zCString translateUPnPResult(HRESULT hr);
' P5 V* n8 \0 l; y! QHRESULT UPnPMessage(HRESULT hr);

! O- v) Z' R9 d1 U! u% d2 r& p
class CUPnPImplWinServ: public CUPnPImpl
' N9 m1 [' n, @) s/ I$ |{
# b0 |8 `8 d6 o" N! e9 [. ~2 _        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
' l8 t9 K) ]( G  j# X6 F        virtual ~CUPnPImplWinServ();
& Y: j9 `& G; X! D: J9 R        CUPnPImplWinServ();

6 m& ]  e- C4 Y  T
1 n+ X" f0 a2 A. |, M6 @        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
- Q+ y# H3 O# @+ V; Q0 b        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
" U! H3 [0 ~7 s/ z5 T, \        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
6 u" ^: [3 b# N9 b) ?" f        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
$ T4 z+ ?; `# J* [! c* Y' g

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
7 A& [0 _5 S6 a+ ]" a2 ]( b- T        void        RemoveDevice(CComBSTR bsUDN);
. {3 m; \: q/ e" F, Y# Q* U        bool        OnSearchComplete();
        void        Init();
5 {9 Q& d& \2 T" y4 M

# d5 V  w8 x! i        inline bool IsAsyncFindRunning()
2 T, S/ j) C( Z5 Q$ ?" Z( O        {
2 G! R2 }& T* r$ N# b$ x                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
6 [& Y5 G* m! m8 ?, u/ O; w( A. J8 x                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
- t: Y- L9 F7 `& j4 ^5 m0 r                        m_bAsyncFindRunning = false;
1 W! y, d' q% [' W, a, ^                }
                MSG msg;
+ {8 h) m& W; \# y                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" i& `/ _, Y2 t5 u+ O. v( c0 M) p                {
# n* Z4 w8 s; V' Z) N                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
: \' X( V5 Y  d9 _                return m_bAsyncFindRunning;
        }

- k3 d8 D+ ~! b6 }# {; w
        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
! h7 O) y+ r2 N6 L8 v. S        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
# @( k! E' ^5 a6 j( p6 K* b        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*);
$ y: y% I/ h, d7 F' Q        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
% E$ c, P1 ?, Q( o0 n9 ]
$ @, K. @2 q% S0 Y
        TGetBestInterface                m_pfGetBestInterface;
$ C0 A* K) t# P6 C5 K        TGetIpAddrTable                        m_pfGetIpAddrTable;
9 H( S3 _# Y( c; l$ T        TGetIfEntry                                m_pfGetIfEntry;

% v' s+ z, Z" D2 q0 L6 ]- a
' l# l( G( A9 N0 t6 r        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
: Y4 J8 b5 E3 c- s5 o$ e  T% z- C        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
8 b" }* S6 t' I) H3 j
: D/ S9 [$ p  v9 [
( L2 V" l% g9 k% ?1 p, v" R                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

- W  `) y; z: U+ i
1 F9 V8 G* H5 f* w& C. R4 m( c6 ^                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
* f$ O0 ~& l" h! U        };
       
/ p# t( b: J* z        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
, l4 S+ U8 G- g2 c, k: @; F+ b9 ^4 b* ]% a        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
1 f+ X1 |$ v9 y3 @2 k        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
) u, C- j, Y( P3 r0 U        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
% N# S( t6 N" W5 s1 T: q- E7 {5 d                LPCTSTR pszInArgString, CString& strResult);
6 T2 C% O. ~- D4 s+ l" R        void        StopUPnPService();
; K1 T! _/ C5 c1 T* s, K) {+ z6 }

        // Utility functions
3 v% L' g3 X3 {. r8 p; z* ^        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);
* X" z! n5 k. V) d' E9 j9 R        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);


// Private members
# y, _4 }+ q/ \+ H8 x1 S: }# dprivate:
6 }1 `0 Z; d$ N, \        DWORD        m_tLastEvent;        // When the last event was received?
$ ^+ e! h$ W1 x; B$ g        std::vector< DevicePointer >  m_pDevices;
" Z4 H- I5 Y8 e- Q' e2 V# u/ x        std::vector< ServicePointer > m_pServices;
9 m9 e/ o- g0 s, G# z+ \        FinderPointer                        m_pDeviceFinder;
9 b" w4 j: Y5 X0 d& C2 i8 q) H        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
# W8 m; w$ u( O/ F# O
: H* u: v4 C6 A0 `* p" |6 t! s
: ~- g, B1 }) {& E9 P2 F( f( C        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
; A0 O- q& B  c0 t5 E/ w        CString m_sExternalIP;
* V4 R1 p1 {8 E! g7 O( v        bool        m_bADSL;                // Is the device ADSL?
9 W, O6 R+ \% u        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
: ?% f/ N6 g6 n. U; \4 I+ p        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
( A3 N( k3 h* |6 _( S* o        HMODULE        m_hIPHLPAPI_DLL;
' F' N3 r2 I: Y2 n* t4 Q9 W5 |& o' c+ e, N        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
' }( s4 a& V" o# B& ]% o* ]3 f% G) s        bool        m_bDisableWANPPPSetup;

: s. B8 h0 j- X& M
};
/ M, t: `1 W' ?7 d" _. E( `

+ G9 I/ _. V5 b8 I$ A2 M# ^6 r// DeviceFinder Callback
, h2 Q; Y" X% R$ O  @; N1 O( cclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
0 v/ @5 C3 V2 I7 ]- d3 g( Y  X{
/ V& `  W& R+ k5 d9 U8 Xpublic:
" i# D7 |* P3 c7 A8 q        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 M2 s; w5 t& L9 ]8 Z

7 K5 r4 D, A( y; N. Z# v1 s5 Y// implementation
: }& L* S$ X* v$ S4 lprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
+ o2 }6 B  y  P2 E. n$ Y" b        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
1 o* \$ u. V# |& v
. U7 b! }% X' n
private:
; c4 d3 X2 g1 e2 V9 B        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


% ]- V- ^0 Q9 f; M; p// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
  J; W+ P# M$ @& |# Q{
0 J7 X) Y/ u& j- |% y1 zpublic:
! x+ e3 U/ w( @& P        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
8 r5 G, k) P4 m* T        { m_lRefCount = 0; }
9 G. L9 Q: A  T! v   
) j. N2 K; z  D   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
- w0 a( r* \: L$ ?" C  P. Q   STDMETHODIMP_(ULONG) AddRef();
: Q9 u8 ^" V) c" M& x6 f' ?& h, q0 b   STDMETHODIMP_(ULONG) Release();


3 C5 o; h2 w+ V. ?( f// implementation
0 s( q" `* n* X" k% k5 \. J( T+ |private:
, X3 i1 J4 c6 w- y) L1 _        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

  g6 B$ J2 \0 d) R1 T( A! M8 E
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 U/ I# t/ D) O};
) ]9 [- ?4 E3 l* z' m
4 l: @7 w- u$ k$ @0 `; V( M6 u( `
7 H5 @3 B! r, p* a9 S0 _/////////////////////////////////////////////////
3 N7 M  `3 t! D
% o% h! p' _; i* h4 _
使用时只需要使用抽象类的接口。
4 B; k: ~9 l2 e8 H7 k! BCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
" O$ |5 }; a" p! p. K9 YCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
1 c3 v& x3 Z1 ?$ E0 J* kCUPnPImpl::DeletePorts删除端口映射.