UPnP

天云上人

/*uPnP.h*/

1. 
   7 }( Y1 ^& i) p( ~3 V% A# p
2. #ifndef   MYUPNP_H_
   
3. 
   8 t- @" @- s" z( u7 M; J, D
4. #pragma   once
   
5. 
   / [0 C6 u  Y9 w+ q, u3 S
6. typedef   unsigned   long   ulong;
   + m) f* `3 T) q$ m/ E
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    0 W$ Y# [1 ]4 U4 O6 N$ i7 x: @' }
11. typedef   enum{
    7 \$ K" T- d: \5 `# J
12. UNAT_OK, //   Successfull
    $ z! W  w. m+ @4 w
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ' y+ o# J: ^& d$ {3 d
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    ) X6 L3 Q  x7 b: Q
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    % h% S, }4 `: @
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    ! x' F: L5 U. ]1 }! P0 [/ I
17. }   UPNPNAT_RETURN;
    3 O5 D, S8 L4 R4 z( w
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    6 O, B: d7 i- @) f4 A+ Q0 }
23. 
    
24. typedef   struct{
    % c/ Q" z* ^2 X( e- |# E
25. WORD   internalPort; //   Port   mapping   internal   port
    5 t, F. D6 K/ G" s0 u  O
26. WORD   externalPort; //   Port   mapping   external   port
    ! N1 ]  W4 Y: K: G
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ( E% i! i  Z" y% _
29. }   UPNPNAT_MAPPING;
    
30. 
    ( V6 A' N: U. _. O; S" I+ l
31. MyUPnP();
    5 y3 F+ y. K# N$ m& h
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    + R& Y2 g4 X' R6 g
36. void   clearNATPortMapping();
    1 {% w6 d+ t1 b0 T/ |3 V
37. 
    
38. CString GetLastError();
    # u9 Y7 k6 J1 ]/ \
39. CString GetLocalIPStr();
    5 D7 u# {" I- r' s5 ^% {$ f7 {5 U
40. WORD GetLocalIP();
    7 q8 R" f# ], `6 ^" ?3 ?) ^
41. bool IsLANIP(WORD   nIP);
    6 `2 K1 a1 B; M7 ?
42. 
    
43. protected:
    
44. void InitLocalIP();
    0 ]2 G; B' o( r; e, g8 o
45. void SetLastError(CString   error);
    " q% U8 l$ X+ s
46. 
    ( r9 R* j+ c3 N6 w- o$ A( L6 o: P
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ! ]8 ]- H8 R8 \" C9 x( H) ?
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    + U  ]) ]  M2 V2 w; V8 m7 o
50. 
    ! |: @1 {* i) I7 f3 G3 O9 I
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    - L/ y5 r2 g2 R- r5 o
53. bool Search(int   version=1);
    5 `6 G4 Q0 p" o/ J( l
54. bool GetDescription();
    - \- k: V4 E4 A
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ( T% o6 E) X5 ~$ w+ h- ?" k
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    ! Y% D- G. F+ e" H
63. CString m_baseurl;
    1 A6 l! N4 k# ^5 [
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ! ~- [4 _* L* \' K) i( J, A5 L
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ) z$ h; L6 j/ K$ j1 x
75. 
    0 I/ }" h! o) f) X7 j6 N' t
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   ) n7 v. s/ U  \
4. #include   "upnp.h "
   
