UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   + j: O6 O; f1 S/ Q6 ]
4. #pragma   once
   9 `5 N" K. V; T% p. H4 L) e
5. 
   ' \! o# `9 c% G% h6 ]3 Q
6. typedef   unsigned   long   ulong;
   0 }* M  U* H% b3 I  Q* N8 l6 ^
7. 
   5 Z( v; N9 K5 V6 a
8. class   MyUPnP
   # H! F4 E* `9 ]4 u
9. {
   8 \% S. f- I8 a0 D! H! z
10. public:
    - U4 e- |( |; Z( S9 i) j
11. typedef   enum{
    ' A' o% d) I: \, o9 T& s
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    0 @, J# h1 D' k# j7 g5 x
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
    & J8 v- q' E+ R7 Y
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    7 f6 c7 D3 P, B( v) G6 k
21. UNAT_UDP //   UDP   Protocol
    2 p" d! _. Q$ D" W: l# [
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ; x' t3 K9 R8 R; C  v$ z/ }
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ; w5 r* f4 G" ~) {4 j+ X
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
      k5 L; c* }& J' g% N( p" e
29. }   UPNPNAT_MAPPING;
    
30. 
    3 I4 v$ D! r( R
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 A3 d1 n; z6 ~5 L; E0 T$ y
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    / ?% u5 v6 q) Z1 e% P/ W* H
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    + ^- o4 P2 |6 L* Q
41. bool IsLANIP(WORD   nIP);
      d# j. r* i/ y
42. 
      y4 x8 b, @3 A  C5 q0 O/ Q* W/ p
43. protected:
    
44. void InitLocalIP();
    . m1 V+ l: W, h! T. h
45. void SetLastError(CString   error);
    4 O4 k" L, i* ]8 W) N7 A3 @
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    : j% A* \; s+ E
48.       const   CString&   descri,   const   CString&   type);
    ( D! p" E9 }2 ]6 [- a
49. bool   deletePortmap(int   eport,   const   CString&   type);
    # n9 T" k+ O) |+ t
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    ; P5 E; F( ]" y+ K, B7 Z) {  g
53. bool Search(int   version=1);
    6 _; N3 g8 B8 L' h/ J. ?
54. bool GetDescription();
    9 S' w$ u$ Z; ~; B
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    : V+ B$ O, C9 U# V# i6 {* i7 @
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    ' T- \+ Q0 y0 t5 G
60. CString m_devicename;
    0 n; ?& W+ H% ]8 m. [' j9 u8 X
61. CString m_name;
    0 S- U9 I$ _9 X& s" a
62. CString m_description;
    - W$ }: `  v3 n  _$ j
63. CString m_baseurl;
      K  c0 q* ]6 s+ ]
64. CString m_controlurl;
    
65. CString m_friendlyname;
    $ z2 N7 c) `9 ?. |1 `
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    ; G5 e6 C* A6 ^" p/ V! g" f
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    : F0 d9 n# F/ z4 k1 |% G
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   / D. o8 ~8 k% ?4 V. s: }2 r# x
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   9 \' ?" h' [/ @  d1 ]8 j
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
     _* @# }7 x+ n' R$ k+ [# g
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    " M% y9 U. D, i# w6 k9 H
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ! _! C% A2 r2 Z* [
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    8 \5 l8 D4 u' R5 N' G) o7 X
17. {
    
18. CString   s;
    ; }6 u1 }7 j3 L% b
19. 
    # ?  T+ h0 b. t6 e6 ^
20. s.Format(_T( "%d "),   i);
    1 S! q8 t" ?4 }* J! t
21. 
    + m0 o5 h, K  Y# {9 g
22. return   s;
    + e! J; v! f) r, t6 O) W+ F
23. }
    % J. r) _) R) L( A, ~
24. 
    ! A, N( }! j: i2 W& l
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    9 E& p0 R' u7 e! h- z7 `- h) ^
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    + L4 H& ]3 o! q
28. }
      k, s; Z; w$ R  j
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . N: y1 `' a# h/ X1 Z+ y
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    7 o" @: n  b: W9 w7 F
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    - F; j1 J4 I- M8 U: v
38. 
    
39. const   CStringA   sa(request);
    % v: b8 D7 L2 f: @
40. int   length   =   sa.GetLength();
    ' J) L6 T1 w. O, }6 R: s6 v5 H, ?
41. strcpy(buffer,   (const   char*)sa);
    " X5 v8 \+ O5 t5 g' u
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : A, w; q( Z0 ?8 ?7 F( s
46. sockaddr.sin_family   =   AF_INET;
    ! E5 F! j1 i: Y
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    0 [/ O# B5 l$ e3 @7 U
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    , o' j4 U: n0 l1 V
50. u_long   lv   =   1;
    , X' z$ Z3 ]. G* X" R
51. ioctlsocket(s,   FIONBIO,   &lv);
    , l5 _) s# @8 d* X! S% I
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    + f' H2 Z' l1 i" F8 j, a* }$ |1 y* L
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    , j4 L" O6 v. F( t+ U7 o
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - M7 D. x' d- E$ L- O; \
57. closesocket(s);
    + e4 U  Z- `) m- H) A( c: r% W% P
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    7 N1 G5 X) \. R- v
59. if   (!rlen)   return   false;
    - p# d# P. t: ?/ |* f- X) f% z
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    1 j* ]$ a/ Z8 T
62. 
    7 F" e* p# n( g- v8 L0 \
63. return   true;
    
