UPnP

天云上人

/*uPnP.h*/

1. 
   7 S: y: W8 D' w) o/ h
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   8 c0 D& Y  ~6 W* `& X" L9 G5 ^
5. 
   ! w8 i* R, N3 Z) P& C  _7 A
6. typedef   unsigned   long   ulong;
   
7. 
   ) X0 n( E9 ^/ d; Q2 {7 }
8. class   MyUPnP
   / ^) O3 y$ l. a- |2 g
9. {
   
10. public:
    
11. typedef   enum{
    + l7 A* k8 r3 O# X2 H8 L
12. UNAT_OK, //   Successfull
    " p( t5 E4 J5 V% U+ S
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
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
      j7 V; m5 X5 m# ^0 z
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    , z- f) h# i) L
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : [4 c. G+ o1 X  H  J' E' G/ s" o
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    # _  p1 d% h3 w+ j
26. WORD   externalPort; //   Port   mapping   external   port
    6 V1 }* `+ j- f: x3 \" W# }
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    - k* F3 d, y+ k9 S& k
28. CString   description; //   Port   mapping   description
    . s' A( r' _- O8 w; R, I+ r
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    ' Q& c( K9 H" v
32. ~MyUPnP();
    ( m0 |! P# r3 j4 A6 j6 I
33. 
    * {6 U7 c5 ?4 |- Q  n0 K* k
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    8 b' P& m7 D/ }# u( l2 M$ \
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    . a! R+ `& I: c$ {$ z" F  H$ G/ N
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    3 h) h' V$ r) A9 T
39. CString GetLocalIPStr();
    3 F( O7 }$ |5 X8 n6 h0 W# f, s
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    / M0 b2 c) n) \( Z8 i" J1 E; F
43. protected:
    2 d0 N6 G- ]6 v  G# R
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    * y# @; z! N. y' |8 b" O6 }' a5 \
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    1 F3 f7 @% @$ D# S2 I# |
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ( a& B6 N, X3 ?- L, ^8 {3 j* J
50. 
    / Z0 x# J/ z0 Z2 Z
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    2 D( w) L% i7 z9 i2 U1 Y
55. CString GetProperty(const   CString&   name,   CString&   response);
    & }- ~+ `4 c9 o
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    9 s! j2 E/ g, r, G$ w1 Q' O
57. 
    ) u8 `$ J  l+ B( b5 w
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    0 Q' \3 R" y2 h8 p3 @
59. bool InternalSearch(int   version);
    7 m- r3 D6 M$ G: V1 p8 t
60. CString m_devicename;
    3 A& B9 e/ s3 h' T
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    5 X7 B, f& O/ h( E% A
65. CString m_friendlyname;
    & U8 W' c' [( h4 Q# H
66. CString m_modelname;
    0 b2 U! A% t- \( y' F/ j
67. int m_version;
    
68. 
    1 K, {8 {8 [  q
69. private:
    , H5 }/ H0 U3 R
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    1 C) A2 a: J( o
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    0 W' \+ Q# _' i; P* ]! l- w- ]
74. WORD m_uLocalIP;
    ' W4 u" z9 J$ i+ @' `# u$ Y
75. 
    ' L* x  n, u5 N7 F
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   % u. j/ [3 u* x+ b: ~
2. #include   "stdafx.h "
   + u# H0 s) w7 {
3. 
   
4. #include   "upnp.h "
   ! ^0 U/ E' p7 h5 C( ~
5. 
   $ y8 F3 p: m1 ^* u
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    0 }/ r) k. _2 t# ~1 E% ~* C+ \
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    9 q! Q/ e) A+ J0 e1 m: g
15. 
    
16. const   CString   getString(int   i)
    
17. {
      M+ G! i" _% y0 Q! S; n
18. CString   s;
    6 ~7 n$ {/ H+ s& [
19. 
    
20. s.Format(_T( "%d "),   i);
    ; \. A. E+ Z- X  e# @
21. 
    9 X( I2 t% ]6 c, H; y
22. return   s;
    8 K6 m/ ~9 P* ~5 ?
23. }
    - u% n* M+ Y- [2 n
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    , P; U+ A5 o* V- s. R1 Q( f
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    - b: I* J/ W( P" P2 C" O
29. 
    ' ~5 U8 H7 C2 T% U7 t
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    * @8 d. d. X+ J# a" x
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    - H- C" z( s. `6 I* ]
33. }
    9 z+ o* ^9 W* s$ r
34. 
    ; A3 j4 N8 Z+ P1 Y" Y
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    9 p7 r! Q) ]1 T' w
38. 
    
39. const   CStringA   sa(request);
    9 k" U$ a2 }0 }7 o0 n' _5 Z+ z
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    / k+ [. J( ^2 ^: V9 n- v
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
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;
    ( Z9 a% A: B; s+ Q6 K- d, U3 |
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    4 r3 o) c7 A$ {& o+ _/ @
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    . t: D7 g, T% p, @
53. Sleep(20);
    9 D% E6 B' Z% U/ ^' Z4 V8 a
54. int   n   =   send(s,   buffer,   length,   0);
    5 b5 K. [  {5 A) _) H7 a( W
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    & r- a+ M4 @$ ^3 B# j: w( W
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    # C. I. t. c9 x! d. J
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    * Y2 c' R- X- H5 B/ H) v; a& o
63. return   true;
    
64. }
    % ]7 R. p6 `6 ^# h: i# n
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    1 F, ]. H0 Q/ [- R+ L
69. 
    
70. const   CStringA   sa(request);
      K/ a) l- l) z2 E1 q/ f  ]
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    & e* `( ?+ {6 a. @* I
73. 
      h% U: i  @* ?' o# X
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    4 t; U) W" S' X; d
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! Z5 Z, u0 Y! l2 M
79. 
    8 u- ]& b+ V3 n; m
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    ' L# {5 d" D: p3 j5 x5 Y8 ^$ E: ^7 m
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    $ G5 q; o' U1 V+ r& X
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ( r6 \3 }' e4 P
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    - U; B6 x5 v+ M4 W
93. status.Tokenize(_T( "   "),   pos);
    % w" L, K& N- |1 T7 e" ?
94. status   =   status.Tokenize(_T( "   "),   pos);
    ) a. x! w& X5 w. ?- Y% K
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    . `* L: r2 z4 H2 z0 m
96. return   true;
    . Q6 _0 G0 y: s) @
97. }
    
98. 
    0 q9 y3 ~  u4 E& L1 P% L% ?
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     6 P# E; ~3 f6 @+ A& ]" n
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     " a, S) k  }7 r1 V3 g
104. 
     9 F& S$ H. H1 s/ Y: m' o" k
105. int   posStart   =   all.Find(startTag);
     $ L& I/ u$ `  d; ]9 Q
106. if   (posStart <0)   return   CString();
     $ [6 }" U. h0 O& f/ e& X
107. 
     / [+ c* t1 j, s4 G5 F! T
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     0 w3 M1 u2 H; ^6 f
110. 
     : @% q9 s  @# f+ l+ M) k" ?
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     - B. H% ^3 E0 _7 a6 B
115. :   m_version(1)
     ( j+ {1 R" B# d+ c, @
116. {
     ) A: s' `4 X. a
117. m_uLocalIP   =   0;
     ) ~+ K. A; L' L; b6 @2 L8 L
118. isSearched   =   false;
     
119. }
     7 r8 X6 J1 B8 E# u0 A2 j
120. 
     
121. MyUPnP::~MyUPnP()
     8 B' m$ @, h6 \! ~; H$ L# |) R
122. {
     : O$ S4 Q8 G/ U% t% E
123. UPNPNAT_MAPPING   search;
     ' [8 m9 h2 K- V; W3 t% |: O2 y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ! v. e: _+ _: \4 V) ]- t0 O- h
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     * Q% x: o6 B  s3 Z' E. X
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     6 p1 ]% N! R! M( s
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     # X! _4 k  ~* i" Z  j
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     $ w5 n1 B0 A1 |" U+ W& D1 I+ ]
141. {UPNPPORTMAP1,   _T( "service ")},
     6 W# p7 v: H# h( V6 i6 N  y, p
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ! K1 t; g9 U  \  S6 D
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
       t! Y, g7 i) y: X% I
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     ! i4 q6 y: Q* o( N+ `! J
149. 
     
150. int   rlen   =   0;
     + {) y+ d5 u  U) k
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     : L6 \' t* d) ~0 {0 A' o7 `: L$ O
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     & }$ u) o0 b( ?
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ( [! h1 R+ f$ m) k# j2 M
155. CString   request;
     ; C+ r8 P- l+ ?# M9 f" d( k
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);
     " D7 S3 |8 P/ b  K
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     . n1 U4 p% ]* w4 R/ H0 L
159. }
     / O, m3 w% X$ k' f
160. }
     . j) a- \! T' l: \4 ^) R- P+ Q! I
