UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   2 H( M7 T7 g4 r8 D0 h
4. #pragma   once
   
5. 
   1 D* _( P; X; Q' c
6. typedef   unsigned   long   ulong;
   / _$ C5 y# B+ s
7. 
   " F3 A8 c) R4 g/ h8 c- i
8. class   MyUPnP
   0 P# V! S! e4 K9 [4 Q
9. {
   
10. public:
    # r7 o8 x# H' W# a% D: R# n& W# U
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    8 |& X. P/ A# i. [
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    1 ?1 \; A. z) M) _5 I% |( B, x7 d" P
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ Y7 o' M, z1 H& n/ N
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
    7 d' M' W- |# V; T% \
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    ) u7 @; I# I: F0 o( S
23. 
    
24. typedef   struct{
    9 J1 d6 D( ~- I2 l" f- c0 @" |' C
25. WORD   internalPort; //   Port   mapping   internal   port
    3 ]7 A7 y# O: D2 M0 C, v+ f6 p
26. WORD   externalPort; //   Port   mapping   external   port
    , ]7 ]& o! M2 Z( T( T  Q
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    ! V; ~5 l8 o1 Q+ c
28. CString   description; //   Port   mapping   description
    % W* V- f7 `) b- l5 E( g, H* n$ n
29. }   UPNPNAT_MAPPING;
    $ @! m/ ~4 |/ Y) [& u: {: U
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    - Y/ E# l! z2 S! I) u& j& n
33. 
    * h( U1 s* P5 L
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    7 x& Q8 e( T$ B5 s0 T
39. CString GetLocalIPStr();
    6 ?; g6 A% C7 M% Q
40. WORD GetLocalIP();
    1 N; P$ h8 u! w. s
41. bool IsLANIP(WORD   nIP);
    3 p% \/ |% h  W1 s' \2 n; a
42. 
    ' ]  o, r0 e( ^+ o; B9 |
43. protected:
    ' g5 b0 [  a5 F+ ]1 h0 i# s6 d
44. void InitLocalIP();
    1 `& B$ U( R8 L) Z0 y
45. void SetLastError(CString   error);
    
46. 
    / J' o: x/ ?* ?: ?2 v6 Q2 P
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    / S, ^, F' T% r
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    % ~2 m, x# {  Z* a0 f4 n+ q
53. bool Search(int   version=1);
    0 V, g; y7 j& B! y( X& e
54. bool GetDescription();
    6 G8 L; p, y) Z6 \1 ~+ p* Q7 C
55. CString GetProperty(const   CString&   name,   CString&   response);
    ' i% S) r0 E* S7 K1 ^& L' ]
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    1 s: z. L: M/ u0 ?8 J
59. bool InternalSearch(int   version);
    ) V- R0 y6 H; x# m
60. CString m_devicename;
    
61. CString m_name;
    " E+ U, C8 m2 b/ d6 @: v3 d
62. CString m_description;
    3 J! g) C  E. b" Q$ ~2 q6 v* u
63. CString m_baseurl;
    
64. CString m_controlurl;
    & f- e# C: M6 u. R7 a. y
65. CString m_friendlyname;
    , M2 W3 p1 ~7 @5 _5 }0 }