64. }
    , @! Y. U' C" D0 R
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    / n: h. k* G/ T. n+ M
67. {
    # _; N: X+ h) S) ?2 C) z
68. char   buffer[10240];
    5 f' R0 n# k& i8 Q, |+ k
69. 
    9 o7 F+ U+ A6 r+ m8 ]
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    ( ]" l8 N+ p6 Q1 V% C
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    6 S, P) [: b2 G( `" ~: V
74. struct   sockaddr_in   sockaddr;
    % S+ ~0 [. A. \6 }/ V6 G  \  P
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    # K* ]6 F$ Y6 l
77. sockaddr.sin_port   =   htons(port);
    * j; o9 O- O6 v
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    * z/ F0 A; u( }, W
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    5 T: Y* f+ `, I: E9 @% L# P
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    2 M3 u7 f& [2 P* f  x8 q
85. int   pos   =   0;
    
86. 
      x& f8 O: D1 _: O
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    # e, m! p2 O1 W, o
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    % y  K1 p6 R7 {! }! }. y
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    7 V3 q& f0 m. r3 z1 k# N, p, E5 S
96. return   true;
    # }! Y' {+ ?- |' P# d
97. }
    4 ~7 m4 W2 J+ F' w( H
98. 
    ! u7 {: v) J+ N  f2 w0 t& |
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     1 Q1 {4 }! K2 V9 J
101. CString   startTag   =   ' < '   +   name   +   '> ';
     7 k' v8 t8 h* E
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ! ?8 [' k: q7 c6 V# h' Q2 z
103. CString   property;
     
104. 
     7 K- P+ K# {2 j1 z7 O
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
       t8 C. h. e* C$ y! j& Y! v: ]
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     * o7 p4 ~1 |7 G: Z
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     " j7 ?. n; N+ m  P% }* v  o
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     3 J0 U( h6 m0 H: X4 T
126. search   =   m_Mappings.GetNext(pos);
     ) ^! {! H. B7 X; W4 i
127. RemoveNATPortMapping(search,   false);
     
128. }
     " o; Z8 e- U" e1 k! Y& x
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     ; ~# v7 s3 l, o, `
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     # Q4 ~& {! h$ B3 N1 |  b
136. if(version <=0)version   =   1;
     ( P1 Y' `' p3 \7 q7 P' d
137. m_version   =   version;
     
138. 
     0 A8 p! E/ |) D, x+ z0 w! @
139. #define   NUMBEROFDEVICES 2
     : K; b& }. e! F, o. p6 `: q/ D4 z
140. CString   devices[][2]   =   {
     & u1 [2 O) E6 ~+ |, y7 N% c
141. {UPNPPORTMAP1,   _T( "service ")},
     3 ~! N- m; m" `/ J: v/ r" b6 q
142. {UPNPPORTMAP0,   _T( "service ")},
     : _6 |2 h' k) K" f2 q3 ^
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ( s* A/ @9 o" E1 F1 O+ p
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     0 W7 n) Z5 M% ^" `/ o
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     , ]& r. i* g* s
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     6 j# U/ B) i7 ]- e* j, I+ 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 "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     + Y* ~+ \, H  b
159. }
     + C* l8 `& E' C% ?: ~! E3 ?9 f" a
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     1 B, |* m9 }2 i' [+ u
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ) R0 O% y; T; }
166. if   (rlen   <=   0)   continue;
     . z8 ?( ]4 P5 o* T, {6 U2 U
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( |) O) ~7 M8 [1 c4 w, E% P; l
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     - d5 X+ W/ [. ]! F1 F4 F; B* W
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
       H7 Y2 o' N7 Y3 W0 t9 D; s
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     7 i; j3 _1 A* W1 ~. _
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     ; a8 F+ ?; g/ `6 r$ g$ r& A
179. CString   name   =   line.Mid(0,   9);
     ! N& R& l  S1 `# i! R
180. name.MakeUpper();
       K+ o- ]9 f$ ^! `! J, r
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     6 g- \/ e1 V% u) r: R
184. m_description.Trim();
     9 o  S9 q" e4 U