161. 
     
162. Sleep(10);
     
163. 
     & e: s% J- J$ p
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     7 `: e3 t& o" ?' z8 {
166. if   (rlen   <=   0)   continue;
     $ E. E! L3 E9 V  Q. E1 Q# C2 y
167. closesocket(s);
     6 \, p/ x# I5 C
168. 
     9 F5 i6 r( G' U
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     4 X% D7 U( f7 p( }5 o- l" }6 T
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     1 T% }" \/ N7 \! v+ ^0 T: @+ h
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     6 ^8 w% w9 I; u
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     % t0 ~5 ^9 d" ~2 t& g: X% k
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     4 A2 P) _3 m0 C! _/ ^9 V0 W
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     1 E, I6 {1 r$ j! M/ ?" p, s/ }# O
181. if   (name   ==   _T( "LOCATION: "))   {
     - p" l0 ]9 v6 t: D4 ?1 V( d5 u
182. line.Delete(0,   9);
     / X& @* x9 c. @# U. o5 ]1 M
183. m_description   =   line;
     & [/ ?" z/ j9 Z, `$ ^0 R7 F0 C5 d9 Q
184. m_description.Trim();
     + h0 i+ q9 C; s' Y# B5 n
185. return   GetDescription();
     : c2 B  ~, a: C4 X( V  ^+ A
186. }
     ! Y/ e* r" D2 W) H