5. 
   9 z% ~9 M# ?) D, o
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   # F  x6 ~8 h& r. b0 U/ Z) K
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   : ]1 a9 m1 i  F! H  V+ T9 M
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    ( J8 `8 `. k' \1 b4 y) q
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    0 {, k. t1 U$ m2 L9 }: D: G* d6 A" e
17. {
    2 [  U- Q9 H* W$ b
18. CString   s;
    
19. 
    0 w  x# v2 L2 c3 F3 V7 [9 W. F
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    5 T! u% j: E5 d. W
23. }
    1 K* L  a: h% C! D: d3 }' D$ q- a
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    0 R2 \3 h5 g. X5 P" }1 n
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    & {' \% S& a% D7 v, O. h: z
30. const   CString   GetArgString(const   CString&   name,   int   value)
    5 k5 ?4 u' R8 T2 j( ~/ ?/ v- G
31. {
    6 O# }9 ^+ T4 [' y  f
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    : z8 a6 a+ k5 \* {) y7 a
33. }
    
34. 
    # V+ W. X( y8 D  k) F& P. ]
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    $ |9 d# r% _; O5 J
37. char   buffer[10240];
    
38. 
    ; m( O* F* U* B* h* f
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    - _( k6 l* G' y* s$ n
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    ! b- e) u  x' A( |# L+ K4 o
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    5 R& P3 [2 @3 u
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    " d6 u6 [+ X- V% F& j3 m' C: s
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( w* _; n( c4 S' X0 J- z/ B& H( [! E
53. Sleep(20);
    $ j5 d! X- ?+ n% g0 h6 T
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ! {6 O4 k, r6 R" h5 k
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    - j- u& m% l9 a! U8 F  ?, Y
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    - f- ?" V8 _$ H0 R
63. return   true;
    
64. }
    ; _, Q) v" g7 h% F, B
65. 
    ' [3 T( C7 u8 @* ^% G
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ) [3 b# b5 F  U! f5 T  D6 W2 J* J) [
67. {
    2 v$ C; L# n. l5 _
68. char   buffer[10240];
    0 p0 j( }% I6 R+ Y9 z, B  O
69. 
    4 q: y0 M7 f0 W* F$ ~! p! H8 s
70. const   CStringA   sa(request);
    $ |3 C2 p! H, @6 e. ?0 u, l
71. int   length   =   sa.GetLength();
    3 q. p* j8 F+ E0 J' O
72. strcpy(buffer,   (const   char*)sa);
    1 a1 u3 N5 A( z6 q
73. 
    $ p1 W. V# p! ]# w* f3 }: |& h% e
74. struct   sockaddr_in   sockaddr;
    8 a/ ^9 {& ~% L# k
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    % X' v+ B% M6 f# @: y9 A
76. sockaddr.sin_family   =   AF_INET;
    ; `" @: Y3 L  H' x7 W
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    $ _7 w% u% N- B8 g
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & d- D6 S+ S" D) W' ]; S
81. }
    
82. 
    , ]% b- Y% r0 d9 C
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    2 f8 x5 d. T& J
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    0 d/ `) a4 J" g* C. @
90. result.Delete(0,   pos);
    7 C, J/ D* Y% R+ `( {7 s6 L
91. 
    * q& L: Z' k% E3 o% ~' t
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ; H7 e0 w- d! B  V2 z3 O0 s$ j
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    % ?$ \  d* I- u% w+ v) L
96. return   true;
    ' q( M8 G1 w1 V! S3 W+ l
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    5 y* g3 W0 |4 a7 |
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     + W7 n+ x5 H3 Z# f4 n1 R6 s3 x
103. CString   property;
     6 D. h  G. V6 G1 Z, @5 f) `6 e2 C. B- }0 v
104. 
     : Q) S& z% n8 K1 _' y9 S, U+ L% D
105. int   posStart   =   all.Find(startTag);
     2 F  Z. U% H' n( j+ b$ s; S
106. if   (posStart <0)   return   CString();
     
107. 
     8 i: {' V. [; o2 y+ q6 K7 L
108. int   posEnd   =   all.Find(endTag,   posStart);
     # T( d, u% w; Z  ~; H
109. if   (posStart> =posEnd)   return   CString();
     & J8 F  F( b: T( o& }- v
110. 
     ; L  Z: o: R1 `% [) }1 Q
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     & V" z4 `- Y) Z
112. }
     
113. 
     " F: ?. F0 G* L$ d. d& b
114. MyUPnP::MyUPnP()
     # i2 z& y& l- ~- ^8 P
115. :   m_version(1)
     
116. {
     ) A/ \" h: b0 o1 Q
117. m_uLocalIP   =   0;
     * C- [0 W" {7 L$ B- k
118. isSearched   =   false;
     & u6 O- P, Y5 ^
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     ; w4 Q' P! }# q, [  t
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     $ w: m( r6 y- n5 f% q
127. RemoveNATPortMapping(search,   false);
     . t0 R8 \% p$ s0 J# x
128. }
     1 `  L. R: v& B
129. 
     
130. m_Mappings.RemoveAll();
     2 t# b- d; f9 n
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     1 r5 g4 \6 D5 `- S& I6 s7 w
135. {
     , ~0 D5 P; Q% E0 v1 [. r
136. if(version <=0)version   =   1;
     * D+ [$ P+ T& v( A  V
137. m_version   =   version;
     1 [  t1 X- z# f1 m" t( ^( N
138. 
     ) E' |9 `3 U4 I/ k/ a( ]9 M# y
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     # Z# |) P, E& M0 h& j1 y
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     " M) Y5 h6 e( t
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ) b. e; t2 H0 R5 d+ X) C1 F
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     % D7 S) v$ k5 t; q! f
147. u_long   lv   =   1;
     + ~' E; T: R8 S9 R7 ?' ~6 Y5 m, I
148. ioctlsocket(s,   FIONBIO,   &lv);
     ' k% l$ n8 x2 Y
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     2 d0 H  n1 P0 Y
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
       i! x( s9 T1 b- f0 e7 w; K
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
       \+ I" I* i) a8 [/ Q8 v: |% l
155. CString   request;
     / n0 d  f: b' u% W
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);
       m; L7 B9 O) a8 Y
159. }
     2 j" ^& _- F3 q% x' V3 e
160. }
     / @& [' r. g8 a) I3 P' [: w: I6 D
161. 
     
162. Sleep(10);
     0 u( N  c; @6 Z* X! i
163. 
     
164. char   buffer[10240];
     + l8 K$ v$ y) Q2 A6 c2 I
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     7 |: t9 H. A' n7 _  m
167. closesocket(s);
     
168. 
     0 J8 H# y0 E# ]7 _3 t! `
169. CString   response   =   CString(CStringA(buffer,   rlen));
     1 d3 N# j1 `$ c  ^' ?# N; {: L
170. CString   result;
     $ E1 [+ e/ l+ m" K# v% a
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     : e7 r5 e' V% ]. @% L& p' \  P
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);
     5 v. J0 n# M7 {! N* V7 u$ E0 g( S
178. if   (line.IsEmpty())   return   false;
     # l8 \) {4 q0 w1 e( O3 Y$ p' C
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     / d4 I8 z) g* R; k8 w) m/ v
182. line.Delete(0,   9);
     ! ~' u% f( F8 J, [4 V4 o/ S; W
183. m_description   =   line;
     % I, s: s# O+ O$ h3 F) S! J
184. m_description.Trim();
     
185. return   GetDescription();
     . f4 `) q* E  v, R7 B( J
186. }
     
187. }
     " v  j  m. S  Y1 w9 ^
188. }
     5 Z9 U! T9 {- T/ G0 e
189. }
     ( Y' A. Q2 C6 G8 L* x/ Z
190. }
     0 W8 w7 P1 y7 T  ?- Z' g- ^
191. closesocket(s);
     
192. 
     * b9 E. S* ]' l' Z5 H* X
193. return   false;
     # z) n) f+ x% K* ^' N$ G- @, a5 s& o# |
194. }
     - v' q/ f5 ^$ r$ d% F% U" w

复制代码

天云上人

以下有关upnp的接口来自emule，
6 N0 l0 q  c$ `7 x# Z

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

2 W8 K/ G3 _# ]9 k% _6 E/ Y
* j. v. H; ~# ]$ q#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
2 A. a( ]9 w  H/ |};
5 C6 P$ D9 p" ^3 |- T6 _# _( x% A
  F  z" F. j: k9 l  _
enum UPNP_IMPLEMENTATION{
) s  r" L2 s% L        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
" ?7 c+ q* t" ]* H9 G};
5 ?4 I! F& c4 ?7 A; F7 p( o


# x9 b- M" G7 G0 f7 s
class CUPnPImpl
{
public:
1 \. Q) R8 d) y2 G0 K        CUPnPImpl();
  \/ c/ N" Y2 u! E        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
/ [5 A# h* m# ]% _) C  e# Q7 d. f$ {& A        enum {
# I. Z6 i; N! X  M) {- D9 a6 I: c                UPNP_OK,
                UPNP_FAILED,
. k# e) K$ `+ \# `                UPNP_TIMEOUT
8 A- T* N- \" ~# E! L        };
# C! x. x, P' X# n* y
0 O% r( J% O3 `& @
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
9 r8 d+ F- ?; C# R( V        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
# e4 @0 Q! n' O/ G! O' c0 N9 N        virtual bool        IsReady() = 0;
1 t  u' h# G8 b# J, j$ k* @- m% g        virtual int                GetImplementationID() = 0;
4 ^& I8 Y7 v+ L4 n* L# d) Z       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
9 n2 L, W. ~1 c0 K% M3 i0 r3 I- o

$ R4 }0 @. p) u9 J$ d        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
% U! `7 e- E1 Q1 d( L- d        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
0 E2 d* A5 I7 _# |

// Implementation
7 [% M' ~- C( G0 I7 a3 s' rprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
/ ~- J3 ~/ @2 O, L        void                                SendResultMessage();
1 d/ [0 s- e9 u6 X, x5 ]        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
+ x3 M, f+ G7 Q% U% s
: {7 d1 [1 X$ d+ g  \: n' w  |% t
private:
        HWND        m_hResultMessageWindow;
! U0 I# w- y# e3 Z* w+ I% N: L0 y# u- h        UINT        m_nResultMessageID;


};
4 z+ H. P& {, e4 \2 G' a+ v3 J

) a/ ^0 E1 r" L7 M// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
, m0 K2 l1 Z* S4 s! x8 A, ~- V8 B) qpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
/ O; V' z( `8 w& k  _. I+ S        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
6 _2 g# M, z2 F7 o+ Q/ d! V" t: F        virtual bool        IsReady()                                                                                                { return false; }
( @  ^- u6 P& R: c+ n2 F        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


6 ]' P: a9 `- B9 F% P# {, v* H/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

0 Y: G; o+ ]9 M9 H2 _/ a' h
#pragma once
4 l7 b! B( u3 o+ }#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
& _% Q& G# t% J/ n#include <winsvc.h>
, X! `( b& r' x  I6 s
2 R8 F- T$ e& P
#include <vector>
# g7 ^" V& N* V  q#include <exception>
; N' W* f3 q7 o  E$ m#include <functional>

7 P9 ]& Z( k- F+ ]( N


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
% W/ U5 r4 c- B+ k7 mtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
/ y; E6 y5 k3 A* r; H7 D% Gtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
7 S) {- z2 S9 C$ F2 ftypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
% r6 \0 n* Y9 J; H

typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
6 q7 s% l& Z  f' I& T( M7 m6 o  PDWORD pdwBestIfIndex
+ M' v: v( W! O);

8 H, g# R% q+ I7 {
0 ]# G1 |) Q5 J' y( t" ktypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
! z2 M0 m" T: g  R, H! g  BOOL bOrder
5 H& \. X* F0 z. h/ M);

' V% ]! _6 U/ [* u+ v& q% ~$ i, D
7 s- i3 c  x  q+ R/ Atypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


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

' \0 Y% r  _/ J8 P
class CUPnPImplWinServ: public CUPnPImpl
! o; O) V0 G) f/ T9 S0 b. n. v{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
4 U+ Z  F" {, O3 y// Construction
public:
        virtual ~CUPnPImplWinServ();
: D% E, K6 |( a' _5 @        CUPnPImplWinServ();

+ u6 ]. U& M6 u
& O3 o0 ~$ x: B- D        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
& W! D' Y* D0 B        virtual void        DeletePorts();
1 N* N' t* V" D. Q" Q        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
' t& s' `* H$ c  f
6 ]' c8 u: [( {5 y  Z/ H2 `2 T" _' o
6 r( y3 _* o. L. [8 H, R8 [9 t0 [        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 M6 r7 J% `* G. x# \  D        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
7 Z( |' G3 x5 U1 Y- j

  V6 x6 f( C' U/ A( B6 J. j7 C+ \protected:
4 w( `. s0 e5 ^+ p& N6 U        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
0 i0 Y. M/ @6 H5 V, F. @; O        bool        OnSearchComplete();
        void        Init();
! o/ \5 t; |) q: r3 g/ m
4 f- j, T' n8 Y' X% d5 o1 D& S
        inline bool IsAsyncFindRunning()
2 X+ |. x- x9 D        {
- o! Q6 [" w1 O+ W                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
9 Y' L$ O8 ?3 ?( I/ Z9 n                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
) h) q5 I8 |( B                        m_bAsyncFindRunning = false;
                }
                MSG msg;
' m  |- }) D" `                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
: y- Y0 y/ L; w2 |                        DispatchMessage( &msg );
( ?  R( z) h8 `3 x$ q% l+ g                }
                return m_bAsyncFindRunning;
$ Q7 `1 U- j4 r9 y9 M        }


        TRISTATE                        m_bUPnPDeviceConnected;

6 w4 s! b: q/ h4 `) |
// Implementation
' J8 a' U" e) `+ ?: F0 S        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
. a5 y5 N% l* I: F9 D$ L        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);
- K* v& t- `+ y  R

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

* i7 Z1 }2 k) V( U
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
3 H0 p0 Z2 i" s5 N9 h+ d                result_type operator()(argument_type device) const
' f7 r+ S4 j3 u# L, x: _4 m+ ~! w. G8 Z                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
$ I% {, A2 x* A
2 H/ m: }" ~( B
                        if ( FAILED( hr ) )
% w+ V  C/ ?5 A. i/ l                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
) e6 Y5 x$ v2 }+ `) D& U5 r6 i, f; s                }
                CComBSTR m_udn;
% c- g  @7 B* u' P7 b        };
       
5 |' q, }9 `: J        void        ProcessAsyncFind(CComBSTR bsSearchType);
  O% E6 {: L3 k- Q" b" S" ~" ^        HRESULT        GetDeviceServices(DevicePointer pDevice);
- e2 _: z- y5 n5 u* o        void        StartPortMapping();
# m' t7 O) h! G$ v9 x        HRESULT        MapPort(const ServicePointer& service);
% z; c8 c: \* |        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
9 D1 I3 w: F- c) N$ U& A                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

$ o3 S6 @3 ^5 r
        // Utility functions
7 K' [+ r( Q# ~4 Q8 Q' e6 R# s        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
6 c6 Z( N& U5 Y* e        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
0 {" |% e! x- p1 K1 J3 {        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
9 p1 q6 x' c  Z        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
- K/ y* d! k- H$ A  m! _        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 p! @  j4 @  v: q3 [* o        CString        GetLocalRoutableIP(ServicePointer pService);
! a3 B  {7 [5 r8 H! E6 }

// Private members
private:
" Y. {& `% W! j8 ~: ]/ M" r3 w        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
  q1 J" _: G$ H' k8 M& z& X+ d9 V$ E        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
; V; I- T- u( y( G" ?        CString m_sLocalIP;
9 C0 i0 `  |8 M6 v5 t        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
' I5 O( q" D; g6 G. s        bool        m_bInited;
        bool        m_bAsyncFindRunning;
/ g3 H  Y3 P3 F7 \9 S4 Y        HMODULE m_hADVAPI32_DLL;
6 l/ J  `1 M4 P8 [; O: E0 ~; P0 J        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
6 q$ P; g: ?! B6 l5 d; _        bool        m_bServiceStartedByEmule;
/ @2 F2 R& N- T2 t  w        bool        m_bDisableWANIPSetup;
( m+ j# y( d6 ?( M$ c. b0 ~" Z        bool        m_bDisableWANPPPSetup;
5 W1 T8 C0 b( F& }; F
* y; T7 x3 d  b: p; z" {: P* a# ?
  R4 Z( @0 z9 J" a% a7 x};
' N8 L: B7 v, n' g* _

// DeviceFinder Callback
7 I6 \5 {1 b( a2 E' ~( zclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
% E" x6 T: ^( W4 a+ }2 n  E% S{
public:
+ l/ j4 p, }$ g; N# O6 U        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
# j, M0 E; c* W2 l: d; Y        { m_lRefCount = 0; }
+ t7 W/ h8 ~1 ~7 [% j; s$ E" I

  \9 E2 z3 z. ]2 q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
3 `+ R/ N5 A4 [8 j" W3 M& g   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
4 ?* p' |# n* _4 [2 L* _# Z+ w

// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
7 h8 n& B* U3 ]# f4 z        HRESULT __stdcall SearchComplete(LONG nFindData);

) e; p' r0 w4 \: m9 U! n; }
, G/ N' N& ^0 z9 ~1 Vprivate:
' C" f- b/ Z+ k& {. C2 ~# l. b        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
* W- v9 F9 t, c; J. Y8 I};

8 J( h( b5 W# X  _# B6 @
// Service Callback
7 Y3 X4 E$ }1 K. b% o% m9 gclass CServiceCallback
3 U+ Y5 T0 y1 w        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& q+ F) x* h9 A% x& I/ o   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
$ h' D7 D0 A( r0 v* v6 [
% T" ]2 P/ r  r
// implementation
: P) U) H  g9 m/ R0 p$ r* s  ?" Xprivate:
! O2 |* l. z  G3 L/ g        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
% ]; x" r, _# b' u: [        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 q6 D. k2 _. A8 i" Q3 i
private:
        CUPnPImplWinServ& m_instance;
& g) N1 H! d0 }+ g+ ^        LONG m_lRefCount;
};
. r1 P8 m0 _: K* h. m, }

/////////////////////////////////////////////////
* F' W! _$ z( f2 t
8 N! {5 [9 C2 i, q+ W7 ]
使用时只需要使用抽象类的接口。
* p  k. {. `" C1 e7 `$ LCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
) r7 o' l4 m2 G) L( w, _' m5 bCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
  v4 D8 x  ~1 d5 i, uCUPnPImpl::StopAsyncFind停止设备查找.
+ v% h% M5 j6 O, XCUPnPImpl::DeletePorts删除端口映射.