66. CString m_modelname;
    & g! E0 R/ S( S$ t% a, e
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    4 J# r- j7 `% k1 |3 v
71. 
    
72. CString m_slocalIP;
    7 g9 L+ }; a/ q# Z0 F: Q
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ) Q2 K2 ~: C6 t( o( I. Q
75. 
    
76. bool isSearched;
    6 f& r5 a- \' Q: Q
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   + X; D( ]" Q/ p+ g4 B$ _. v/ G5 J
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   : l' D  n+ r7 [2 O
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   9 v6 W$ d2 i  M8 I
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   , m# o5 F: r. t; _$ i
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   $ Z- {, d7 \1 g2 h; M7 T+ Y
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    7 [" R0 a$ t. E3 c. ?" J
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    " h9 c9 E* v1 X. q! h6 A- T
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
      T, |9 X1 s! H9 p9 ~
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    , o: l+ _) i. c+ G, p$ K) ~
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    % x8 h+ |  C3 n
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    0 d: u+ ]/ {3 A" Q( |; H
31. {
    ( _# ]# A* {* ?4 z" \! L+ V
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ) D. \5 H) I  a* w( }+ ]! X
36. {
    ; I" d8 l4 f- s" D8 {7 t7 ]
37. char   buffer[10240];
    5 {: P4 [8 T. B0 W6 f
38. 
    
39. const   CStringA   sa(request);
    7 ]  W9 e8 B& E- M. L4 B% @  U
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    $ g) s  \, o& A! e
42. 
    0 \$ L9 B( V2 Z- ]! N! G
43. uint32   ip   =   inet_addr(CStringA(addr));
    5 ^) N6 P. x. Z& j) _+ G
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    2 t+ k! a' v' r# X
47. sockaddr.sin_port   =   htons(port);
    ! `0 P9 e. G4 Q/ ]) L  f
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    & Z' o+ ]1 W* ~# f  w6 M* C7 T6 @$ z0 p4 F
51. ioctlsocket(s,   FIONBIO,   &lv);
    ' z# T3 ~$ a# E2 I0 w: Z
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    2 C' F* ?9 g8 b
54. int   n   =   send(s,   buffer,   length,   0);
    + y7 B  y6 D) q- S8 _
55. Sleep(100);
    / K" c& ^1 D- X/ V" C% P% x2 I3 F
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    . ~. K7 B/ x% V, f0 q
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
      r4 ?3 o6 H+ x. C2 |( |- D; \  l* J
59. if   (!rlen)   return   false;
    ) A0 h% U* k$ c& Y. k4 \
60. 
    ' c9 ~7 O0 t8 I4 e# S; B
61. response   =   CString(CStringA(buffer,   rlen));
    + @, X+ Z( W7 Z/ t
62. 
    ( e" v3 Q8 v- g
63. return   true;
    , x" M! N  f/ `3 l* v
64. }
    0 \7 P1 s* m$ D; U4 w" y" w
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + z4 f6 b8 q8 @) @3 S
67. {
    " J/ R8 d% X# Y6 w9 C; [
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ) @" z. n8 z6 i. f% z$ f$ B
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    4 N5 D5 s/ x0 |$ ~
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    / n, P( S  {% H
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    3 X2 F1 w7 T) P6 W, Q4 }
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & U, O( Q9 p+ X' |' e# J% F
81. }
    ( Y* X* \5 F+ e* F" y% k6 \
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ' `' D1 I4 |4 b, D+ G: f1 M% u
84. {
    6 P' I- r0 a2 Z3 J1 V+ B, E
85. int   pos   =   0;
    ( j# i: G8 K9 ]2 _/ a' Q" M
86. 
    2 m! {# j$ F5 i1 O2 U3 m& s/ E5 s
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    # L9 L" E2 n; B% A7 |+ |% Y3 y4 w
90. result.Delete(0,   pos);
    ! l0 s8 g2 l9 {1 F/ m
91. 
    . s) I' y7 U' _
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    4 n$ `& H$ v3 X3 ^" W4 j
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    , o) `2 H% I: E! Y" W* ?
97. }
      M' j# c7 o1 F5 }; a' k
98. 
    6 j) {. p' M6 ?6 ^; M/ e
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     & |5 C5 G) k9 ^' y  t
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     - _) e) {. k: u
107. 
     2 N5 ?) y3 C  Z& Z- B+ Z
108. int   posEnd   =   all.Find(endTag,   posStart);
     5 W: |3 }1 U, v4 H+ q
109. if   (posStart> =posEnd)   return   CString();
       X6 _/ G( u3 N3 G3 z0 f9 A
110. 
     2 ?3 c; b1 h! J! }& N0 F3 V! j
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     ) }5 Z0 q- J/ J% U0 H- ~, h  l
113. 
     
