UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   , G9 U' `; ~/ x+ N5 y
4. #pragma   once
   
5. 
   1 J; y; R# p# a, r
6. typedef   unsigned   long   ulong;
   / _% o6 l: N. }- X6 }) ~
7. 
   ; l/ s( Y/ e" q9 y
8. class   MyUPnP
   # j( o8 @! h2 I" H; V
9. {
   & M7 O4 c/ `' R7 r5 ?& {* w
10. public:
    
11. typedef   enum{
    2 J0 B% N- B/ g3 Y9 Q7 `
12. UNAT_OK, //   Successfull
    / x5 U5 O9 i. a3 R" Q
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( j4 @2 j2 q2 g' U2 N+ S% O, z, \
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    $ K5 A8 e9 [5 R+ {$ K) d6 Q
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. 
    8 S1 y- {5 e" C. k" _6 S" V- ?
19. typedef   enum{
    , U+ b" Z6 v9 P6 D: T5 y: L
20. UNAT_TCP, //   TCP   Protocol
    ' e4 n0 A3 v) S
21. UNAT_UDP //   UDP   Protocol
    * `* g3 p& A- n
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    ) k( p  H, s! p
25. WORD   internalPort; //   Port   mapping   internal   port
    # S" S% U5 T1 i  K
26. WORD   externalPort; //   Port   mapping   external   port
    * S% X& D& k0 o+ b
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    $ a; {# _$ t3 c, q, e
29. }   UPNPNAT_MAPPING;
    3 X& p/ l0 `! M* l- A& g2 x* ~' r
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    " s+ c5 t% x& O* a, g  X0 q" e, ~* Q
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    2 N3 g3 H( b& L4 B, X! ?
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    7 _! ^" N' P- `% B1 L, V& d" u
39. CString GetLocalIPStr();
    0 w" a) w- ]- l0 B  E  |: B, o
40. WORD GetLocalIP();
    : Q" a, w1 W, @! R
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    $ L7 M- b: |0 J* R  h
44. void InitLocalIP();
    0 A- I' S( K' w- `6 m* Z
45. void SetLastError(CString   error);
    
46. 
    2 ~& e. i9 \% u0 ?
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
      {+ F; g8 Q2 B. N1 d
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    + U! C% G9 q5 A
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    : D; }0 \+ J) R. L
52. 
    
53. bool Search(int   version=1);
    " z  t: H) M* ?9 ~
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    1 J! n' x* b0 r" Z1 L8 I
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    ; L; r0 h, p$ R" Z0 X/ y( K
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    ( ?  S" T, M7 |  Y; T4 R/ a
61. CString m_name;
      V5 c5 N, @, u" c" n
62. CString m_description;
    ' }$ C# g- l/ G& o# t
63. CString m_baseurl;
    2 m; h  I" a4 P& _0 H7 n
64. CString m_controlurl;
    ! `+ C6 t& N1 y8 K# s5 D! P: _
65. CString m_friendlyname;
      D4 ^/ }, n' N" H1 y3 V* r
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    ! j) n' Y, d* q8 L) ?" t6 o" ~
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    / A- \% R5 z8 [4 t; e0 j+ {
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    # u, U9 ~+ o- v; c# }- H
74. WORD m_uLocalIP;
    
75. 
      h/ }! k: a' n$ Y
76. bool isSearched;
    ; T( [( u2 r% k' I7 U) L# ~) {
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   . q6 `. k6 c$ n! K. j
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   5 G" _3 e+ [# I
5. 
   6 q4 g# S6 Q2 {8 f) V% p7 ~
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   1 C0 _1 [+ h% i2 A6 o
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    . P% z+ n$ p: w0 f+ e: X- D9 m
11. 
    ' \) X- u# c* W! Y) y( y
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    5 U4 s& G( |" G6 A8 m
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    $ k1 h9 F; t5 x6 z# S  |; b
16. const   CString   getString(int   i)
    ' |" [, Z% R9 n8 D( G5 `
17. {
    
18. CString   s;
    8 h, n( @' I7 i" R+ V. z
19. 
    ( A0 |2 b& ]7 U* s+ I. N
20. s.Format(_T( "%d "),   i);
    , j6 U0 u; o1 v! X! W  H
21. 
    # i* L# ^# g) x9 A* C: D/ c/ T
22. return   s;
    
23. }
    # ]$ _) O. M; Y, Y
24. 
    1 ]' I1 T: h5 J  I, K, T
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    . i# N' i; Z9 u! S# e7 ^7 Y
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ; H" ?1 Z) w: g$ g4 H
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    2 `. q8 n" ?- ^0 m! y- p% T
37. char   buffer[10240];
    
38. 
    " ]8 l! {( y4 O
39. const   CStringA   sa(request);
    ! Y( e7 Z. X) X% d6 b; b
40. int   length   =   sa.GetLength();
    9 A" Q1 L# l1 @* w5 f' ]
41. strcpy(buffer,   (const   char*)sa);
    * t. M+ e  |; z/ @5 Y( {4 ?# }
42. 
    ' y# q* J5 {1 V5 [1 Z2 t& r0 H
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    7 M1 }3 |0 k, r6 U
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    ' g: ]9 _$ A. S( W7 S5 \# F
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ( o. S  B$ ]( p! V
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    0 N1 J$ ^4 x/ g$ H
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # \* q8 B( m1 P0 \$ ]" E8 @3 f
53. Sleep(20);
    : I  m8 D1 r+ D7 \8 d, z
54. int   n   =   send(s,   buffer,   length,   0);
    - F3 c! Y6 n7 U% Q5 h
55. Sleep(100);
    4 B* J; i, g& E6 g6 n* E1 w: [" ^
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
      m: o2 X0 G5 T' e2 a% X$ ~# l2 M
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    / t% w$ v  M& Z1 `. o
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ! B( h- X; y/ R* ]
63. return   true;
    
64. }
    
65. 
    0 E7 v1 o& V. z5 H
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    - S9 E+ g# U! c$ E+ L
67. {
    : z% @, `$ G3 d+ _) a
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    3 U; B7 ^4 j/ @' K" I
72. strcpy(buffer,   (const   char*)sa);
    & y4 m0 H7 ~' L' D% v
73. 
      u. Q; C4 t0 w. h# n
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    - D/ m9 R4 K) W4 E6 Y6 Y8 E6 ?$ v
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    & n3 n( C3 o( A: T9 F! Z0 _$ b- X
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    ) @/ L+ m8 I9 W9 t" P8 P6 |- ?
82. 
    / E, n3 j# d% W8 B+ o# J
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    3 _5 m8 V0 n/ B, _& R- E6 L
84. {
    5 [" v8 V8 t6 J5 j
85. int   pos   =   0;
    
86. 
    . w8 r. [; h. ^  K3 j$ Z* z3 ?
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    + Q4 {4 P8 _, ~# I3 @- i# a  y, e) ?
89. result   =   response;
    
90. result.Delete(0,   pos);
    ! Z0 a/ H4 E: `$ k
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    & }8 G$ l% c. w) W2 K
94. status   =   status.Tokenize(_T( "   "),   pos);
    ! {+ M) Z. L3 E5 L# g
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ! Y; Z7 T" K6 R; W! ?) ^
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     : X8 J2 k9 o. Q3 G% a# s6 c
108. int   posEnd   =   all.Find(endTag,   posStart);
     - V8 J3 H% P; A; k: g4 f; R
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     6 M8 C+ K3 j0 O9 d& o0 w
112. }
     ( e/ ~6 Y; V: b( s8 F4 X: h
113. 
     
114. MyUPnP::MyUPnP()
     5 T5 q2 n/ ?7 ~
115. :   m_version(1)
     . T& D7 \" z/ C4 l) ~
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     / [. v! R; p( X* r. P% I, D
121. MyUPnP::~MyUPnP()
     
122. {
     / _) C% y/ o9 F; r1 a
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ; |5 {. G- O+ ]1 k
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     * g6 }, @7 s9 f8 Q
129. 
     5 v9 ^6 @$ f9 S8 }& Q7 [3 z
130. m_Mappings.RemoveAll();
     3 ?7 r% y2 w1 y7 T( b" R
131. }
     , F. i$ _( O) o2 M, E
132. 
     & M  {/ W. ?1 `
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     7 U, o' A" t( g! I. g8 o
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     8 U' o8 W) @5 C/ ^0 A& A% e9 c
141. {UPNPPORTMAP1,   _T( "service ")},
     ' x/ k- ^" `. v* ?
142. {UPNPPORTMAP0,   _T( "service ")},
     * V( k6 S: I8 N5 F1 O: e
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     1 N! b. b5 U6 E: K9 A( E+ `$ ^& P
145. 
     2 U  \) w" L, p9 `
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ) m& C9 V, S6 x  ?1 o
148. ioctlsocket(s,   FIONBIO,   &lv);
     6 [+ [: s/ \! Y1 D! Q! j0 |8 |
149. 
     
150. int   rlen   =   0;
     - a% N4 s3 q8 H! f
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     5 p  ]; F5 Y6 M: F6 Y+ c& B/ F
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ' @9 y  E" z; \1 m3 }
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     + `& `1 A5 ]1 [  \
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 "),
     9 i) X9 }; k/ |1 l  M4 ]) y
157. 6,   m_name);
     * r% s$ d  F5 S9 Q' ]1 J  X  Y
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     0 ~( j' x7 |6 q: O/ f5 V6 B+ l* C
160. }
     
161. 
     / G+ {4 D! N' U0 v2 {6 T# C
162. Sleep(10);
     
163. 
     % w1 B% P/ V7 q, r" j5 o  f
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
       K9 b$ L9 o5 n3 ^7 R
166. if   (rlen   <=   0)   continue;
     0 q: g, V3 U; M* D5 @( N2 ]; D1 J
167. closesocket(s);
     . d5 d' A0 m$ U
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     8 n$ q% _, |1 f5 R. O9 w0 h
171. if   (!parseHTTPResponse(response,   result))   return   false;
     5 c' ~  ?( z2 U: X" x: @$ H
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)   {
     ) g' _& e6 h; R4 `4 k0 `
176. for   (int   pos   =   0;;)   {
     , K! z5 Q' ~5 ]" g6 V: n
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     : h2 I) J0 x# f$ L0 Q6 \/ }& w
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ) |- e+ C3 t1 ^: Y
180. name.MakeUpper();
     6 x! h$ c$ R2 @0 E9 C* R
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     " k2 G# \( D$ r( c7 a4 P* o8 a
183. m_description   =   line;
     % X( S( \8 K) A% ^, g
184. m_description.Trim();
     
185. return   GetDescription();
     ( [& F4 K: c* @" x* ?/ y/ l; X3 k7 M
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     * g0 P/ t8 b1 O' I
191. closesocket(s);
     
192. 
       r9 F1 g$ g. e  o0 K
193. return   false;
     
194. }
     # U) Z4 Y+ {: |4 g

复制代码

天云上人

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


///////////////////////////////////////////
; ?2 b1 p1 z* ]2 E//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
- [+ `9 A4 S4 k0 p& _
/ B8 F) A4 R9 s0 a3 k  r; E
; `  [. w; t- C( ?#pragma once
#include <exception>
% I+ {$ ^! A: c: q

2 g9 |' y# G) s$ @" `3 d& |  enum TRISTATE{
4 ^/ }' e; s' e1 C% {8 R        TRIS_FALSE,
' y% o: W6 T4 ?  L1 V        TRIS_UNKNOWN,
/ {" x$ G% x# P+ Y( X6 b- T( Q        TRIS_TRUE
* C8 @$ X4 W2 z; d7 w};
% ?/ H5 Z0 s: D6 t) O) Z3 p
; x$ E. {  k" b
5 D/ g5 j  B% e& w& Wenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
; [' P- c6 m/ c& J2 O" |};
/ `* T# p6 Y! X
$ d  ~% k6 ^) `# X: ]9 j
8 @  p- ^6 A8 f0 M' D; s
  d, n9 b( P6 B! n! Z$ j9 e9 H0 C, u
& R1 ]8 @. F9 ^  w8 vclass CUPnPImpl
- a, |, \; [! _, C9 N- s4 ~; ~{
public:
( j/ s' B9 Z" L3 U- p        CUPnPImpl();
9 J5 S/ a5 a$ |) d# N% [" N        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
3 y0 z$ L/ c3 T4 V                UPNP_TIMEOUT
        };
3 G* z. r* `: x' O! L

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
9 J' l# t9 D7 g8 N. B9 ~        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
. q7 f! `* G6 F# z9 O3 v        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
- m4 u3 O' Q, g4 ?2 F. u       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
  R( A! U  D* a; Z3 f        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
6 D  @2 ^" G' n) R: ^! y5 p

// Implementation
' T6 ^; p# R1 t0 s# s: @protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
/ x$ L9 O) x# `; {* [! v# l4 x( K! y        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
6 y. s+ U/ M2 }5 ~4 l        bool                                m_bCheckAndRefresh;


' G6 S* b* v1 D" k! mprivate:
, o" a8 Y* E6 I- n$ L8 w, j        HWND        m_hResultMessageWindow;
$ D$ e  l. Y5 ~4 [3 @9 C0 h        UINT        m_nResultMessageID;


};
# T& s; T$ i6 G/ l3 R' P; p8 t
3 e" n3 i6 b, T: F3 Z. D
// Dummy Implementation to be used when no other implementation is available
# i  b5 _4 Z9 p0 aclass CUPnPImplNone: public CUPnPImpl
{
public:
$ T7 C5 [% B- r! u' I) r        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
# Z# m: ]" \  d0 s( s$ o! ~        virtual bool        CheckAndRefresh()                                                                                { return false; }
' D, g- i8 Y# W# @7 r        virtual void        StopAsyncFind()                                                                                        { }
* ?2 @( d2 C: ]$ {6 _        virtual void        DeletePorts()                                                                                        { }
  X! L  I2 U% |+ D, v        virtual bool        IsReady()                                                                                                { return false; }
! `1 t! b2 T" V9 b3 [9 V        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
  a3 o3 y0 g& T  c$ P+ E" [8 J
& Z$ w9 v: v& Y, R  ~
/////////////////////////////////////
' p( h. B8 {+ X4 s3 V  [//下面是使用windows操作系统自带的UPNP功能的子类


8 Q" H  ^1 q8 |! n  D7 m! `$ `5 {( v#pragma once
+ ^8 o+ s; v' h5 v#pragma warning( disable: 4355 )
; Y$ L- y1 F8 @" q) Q" \
6 w5 w# T% Z, l+ Q* _
  B3 \! B/ n+ K% [( t#include "UPnPImpl.h"
8 u! E/ G  j' f, ?#include <upnp.h>
#include <iphlpapi.h>
, ^* G' X- P* \* @! Y+ }#include <comdef.h>
8 A, ^& J, a: J( ^#include <winsvc.h>
: d9 Y' |  Y$ P$ Z# i. Z

#include <vector>
7 s& W4 Q# l! t4 T* |#include <exception>
% C4 {" X0 R; k( H* G3 n#include <functional>

) O  Z  O: p- e/ |

5 W/ ]% W" l6 ^
* s3 f. f5 ~7 jtypedef _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;


2 w& w: E% y& w: Utypedef DWORD (WINAPI* TGetBestInterface) (
) T8 R) p1 Z, u$ G  IPAddr dwDestAddr,
, i0 i  C5 N* z, _8 e$ J6 J  PDWORD pdwBestIfIndex
: o" v9 _0 T+ |4 _1 |, n);


" a' S, o/ E: Q5 Btypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
0 i5 |# a. t1 _- A$ d5 U! X; U  BOOL bOrder
);

  ]! s+ e9 \) a* S
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
' v  ]. ?5 b- T);

6 S4 `3 W) J+ l+ R( A1 V1 a5 `
! S5 N/ h" [" e9 z  zCString translateUPnPResult(HRESULT hr);
; g, ]. g. @# k1 U9 J6 xHRESULT UPnPMessage(HRESULT hr);
$ [4 _/ G5 o5 s9 l

class CUPnPImplWinServ: public CUPnPImpl
{
5 G3 U9 ^6 {, A% U! a        friend class CDeviceFinderCallback;
: a* g$ O6 K: i2 n8 c8 J        friend class CServiceCallback;
// Construction
; }. q5 r- J* N% f) t3 hpublic:
: W0 w9 d1 S3 [" u. M6 _5 f4 l        virtual ~CUPnPImplWinServ();
! J1 |, H0 @+ g' k! D        CUPnPImplWinServ();
- }% ^# J3 F1 ^2 ?$ v4 a1 Q

2 f3 z/ X' b# Y        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
7 Y0 G' [% p: u) i  \        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
$ D2 U) f/ s" c6 h1 `

& i' ~2 z+ r5 B$ `4 W$ ~        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
* ?7 f6 _5 M- h3 l2 ]        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
8 A" X0 m! [* J. }7 x        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
( f1 b$ L$ L$ g5 v2 B        void        Init();
6 D3 o0 m; u7 K% Y" b
3 s; h0 d1 K% Z) e: R2 `+ n
        inline bool IsAsyncFindRunning()
# \; ^) d1 ?  U2 Q( {: l. S, t        {
8 y0 E& b( h5 e- ]                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
. G$ z* L8 ]" k  S1 U                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
  s& C) F; f' c5 [                        m_bAsyncFindRunning = false;
                }
; [) `( i% Z. t# b& s/ M                MSG msg;
! E5 J" A2 y2 ]; s+ O$ r' @                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
- G, f. {# x5 ]( g+ z# J& k                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
0 u1 i% b  s: u( s; r# ^

0 {/ C% ^, Q6 D! l6 x        TRISTATE                        m_bUPnPDeviceConnected;


- ^; u% r3 t( A, w// Implementation
        // API functions
5 V8 d! \8 u' X  @9 J, X        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
0 J8 m/ R% W# o8 f0 r        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
  N# j+ K! p8 }, P! }: O        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
  U& f4 f" o+ ~
) w2 U+ t6 K+ U4 A
- S5 M  \* b, @7 |1 ]+ M9 F        TGetBestInterface                m_pfGetBestInterface;
' w7 ]$ X# {6 ?% `3 B1 }% q4 j7 K# K        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
+ v: W' o* y$ j
- G/ P, l2 }) @! O5 C2 x
- x. S8 T3 E. T5 C        static FinderPointer CreateFinderInstance();
! t  M0 S2 Z3 a1 L; P% p* r        struct FindDevice : std::unary_function< DevicePointer, bool >
" ~$ e; X% g7 a4 y        {
5 l# d4 U) y' M3 K0 L                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
  U0 s1 C+ m! E3 V5 n: T) m                        CComBSTR deviceName;
; x+ y# v; c9 Z" {5 j                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
  @" V4 P' e- I8 Y

                        if ( FAILED( hr ) )
% X6 n+ a/ {5 D+ a% V. W. D                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
3 V) M. x% d! O: k: ]+ W1 w        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
0 g- a: c2 l% d1 M/ V1 D        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
+ g7 S) f8 d1 {. f        HRESULT        MapPort(const ServicePointer& service);
9 n9 q- z( a+ W1 ]( q+ i0 r1 Z! n        void        DeleteExistingPortMappings(ServicePointer pService);
7 k6 Y) W9 w6 y! D: [        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
7 P& N4 d* n3 v3 Y- E        void        StopUPnPService();
( x: c' z) Y& z5 D. T2 B6 `4 w/ F
5 M! t: K5 D) l4 S. O0 m& n
        // Utility functions
& O' G. ^& \4 H6 ~+ M        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
" H2 o; b% A7 f: ~        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
* _9 A1 U) K) v        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
5 `" s" K- A6 ~        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 K# y5 D, y# Q        CString        GetLocalRoutableIP(ServicePointer pService);


  e0 y4 n8 p0 B0 }# D7 b// Private members
private:
% l+ v' z. b& d/ m0 k        DWORD        m_tLastEvent;        // When the last event was received?
6 n  d( Q) [, W5 x        std::vector< DevicePointer >  m_pDevices;
1 b' e! P) W' E8 f+ A5 |        std::vector< ServicePointer > m_pServices;
8 _# m# Z' }) D- j( m9 j        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
% Z. [' `" x6 O8 Z* c        ServiceCallback                        m_pServiceCallback;

* n" Y( s8 ?4 j; I) {3 F
9 \# z) `- l- U  @. p        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
" b6 I7 G3 E( P        bool        m_bADSL;                // Is the device ADSL?
3 V% O/ ]# i- t* t5 u: c1 h        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
$ e* N) i! _% |+ e7 B5 }. m        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
# R# V- K" ], |  K" a# A        bool        m_bDisableWANPPPSetup;
2 I4 f% Z- Q# R# y* f6 ~
+ Z6 v; x1 T4 ^" G
7 ]% {( G1 W$ L' |7 I' ^4 h};


- f/ x# j, F4 g+ ]// DeviceFinder Callback
) E2 ?; K7 M$ r& E% T9 |class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
+ L7 @: v4 R+ n) }{
2 [. [, Q+ Y3 E5 q9 E; Jpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) ^& s  X2 a: L1 q" N                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


1 d+ \5 }. U* `// implementation
private:
! o. \8 a' k4 }( |' O& F        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
+ ^; u" ~" L! p5 E: ~8 j/ j

private:
        CUPnPImplWinServ& m_instance;
# _1 y& ^4 u) k" q        LONG m_lRefCount;
  d, H! _9 J& Q};


// Service Callback
' R2 N* B/ ?2 T- Nclass CServiceCallback
+ v0 x7 E% t! W, Y* K1 ^        : public IUPnPServiceCallback
8 `/ h* r- O+ G{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
* I6 @, U2 ?6 ]1 \0 L   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
: h6 P$ h1 e# C% J$ l   STDMETHODIMP_(ULONG) Release();
6 y6 r# P$ h9 [. @  Y
1 [5 U* W! R$ Y, A1 E" t; p- g
% r  `' P/ k& V* ?// implementation
- l- H# l9 a, s5 gprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

" z3 w/ h9 l$ d  _  i
private:
- q$ H  @; h: c/ G        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
' J3 b1 I( G6 G$ F% D8 {5 o};


, L1 a2 e1 L! a. ^/////////////////////////////////////////////////


* z5 Q1 _$ J1 D2 E5 y7 U使用时只需要使用抽象类的接口。
3 B/ [$ m) Y* X. a: F/ M, cCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.