185. return   GetDescription();
     
186. }
     " ?4 Q0 x' t* u5 P5 U
187. }
     5 e% D! W8 i# g$ e
188. }
     # }1 o! g* ^+ ]" d5 g3 |; T
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     . C9 Z( j3 J, B% n
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
: s/ C* M7 W; X* E7 l/ G( j

+ ~% H. }3 \/ q  F* X///////////////////////////////////////////
+ y  C' ^, L- u# ], x% ]/ A  o//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

- X; L7 w9 B0 ?0 x1 S
#pragma once
/ q# g% r. B, N0 {; l# o* `+ i, b#include <exception>
+ H, C9 J/ t# d- h/ J: Q
, ?+ r8 \( T" s& W6 v( q3 j, [9 U
' ~& a. ]8 T" N' w. q8 y2 \: I  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
( a3 Z' A& w- C6 V* j' Q5 M9 W};


* S1 T& ~( B+ R$ renum UPNP_IMPLEMENTATION{
& A/ M3 N" m7 }8 R4 {        UPNP_IMPL_WINDOWSERVICE = 0,
* x1 z7 M* X( i8 z, {7 ]" k        UPNP_IMPL_MINIUPNPLIB,
& s; l) h! q! j! ^9 g/ U: B        UPNP_IMPL_NONE /*last*/
. K) N0 _' [8 N" _- p- u5 ]};




class CUPnPImpl
/ T* J+ N4 }) D# U  V' s4 {{
public:
, m* z. [# j& a2 S# E$ H        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
' t5 y1 V$ s+ R6 ^2 S) r7 \6 v                UPNP_FAILED,
                UPNP_TIMEOUT
2 c7 T0 d: w, J        };

( S# h8 M2 v& p9 N
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
1 Z" V. z$ z& i' X, e: k$ H        virtual bool        CheckAndRefresh() = 0;
& A& Z+ t, D1 l& D9 C        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
8 @$ r+ C* n! U4 ], X+ O9 {0 n) o' x, r        virtual bool        IsReady() = 0;
$ a" ?! v$ K+ O9 r8 Z- G  W        virtual int                GetImplementationID() = 0;
+ B6 b+ Q; S  O- \  ~       
% i* @; v8 [% G6 R1 S        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

: S% C6 d; [% {' ]
: G7 ^. V# ^0 H4 c5 u" Q5 a2 J        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
, a4 c) f: Y9 c* j- F) E        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
6 m2 d9 z, L5 b8 E/ A) A        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

! L: _- f3 x1 m( i1 E2 Y  l) y# X
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
/ z9 Q) K6 V7 Z, n/ q        void                                SendResultMessage();
) C+ ~0 K# g1 r$ U! {& T        uint16                                m_nUDPPort;
7 N% ~( a  c( u$ S        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
) K6 r6 e' O- M' Q8 U        bool                                m_bCheckAndRefresh;

2 K2 q9 m4 @8 u4 [0 W4 T9 n
2 v% l6 U" J+ F9 o- l6 Sprivate:
* n! s7 p/ Q7 ~7 i8 h! k        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
: _% Z. t  k8 ?3 c) B
! w) e) n; G' R, l3 A2 Z
};
2 N, {4 J) w  v% c: J* l
6 N- x) r1 c- a3 ~5 }: @/ \3 K; e
// Dummy Implementation to be used when no other implementation is available
) _# g5 v3 b3 l% F& J+ A) Dclass CUPnPImplNone: public CUPnPImpl
: T$ O% J7 H; d4 x1 O% L{
# M, F3 ?* @# B7 E) Ppublic:
# R+ V1 [0 G- `' Y' S+ y        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
! x6 W; D" `1 V        virtual bool        CheckAndRefresh()                                                                                { return false; }
2 d8 l9 D, t6 G& k) N( ^2 s! n* n        virtual void        StopAsyncFind()                                                                                        { }
; v# n* q! q* K3 e8 Y8 S        virtual void        DeletePorts()                                                                                        { }
  f& b5 ^' _) N3 ?7 s        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
$ G% \0 T' O2 M. X//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
' _! L$ {4 d; A' P* e! g$ R( _#pragma warning( disable: 4355 )

) I7 g. ]* I1 A0 ?7 w
#include "UPnPImpl.h"
  I! P: T) E2 _# O# |9 `$ K# W#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