114. MyUPnP::MyUPnP()
     : v3 a6 K$ {3 F5 E/ k* y5 w
115. :   m_version(1)
     
116. {
     ! X5 C' L3 a* [' N: U4 v
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     " E; R9 r0 q+ U
123. UPNPNAT_MAPPING   search;
     3 p3 K% F( K7 j  m9 g; N- v
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     % u9 G* @9 e8 Q! \, k
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     5 N, R, T- u( s$ r& S( ]" n9 h
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     4 t. p7 _/ r* p; D+ y# @+ e
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     ' R8 h4 j  f  `
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     $ ?7 N3 D8 F& {' x6 ]# G; O, Q
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     / u# X* f- V4 ?; _) u
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     # d( m. R/ F! A! n6 w! ?
144. };
     
145. 
     9 y7 f& T$ N6 g& |7 @
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; [) k1 a; f  t0 l
147. u_long   lv   =   1;
     * T' A- \& L1 ]. G* y
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     / h' ?- `0 b) z4 s* D( {1 I" _; R8 b
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ' F% ]- |5 w: E+ r
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     # y+ `# T1 Q! r: I6 `9 h
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 "),
     % w5 f- F3 I; C" w; V/ k$ L  w# E
157. 6,   m_name);
     ! }4 u  `* |. p, L* Y
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     % O: X  T! X% U; ~
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     5 E0 O0 N; H# [& q6 ~
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     % Z( R( n" w& X2 i; u5 J
167. closesocket(s);
     - @# l& Y+ {+ D3 B" B  N
168. 
     ! c3 t1 \7 N7 n, ^/ R
169. CString   response   =   CString(CStringA(buffer,   rlen));
     0 J( t( q: q: s9 ?* g
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     6 }, x+ z2 e5 }# B+ ]3 U( W
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;;)   {
     " X$ T) ]+ t' Y6 X
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     # q6 U! r8 y, b+ G+ E
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     $ f1 s/ B5 G! S0 ?) u8 |' c
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     ( X: Y+ d+ h% c
185. return   GetDescription();
     ! W- o5 Z/ |) |' Q
186. }
     
187. }
     ; Y* q" _/ y+ d$ K/ `
188. }
     
189. }
     9 |) V: I3 p: u5 Y' M) `  m( c$ ]