187. }
     6 o  G% R. n8 j. ]; r
188. }
     4 M" w: l7 {/ J% c
189. }
     
190. }
     
191. closesocket(s);
     . N% Z2 e+ w) I6 t( e6 I  o
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
4 y: q* z0 l! m

9 o" |4 N' A% f5 V6 v3 s% I2 {$ y///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

. E: f5 q8 R1 ~% Q* m  ]
7 O* {1 z/ F$ m1 B# |9 X#pragma once
#include <exception>
% j( ]/ i. H" q( Z# |  J+ K) l' o
1 N% `5 t$ W) p! S9 P. u" V" a
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
3 G( j, b+ E: N6 r4 n+ Q* G, I};

0 x7 u0 S9 k( C5 o4 X
8 X( T! a* X: ~6 Tenum UPNP_IMPLEMENTATION{
9 q, Y, B$ U( U3 Q6 ^3 {( d        UPNP_IMPL_WINDOWSERVICE = 0,
- o2 M' s) E% O# M2 P2 X        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
+ N7 ]" U$ K9 b
7 e  W% l' R! W" g2 ~

# D" t* l1 u) O' k" Z
* m2 W# w, b9 {' K) ]class CUPnPImpl
  T" q: A! V% f! T/ M{
' ]: B0 }2 J% e% _public:
        CUPnPImpl();
% J# v$ Y' U: K7 `. `& q! I2 r        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
; R7 l) N% Z3 ]9 m                UPNP_OK,
' x" @7 S. ?6 d                UPNP_FAILED,
                UPNP_TIMEOUT
; B) b4 ^* R2 L; c# B; w        };
. F' i3 [% x' ~' t/ h2 y
; L0 N. _* s' }( u+ S" |. _& A1 \
' J: ]  L" y7 z  u! J. |) m, E        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
2 y- g8 M+ E) ~7 A/ M        virtual bool        CheckAndRefresh() = 0;
3 a  ?) m$ c# L- J/ }# [7 Y) J, Z        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
( Z  M: K  n, C4 E        virtual bool        IsReady() = 0;
2 \( i% B% c9 P4 t, f        virtual int                GetImplementationID() = 0;
. V# I2 o4 [& T6 D! ?       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
- y( W, X+ f: @6 w! j/ H: X$ E

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
2 r% I; n3 P/ T1 I* E8 {        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
) s# w/ q1 j2 R& s% |) ^) |/ t        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

  O5 s" V" E) z$ c, L, a
. w% e/ j  ^; I6 j5 z" `// Implementation
protected:
- C/ S0 N1 t) y( q# c" u9 {        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
% U9 k& ^4 b3 Z        uint16                                m_nUDPPort;
& \5 E0 k# ]$ a( S        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
3 K/ w# w( `0 i        bool                                m_bCheckAndRefresh;


8 A" e! i: \7 P, Cprivate:
7 B  K. X. A+ Z1 a: h( [9 f1 W        HWND        m_hResultMessageWindow;
: m1 ]/ Y. a' D; N( c) L$ v        UINT        m_nResultMessageID;
% ^& G. |$ s6 t8 n- I  \) a

# K2 p- n$ T1 ?* E};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
7 A6 A0 j# t9 T6 K5 B6 _$ ~4 ]4 A{
" Z+ M' ~  I; n+ {; D, O- z- ppublic:
0 h: I& n) w( A' k9 W) X        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
% _# u7 w" @: V7 E; s: U/ j        virtual bool        IsReady()                                                                                                { return false; }
$ L  \2 Q% P1 [$ q) ~3 Q        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
0 w- n$ X6 d$ k# ^2 X# f* W1 l};

! z7 T) I" G2 C; a
/////////////////////////////////////
, h: }" l( H7 c6 Q2 _& M//下面是使用windows操作系统自带的UPNP功能的子类
) q- H: P5 _# ]) o- N
' w7 p9 w& Z9 \. ~
" ]% Y) [  L' P: E! A6 N) {#pragma once
* e2 \# M# V; M7 W- |9 c$ x: l#pragma warning( disable: 4355 )
: A9 ]" j* u" y  A
: j& K) g# z0 L% R3 Q- e1 Y! A
# A1 m* y0 x5 u. u" D4 o6 f#include "UPnPImpl.h"
3 k& ?1 C; ]5 o" E% A1 z#include <upnp.h>
4 a4 O7 n* k6 h" ]3 J" ^  L  q1 ~& U7 E" s#include <iphlpapi.h>
: k) [3 E+ Q  P! ~" j8 f' w#include <comdef.h>
& s9 E9 T' a' x* B* W1 A#include <winsvc.h>

* p* B9 l0 s5 P2 o5 m0 ~' c0 m
#include <vector>
2 J! I! t8 z( t3 m, @' f2 g#include <exception>
% q1 _& i" e. M" ^  k; L#include <functional>
0 W$ C) `7 p6 ^8 o- F
! B! H% I& s4 o* Y: R


; }: X; R' a3 p. N& u2 q& G) qtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
3 A7 l8 O+ _  A; E5 V6 \/ vtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
9 j4 L% d! J$ ^6 o  x- qtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
: h! K' l! m: `$ |typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 _- C7 A! y1 Vtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

1 C9 m6 B! X/ U4 @6 H/ ^( |6 ?
typedef DWORD (WINAPI* TGetBestInterface) (
% U  B( N0 y% d0 |  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);
  D- s7 d, W; u2 P4 z

typedef DWORD (WINAPI* TGetIpAddrTable) (
. o1 f% n5 b  _/ R  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
( v+ u6 Y& g9 V7 D3 {  BOOL bOrder
);

: ?) q! t/ m/ k4 ]( s
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
& ~& o+ `  t+ q+ y9 }# Z  });
9 [0 n9 {3 f  ?/ x

( t/ K8 p3 a) |; D2 j9 ]CString translateUPnPResult(HRESULT hr);
1 p, u/ e! `% d4 M2 PHRESULT UPnPMessage(HRESULT hr);

) j! j; q+ K3 \1 T( R4 @8 c
* e, M7 d! f  d. mclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
public:
. K5 ?0 a, w. k6 |) X3 v        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
- ^& _2 j! I+ ~8 i* ?9 Q  T3 T        virtual void        StopAsyncFind();
/ b5 D# O8 \8 C        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
2 s" Z  z  u/ i7 f# i
3 A, B: V) a0 O) E
        // 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
. d0 W) P' A5 p' M6 J9 }7 R7 c. W        virtual bool        CheckAndRefresh()                                                                                { return false; };
- ?& r* T1 [+ x6 G

protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
" l. }  ~+ T& }        bool        OnSearchComplete();
        void        Init();
$ X* ?" b9 _% }1 {$ O  ?; `
3 D6 G+ a$ O* K1 S! Y( x7 h; W
        inline bool IsAsyncFindRunning()
! r; l% q5 ^# ^' i( }        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
' {* i  \: F5 t9 s4 u& d                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
- a/ ]" [& m$ W! }  r& Q8 f0 w                MSG msg;
% \, @2 R$ z( _: W2 c7 B( D7 c# S                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
  [. L  X" c) n# ^                        TranslateMessage( &msg );
- Q0 ~6 |0 q" l/ g                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
) L5 T0 E; S/ o        }

6 W0 H3 w4 E/ n6 D9 g
+ A, t; Z: v+ o! D7 E4 F, k% |        TRISTATE                        m_bUPnPDeviceConnected;


0 ]4 w+ C& f1 H, w4 H5 Z// Implementation
0 ?: `5 w2 T, k4 e# s- `  @        // API functions
- P8 W1 V2 E0 \3 A* H- N4 W6 p2 D        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
3 {+ _2 N$ w# ]; t        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
" _7 g0 i0 U1 X7 [' x        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
* W. Y) w1 x6 W: l' {' [& p; W        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 ?- J  W( h% Q. X$ S" v" J9 O" H

6 ^, D6 W( m1 ~4 V4 H) E        TGetBestInterface                m_pfGetBestInterface;
4 [3 l5 ?9 N6 f6 u( Z: D9 w* g0 [        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

3 G: Q1 p- K; D6 t1 Z) R; g  y
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
1 O: \# d; p3 {' L$ v0 P                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
: k' v+ }2 \+ M! K6 ~                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
, m8 Q4 \' |6 Q3 |( Z: y4 t1 m5 R4 [. f

                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

- i1 E4 k7 ]9 y9 v
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
# i8 g0 @0 A& G4 M        HRESULT        MapPort(const ServicePointer& service);
4 y% I- a! y, t        void        DeleteExistingPortMappings(ServicePointer pService);
' J4 x; |: G2 {& d6 ]        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
( J% T6 k/ b& N4 b! ~( g# d" E/ i                LPCTSTR pszInArgString, CString& strResult);
7 ^& n% q; H7 v& M        void        StopUPnPService();
5 N8 D7 c/ {+ z  [! P

" g8 Q7 A1 R5 Q; d5 S        // Utility functions
# w: |* U* u1 A, F8 }; l        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
+ V; S) a7 P! y* U        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
3 t1 B" r- n3 s# o& G        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
& ]* r4 s) `: x* u& h! P5 o        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
  q  m; p4 U% W: L  F' {9 T$ l- n        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
4 q" o$ L+ Q0 Rprivate:
2 ]  v$ I' L" I0 S* d        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
# Y0 ?8 E! W$ O        FinderPointer                        m_pDeviceFinder;
" t: K  s( n4 k; B        DeviceFinderCallback        m_pDeviceFinderCallback;
8 ]) ~! B7 A/ w% I5 k, s# K        ServiceCallback                        m_pServiceCallback;
, k% E; l) V$ _% r  c4 z
4 w3 X, ?/ \+ p1 ?' h4 W2 T. ^
8 r( G; M8 Y5 M, n8 _        LONG        m_nAsyncFindHandle;
% D1 C8 W2 z0 C        bool        m_bCOM;
( L' {" @0 ~* R        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
' T& V8 U& y$ a9 t. V- W        bool        m_bADSL;                // Is the device ADSL?
: T! o8 h2 K/ F$ m( {2 l        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;
$ R' ?9 v; T9 i        bool        m_bSecondTry;
3 g5 ]( n' F% N. X2 @7 n. c        bool        m_bServiceStartedByEmule;
) q' U0 F: K* {3 R" y        bool        m_bDisableWANIPSetup;
8 _( e4 y4 ~' J) U- Y3 R. }        bool        m_bDisableWANPPPSetup;


};
: G+ n. I* \. {+ z4 }

// DeviceFinder Callback
) o* O1 e* }+ [. @: {3 a* Sclass CDeviceFinderCallback
5 [$ Q# e2 W& F9 C& M& @        : public IUPnPDeviceFinderCallback
{
7 I& @& [5 Y0 ?$ w: `: epublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
% [9 _! q! C5 P9 e9 P        { m_lRefCount = 0; }

0 |0 R' E) o% M& a& J' G
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
6 ?# l% X2 v. h- J   STDMETHODIMP_(ULONG) Release();
# U. _7 Y- f3 l- C9 k

// implementation
private:
) T3 e. M' @( z/ @8 g        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
- n7 q, U0 ]+ c. x8 \% T        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
+ R- }$ {! I8 x+ t# b! T        HRESULT __stdcall SearchComplete(LONG nFindData);

+ a- q8 n+ \! Q" ~/ s5 i" r
private:
" r% a5 W. {( s# E        CUPnPImplWinServ& m_instance;
) I  ~1 F) s! b" s        LONG m_lRefCount;
2 |& B: x( w) Y/ x3 z6 m};
& W4 i; q# g& q

8 n! Z6 E* T0 Y1 d. I$ z// Service Callback
class CServiceCallback
2 s6 Y2 w8 Y0 |2 d( c        : public IUPnPServiceCallback
+ e7 k% m; G, P6 v0 I{
+ T4 K  z" a! N5 l1 Y3 D% q! j$ N9 apublic:
6 s' T0 |( F1 Q        CServiceCallback(CUPnPImplWinServ& instance)
9 r1 L6 R! H! N/ c% ~2 w8 Q  P! {                : m_instance( instance )
        { m_lRefCount = 0; }
6 m" S" j5 ?# u# u6 |+ D   
8 s1 I; {: C+ |" H: u/ i. n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
* M* d$ ~7 S8 [9 X# X5 j) v1 s   STDMETHODIMP_(ULONG) Release();

6 A' i- u6 w+ k6 S
// implementation
* ?  I6 n, S' S$ v& Uprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
8 V4 n: V5 C: `        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
4 B8 p8 [4 e: X4 Y$ ^. f

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 Y& a8 s) o) }};
, D$ d- ]* N" F& z' g5 T1 _
' o  Y- N1 O+ E
/////////////////////////////////////////////////

4 l4 I1 |# f, k
使用时只需要使用抽象类的接口。
+ n. o8 U2 J* j: z3 K( m* c; Q' \CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
' ]- o: N) g6 t7 BCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
1 q& d! E1 B: _" n6 D# ?CUPnPImpl::DeletePorts删除端口映射.