6 j- c4 L- S8 ~$ }0 l7 [# _#include <winsvc.h>
& n9 m* B& z. ?" n* I5 h9 }
# C" \, P, [1 M( F
2 z; z" x" W, v#include <vector>
#include <exception>
" {' {' R+ r& Q! S* b8 H#include <functional>
1 c+ x4 u+ |# J# [8 X. y. w
2 }  x' s7 s& q6 K


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
+ f7 r/ V) C0 i7 A' ctypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
- K( P" C, U1 N! x; u6 ~typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
' a- `* n0 [, f& w: n5 |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;
  r! Z6 b, e2 q( U* i5 ctypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

) _) l% \* W4 Q' u/ P  v" o! Q- J
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
. H2 k9 Y# V; q  PDWORD pdwBestIfIndex
- W( R  N  c( y! [0 `2 O);


, N% f+ S9 d8 Y4 T/ b+ c, _typedef DWORD (WINAPI* TGetIpAddrTable) (
, o+ F3 z' Q  L$ C0 A1 p, v  PMIB_IPADDRTABLE pIpAddrTable,
! G6 L4 z6 Z! }  PULONG pdwSize,
  BOOL bOrder
);
- t' \2 T  i1 l0 O

typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


  ]  l3 C! U$ D# p: N8 lCString translateUPnPResult(HRESULT hr);
6 r$ Y% u/ A' V( @HRESULT UPnPMessage(HRESULT hr);

  I3 k8 i  }0 L( T. p
  ~* y' C( P' P$ nclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
9 A1 J8 T# c- B) F, m0 O// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
! s( x5 Z1 D" i

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
% M" C2 D# y) E" Q) A5 V        virtual void        StopAsyncFind();
9 J" b" j; X, j) q- `, r        virtual void        DeletePorts();
/ @9 \  _, g  j        virtual bool        IsReady();
4 G( @. A5 a. y- ]) X4 _        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

) Q" K- `3 f" F: J
2 {! L, C8 ^* T* q: O# A        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
7 n, N- S3 Z" h; t, c' w( c) b4 p        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 a6 v* W) H3 A- b

. u9 q# G8 ]; A6 r9 [, Sprotected:
" h: L% K3 E, H3 K: u$ l        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
8 {; V2 [- g+ a* p6 Y0 s! h  r        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
! k! t) _/ @3 ^: z        void        RemoveDevice(CComBSTR bsUDN);
7 _" q: T* _1 t# ^# j2 o        bool        OnSearchComplete();
        void        Init();
3 }& ~$ H2 \) L) K& Q' L8 y3 ^

: f; \  H) G9 s+ E  H+ ]: g        inline bool IsAsyncFindRunning()
        {
& H$ n/ q% l6 w# p+ Y6 x7 y                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
* s0 a# ^9 L# r% ^7 c6 S                {
8 |6 j# n9 q( a4 h) e# N5 o                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
2 T' c! r; Q2 i2 R( N/ I                MSG msg;
# e, Z8 F6 u9 v+ F                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
6 y/ ?6 B9 A, }$ R& o8 H# v                {
                        TranslateMessage( &msg );