190. }
     5 H. {5 I; r( K" r
191. closesocket(s);
     : q( J' w! c* @/ {. F3 \
192. 
     & K" q& K- h/ V% B+ f
193. return   false;
     
194. }
     # j9 X5 A7 s: c8 M5 w+ ?

复制代码

天云上人

以下有关upnp的接口来自emule，
: C4 c( c  ^5 {
" V( L* r# W4 W, r5 _' L
0 Q7 g% ]9 A. S3 S- d5 ]( l///////////////////////////////////////////
; s3 _$ M9 Y0 L4 C, z; K//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

- {. m0 |( f- ?( O5 |$ t' O8 T
1 P; P/ `( }; o- l; }; H. S+ s#pragma once
' \; K8 ?1 p- y* d/ [6 y, y) z#include <exception>
3 i" L7 X* n. }. t+ b. ]

  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
- Z" I* T6 Z3 d( u        TRIS_TRUE
" L- H0 z5 d. U( L};
1 j+ S- V: c: N3 p: G, f

' {1 m* D1 h, c, w8 u0 [enum UPNP_IMPLEMENTATION{
" N( m5 ?$ L( Z, A; S        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
" j( R2 ?2 c! t        UPNP_IMPL_NONE /*last*/
};
7 J( j: @+ W) _) T  t

# ~' _1 S+ R& Z9 w; x5 A
' c5 I& d7 J) C7 D4 q0 d* ]9 N
3 A& t$ e6 {1 fclass CUPnPImpl
6 w% Y* k9 Y  [7 w{
0 r4 b3 S; Y) J+ M4 I3 f+ H5 y0 s: ]$ qpublic:
  f$ }$ B8 I- P5 f        CUPnPImpl();
$ D# `( C/ K0 f+ c        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
# d7 m/ `4 E9 k& a6 z                UPNP_FAILED,
                UPNP_TIMEOUT
, ?2 D' [- r/ i( X        };


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
2 F& e( Z& L% @        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
# X- \6 N/ N( h& J        virtual int                GetImplementationID() = 0;
1 s0 @" f+ D& E' F1 s       
" w, f# C! g9 r2 ?" p+ W. e( P7 [  Z        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
# @$ I! e5 s1 o0 o$ C: ~
3 @1 m( B+ B# f/ a+ j1 f& u/ s
4 Y- {+ c# h& a* g  ~        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
2 |  Z* f; g( m. j7 X        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

1 G( C# P3 W2 X4 v5 p9 N  |4 `4 }
: g6 [4 }& {' c// Implementation
protected:
8 J% `7 d, z" s5 ?5 ?& ^; u. b! i        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
" W5 E5 o. f+ p8 Z( q: c# w& C        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
# k! h' f; @9 [: a        bool                                m_bCheckAndRefresh;

8 m! T7 L8 q' q3 k8 V0 i
private:
        HWND        m_hResultMessageWindow;
: z" s' ]$ [  p5 l$ V9 P        UINT        m_nResultMessageID;

) m$ d! g6 q1 r! L* D+ R% c
};
$ ~; ?5 E- E) u, w
: v: K; _6 @0 H. ~
// Dummy Implementation to be used when no other implementation is available
( p! C! A; f% Iclass CUPnPImplNone: public CUPnPImpl
% m! m; m& c& [: T4 k3 ]1 \# z{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
/ `1 S) v2 N$ A0 q        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
3 O/ K- t$ h! {* v1 E        virtual bool        IsReady()                                                                                                { return false; }
! t2 [6 t3 {) K1 ~; L- y# s        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
1 w! a$ Y' u6 l( ?) p" E};

9 W5 {. v& S0 d' k% }# H
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
5 d3 x  G$ v7 V( F1 o2 J! S

#pragma once
/ f0 W" P0 A0 o* @& j% f#pragma warning( disable: 4355 )
8 ?8 h- ^9 Y) `( E; M" f* D5 p
, F6 @% W- x3 q) J
2 M, o( l% q* ]9 ^; w0 c; z; x#include "UPnPImpl.h"
0 w8 U, }9 [8 A2 C9 V2 N6 c#include <upnp.h>
+ l3 o% @% p8 M; D1 \! [$ V#include <iphlpapi.h>
( ?+ Z6 `9 ?) H* e5 E3 W$ T8 K  W: }#include <comdef.h>
8 n7 u8 y' J5 u% `" [#include <winsvc.h>
  N  y5 i! H% w1 }

#include <vector>
#include <exception>
#include <functional>
4 ^5 ~) h8 \- e% w$ z


& E. n. r* g- y5 _/ O
% I/ A( V7 i1 D9 Y' Ftypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
3 D& k# l3 k5 Ttypedef _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;
% I4 y$ D4 r; L5 E% u1 b8 Atypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
3 D- i( O. h0 r; z6 n( }" z- itypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

" S2 r# w9 S4 q; M2 v
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
9 D8 n8 d7 g) S* X; g2 V( Y2 H; M
4 [9 j4 Q3 ?9 I' i+ r
typedef DWORD (WINAPI* TGetIpAddrTable) (
2 {' L' K7 C! _* x% i! j  PMIB_IPADDRTABLE pIpAddrTable,
) U) a) a" M( V8 y! c& [  PULONG pdwSize,
+ ]3 ]: m' S# `9 J" ~& v  BOOL bOrder
; i6 u5 v7 U8 W);
4 a# F% t# q- j9 W& ^
; K. o0 G' i; @
typedef DWORD (WINAPI* TGetIfEntry) (
3 J4 z7 u1 K5 q) p$ {7 E) c% U  PMIB_IFROW pIfRow
);
# P4 D6 V+ |3 `

CString translateUPnPResult(HRESULT hr);
1 h8 U4 u5 }% i& M9 `8 e$ PHRESULT UPnPMessage(HRESULT hr);
9 Y, C; p! g: h! I0 O& K! e* T
* B+ F0 z. z- x* e
class CUPnPImplWinServ: public CUPnPImpl
{
2 [, y# R3 Q) `8 U" S2 v        friend class CDeviceFinderCallback;
9 S: g  y/ |' \; r        friend class CServiceCallback;
% j3 o2 _1 [8 Z( b+ l" }; R. E// Construction
& n# j: a# u( }- n+ G* f+ Tpublic:
$ B/ n0 Z' q0 F- P) T        virtual ~CUPnPImplWinServ();
4 t+ A7 G+ l" e$ z2 ^$ \: n        CUPnPImplWinServ();


5 V5 J( H/ s3 w; B" o        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& Y9 I0 Z: ^/ A! f: }2 x  k        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
" Q: i* X: u4 }+ n        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

! t5 P# F# a2 D+ H7 ^+ [% i  j
/ b4 m8 X2 s2 y+ L( w        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

0 r  X4 ^# m7 j% u- r4 U
& c% F. B& t6 Fprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
7 t4 E2 R1 X0 }: ^, W2 ]1 j3 E        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
/ \1 Y& Y* O$ r" \        void        Init();
# I5 w) B$ m! i% `6 Y' V. G# j

        inline bool IsAsyncFindRunning()
7 c* Q( l# N8 h# r( d        {
5 A9 Z2 v3 T; @, d- p/ v                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
( d9 b  E& z7 h( `+ k                        m_bAsyncFindRunning = false;
                }
8 T1 M. \" ]$ P1 x' C                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
4 s" V- J8 O8 M! {) i7 q% o8 x                        TranslateMessage( &msg );
2 @8 b* h- X; ?% B* _2 g" G, ~. O                        DispatchMessage( &msg );
" a) v7 G/ n! i5 I; |5 n; R. {" _                }
                return m_bAsyncFindRunning;
        }
" P) ]4 ~+ \/ s+ H- {6 o  l: B
/ l" b7 E4 x3 p+ v7 k
( _( X, E7 j: p        TRISTATE                        m_bUPnPDeviceConnected;
7 X# R+ X7 p0 R/ s9 z
" w$ N. |4 \  o0 x8 r- M8 \) I
// Implementation
8 ]/ ]$ J) p7 ?) L        // API functions
% c7 g; @% b4 }, n( N6 T        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! ?8 `4 @& u2 e2 c        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
7 a0 a' o! ?1 @5 F- K+ d+ I        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
* p% ?5 P5 X! X  p        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
2 p% C; [5 s6 [  ]7 t

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
# [0 x5 X- x2 ~5 ?3 z+ ^

        static FinderPointer CreateFinderInstance();
  U" _7 t0 O( N( B        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
  S5 V5 M7 p9 |4 e, k. o                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
0 f: B& t) N) b" P                result_type operator()(argument_type device) const
0 V, t" [( `) Y8 C; D: r                {
3 \- A5 ?- U/ T: w' z8 X                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
' C9 e- e* S& G; ~4 k/ {% q% P& G
3 H/ }* I6 W9 M  Q: I
                        if ( FAILED( hr ) )
. H0 K2 W. C6 C$ P- R                                return UPnPMessage( hr ), false;


6 c! U) v! w: t: v                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
& k) a7 p1 s% M  p: I+ k7 O# ]       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
. w5 j: K7 o) J6 v8 ~        void        DeleteExistingPortMappings(ServicePointer pService);
/ [5 H7 B7 w6 G" Z9 ]; z        void        CreatePortMappings(ServicePointer pService);
- d. Y: q0 X9 R0 ?        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
5 k1 w- b2 ^) ], F; `        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

% V; W$ |9 y5 U0 U, D% V( Q
: `  q, Q& Z( ^, a5 \( y- F        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
! E" [# q2 Q9 A# p/ j        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
0 r  S5 ~* F; j9 b5 R0 N        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
2 U5 ^8 m2 u% @        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
  W. E- _7 G+ w3 F! d8 w        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
* d  |; M* ?* d# ]2 h- _4 j+ d- w/ s        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
6 Z- z% M7 _$ l! U- }  L. P        CString        GetLocalRoutableIP(ServicePointer pService);
- `' X- Y3 L; t" Y7 k* e

// Private members
5 r" ?% L1 |7 v5 l7 wprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

& R3 \; c) b4 P! `
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
1 I# O0 l5 E3 c# c3 `" e% X        CString m_sLocalIP;
! c  Z2 C3 {% m- z- }( Y/ P        CString m_sExternalIP;
& P0 W$ {  A- R9 w) }4 u        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
( h) r& ?9 Z9 x( @, i# F( O/ @        bool        m_bInited;
+ W$ N% ]+ t9 W        bool        m_bAsyncFindRunning;
, A  p+ N8 |2 ^% n1 @        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
- Z! O' R- d; w9 U% U        bool        m_bSecondTry;
1 J9 I7 P$ ]  z- F) U' n        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
8 V/ t! L6 G. D' a8 Y3 U2 q        bool        m_bDisableWANPPPSetup;

3 N% K% N& S4 g" s
2 p' ^& C( J9 v};

  k! W+ g% B7 K0 U7 a
  I+ G5 h+ s5 K! Q+ p7 `& W* g; E// DeviceFinder Callback
/ s, R9 m  O/ t9 H: Jclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
# w/ c* K( e" \5 |6 c{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

. F1 p1 e  i. T. R" x
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
: s' _! x- s% K. @# K7 o6 j( o   STDMETHODIMP_(ULONG) AddRef();
: {. n- [- @8 u# d* c) t; a   STDMETHODIMP_(ULONG) Release();


// implementation
private:
( z8 s5 S: |1 m6 o7 q0 X9 g$ [* h        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
  {, o4 H6 j8 a8 L        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
. `+ H5 p: U5 B/ {2 ?% `) r

' v7 N$ D  x/ p. l" `private:
  ?' C8 m9 @3 e' m        CUPnPImplWinServ& m_instance;
; }: x( ~& M5 J0 S/ b        LONG m_lRefCount;
};


// Service Callback
1 Z5 @2 N6 p. }' i4 B+ R% |class CServiceCallback
        : public IUPnPServiceCallback
, _3 h8 n' p7 b{
public:
( s5 P, H% A% |+ u! h        CServiceCallback(CUPnPImplWinServ& instance)
5 B9 u- {; O0 V7 C- ?( M4 o                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
$ d. C$ O: X- {1 x   STDMETHODIMP_(ULONG) AddRef();
; }% O* D" }1 ]. T& a   STDMETHODIMP_(ULONG) Release();
; t* d* c6 i- O$ C3 p' R
: N0 G. i& N: h+ D4 N: s8 _
( @7 Z: _% }, d4 u" ?// implementation
' L1 n2 g* u7 B! f# sprivate:
- {1 F2 U4 j, P( q        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
. f( u. Z: c. }5 l3 ^6 e$ ?
; x" \! ^7 F; f  d/ \. l4 a1 Y% E
6 \, C4 {- E$ k' A* A0 c7 nprivate:
% x/ s6 g$ Y/ c9 i+ R        CUPnPImplWinServ& m_instance;
  B1 f/ a2 O3 [1 A9 V        LONG m_lRefCount;
};
' [4 x1 N3 O1 k! e; c
* v# D1 h  e2 A- e) @% f* v2 J
/////////////////////////////////////////////////
9 l* X; x8 D, c

使用时只需要使用抽象类的接口。
! j3 z6 O* z, p& ACUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
) y- k8 n" {" g) x1 n( z- F9 RCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
9 U/ i7 B8 y! g+ P; ECUPnPImpl::StopAsyncFind停止设备查找.
8 }* H2 a; V0 H  M! P, }# p# \1 gCUPnPImpl::DeletePorts删除端口映射.