- `- g" x4 J& d" N- n+ p                        DispatchMessage( &msg );
                }
9 t0 M$ g. ~; K  d0 e                return m_bAsyncFindRunning;
        }


        TRISTATE                        m_bUPnPDeviceConnected;
# F6 ]3 v( X) y, S
; o: ~9 H) u4 M- i0 w8 R
: K+ T- _1 {( }% A// Implementation
1 M8 h- {# D, R3 L        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
9 G$ `& b% \9 z( Y, j" M% E        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
7 n- O0 T- ~  I4 B# [6 Z6 ?, P        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
! ~% M+ e3 }- E4 E$ n5 z. z        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
$ Z2 g) e- j7 y( p" L1 Y: y

+ y( n+ q2 o/ }        TGetBestInterface                m_pfGetBestInterface;
  q3 O7 V1 s& A* E2 L        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
% h! V* ?3 |. C2 e$ K& y! V
+ E# D: T" Q" F( ?, o0 v/ o$ n
        static FinderPointer CreateFinderInstance();
! s9 a. p7 J+ y) v        struct FindDevice : std::unary_function< DevicePointer, bool >
5 I- f5 _( }6 _' S+ v7 q5 @        {
. h. W( ^( R0 e) o9 a3 i# `                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
% u, ], f; p& c* h, D( ?                {
- w2 G' _" v/ O6 E4 E" ?                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

/ O( `& D, b0 ?* D/ \. ]- {
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
+ a- q# H! c/ l, G8 R                }
                CComBSTR m_udn;
- _! o* d9 v- ?9 |        };
       
; u# `. @( Y. H  N7 s. W8 J$ m        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
1 J) u9 J! |0 N8 X9 L        HRESULT        MapPort(const ServicePointer& service);
6 X9 r( l0 I. M6 i  x; Y        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
1 r* B$ H1 W4 Y# b3 H4 `        void        StopUPnPService();

. B6 [. {  T( F) n
4 p: M: z5 d) U' s# P+ s$ ~        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
+ f1 C/ m5 ?# r        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
( ?! M- U* s# r) z        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
! G3 a' K9 _+ P3 r: Q; {7 X+ K$ ~& k/ |        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
; s2 y% N+ E% |8 _9 @4 P: n( B        CString        GetLocalRoutableIP(ServicePointer pService);


" g: r2 J5 j2 F5 X& q// Private members
8 @- s* u8 R$ T6 Y" w& ~; Oprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
8 {2 k3 N# a2 a  U; S        std::vector< DevicePointer >  m_pDevices;
. d( h, f2 P6 Y' @+ v' p        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
3 p3 {2 v" k; I- M6 H9 y" d        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
: q4 T3 f8 G* p; i1 N- x0 P

. g, p) _/ o; k: o        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
: t/ I/ a* b+ T. _        CString m_sLocalIP;
/ N/ E" M/ T3 y) V" A2 R, R        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
% C8 |, j  N, F5 x" l        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
$ O+ E. Z. ^+ [& Y1 O7 Z8 P        bool        m_bInited;
        bool        m_bAsyncFindRunning;
# p5 N! g& b. y        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
6 s) |- Y- e5 S& F        bool        m_bSecondTry;
& v; f- `' r1 C) }& a        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
, h( w: P  x$ D  n- O( j
3 S6 f# o# s6 o! q2 h! y, R
* [- T! z9 i  [! \4 K' Q8 E! B};

  Z  X! |7 N/ V* Z) u! }1 {' k  F& W
// DeviceFinder Callback
7 k3 s1 Y' D  K6 ^) qclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
$ {# N' l; L+ e$ [7 G+ K{
public:
: [7 M* D5 M' |' r        CDeviceFinderCallback(CUPnPImplWinServ& instance)
4 s' N; ~1 J$ s% c, L: [4 X# d2 _% k$ r                : m_instance( instance )
+ E3 I9 ^  P$ O/ U( f3 v4 [        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 D. u! F/ h9 w) y4 v' s, o; N7 U   STDMETHODIMP_(ULONG) AddRef();
7 g. [/ O# b  w- ^4 D   STDMETHODIMP_(ULONG) Release();
) ~/ \% N" x# N* ~  s" Q- y4 k

! s* H' W4 P- H- q// implementation
private:
) ?6 H" L3 m; G        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
% b6 b/ z' U. [( [
9 D% i* I) Q5 @3 x1 Y6 J
( F. k2 y; t2 L% L; E( y' d5 Iprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
- A5 y" l. N' \( k! ]& l4 B

// Service Callback
) E& }( J- J/ F) jclass CServiceCallback
        : public IUPnPServiceCallback
( d9 C- _) Q8 A. a- z, Z{
public:
% u2 h) E9 ]: G8 F! i/ ^2 p        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
" X: A( {& V* F) t2 A1 R        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
  v  }( a1 n  p9 A0 J
1 [+ L/ H  b* E' y
// implementation
private:
: o$ B: d! _1 g; w. n        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


! w# \( w7 J& m; b% M9 k2 Eprivate:
* Q0 T3 |  Q8 ~- F) q; K9 {) a6 W        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


( c1 \/ A( ^1 A" _  s" ^3 j  S/////////////////////////////////////////////////
0 K9 }/ D- p9 g9 }
* Z/ _, ^) m) x3 z& V* Y
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
. b, ]& \/ ]# q; k( Q5 k: v7 c% MCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
$ q8 ~- `. o4 d  U0 U; ~CUPnPImpl::StopAsyncFind停止设备查找.
. h4 |3 U: |4 vCUPnPImpl::DeletePorts删除端口映射.