UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   5 K) b/ l* e; b8 \
3. 
   9 b4 a% |! u1 y+ p0 i
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   9 I# M4 M4 f, G+ B7 C! _! r
8. class   MyUPnP
   2 M/ ?! m0 Z0 w0 ?+ i' Z4 T
9. {
   
10. public:
    
11. typedef   enum{
    & u3 n# a: b  J( h$ I" J/ G
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    9 C" C2 y. S5 P9 V9 s& W/ E
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
    ) s' @" Q7 F- D* N4 z9 F7 r
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    / f! g( Q  l" m- y8 H
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    " ]; p1 a% G  p/ e6 ^* v: T3 @
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ) n8 j& o4 [7 N3 v8 v
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    7 k1 W: R8 G- u# ?& t. K
30. 
    6 x0 F1 k$ G# D
31. MyUPnP();
    
32. ~MyUPnP();
    4 `$ H7 ?+ l3 D5 I# ~/ s8 }4 ?/ ?* \
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    6 d( r: K* m2 ^9 l/ d) g
38. CString GetLastError();
    - o5 k. o! F( v: D) q1 A1 u( \$ ]- F
39. CString GetLocalIPStr();
    7 `% `+ x! Y( R- |5 V) D' @4 D
40. WORD GetLocalIP();
    % p3 g& R1 x! f/ v3 S
41. bool IsLANIP(WORD   nIP);
    ( N% \- ?5 x2 m# g. e
42. 
      K$ o4 [- }: S9 |& S
43. protected:
    2 Y! r/ q  m0 W
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    ' O2 y5 v6 d! K" |* g9 X( {
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);
    5 l$ C' _- z# x! \
50. 
    - P8 o, r2 \7 v' v& J
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      s+ L0 H1 Z: t5 ^% D4 b
52. 
    ) c6 ]% I5 h- P
53. bool Search(int   version=1);
    " u" V% {; D1 E. ^& ?
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    $ Z( ^. C) M# s5 a4 Q6 U4 }) {
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
      J3 E4 q$ h5 ]% p
57. 
    $ @9 U; }3 |5 {* v/ v
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    4 u6 {# Y5 N  g6 m
59. bool InternalSearch(int   version);
    - I' |9 ^/ a  o- F$ W/ R
60. CString m_devicename;
    & p- D8 Y1 t8 \' d
61. CString m_name;
    
62. CString m_description;
    $ m) O3 ], ^% l! w0 ^
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    / @1 l" F5 o+ z
67. int m_version;
    ! C: g" ]/ W8 h( {! m& X3 K# u
68. 
    
69. private:
    8 c- m, z1 b+ d9 v! M" v  [0 `6 V
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    % M+ Y" G( d; [) L  b+ e2 m; e
72. CString m_slocalIP;
    9 R' L. U5 u9 R: d  e; C, r* ?
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    9 h8 b4 y% v3 k
75. 
    : G2 g  r) c% A' k
76. bool isSearched;
    ) d5 T. G% B. J) W: w
77. };
    & ]6 S* B8 g+ \! Y( b
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   * ~- P1 h# m8 _, r' y( k
3. 
   
4. #include   "upnp.h "
   
5. 
   3 G  Q! c% o" ^2 K
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   / M$ D  b+ w5 d7 h  m
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ) _& |$ n9 O- k7 |1 ^
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
      _+ t, o+ Q; y  G0 l/ L
11. 
    ( m4 s( [4 G3 x) {4 U) O5 L
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    7 N3 r: Z+ }2 `2 g' V9 j* R
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    % `) |/ w8 j% F  R  e
16. const   CString   getString(int   i)
    * P# f+ I  ?% G* D* ]8 j; w
17. {
    , v+ y8 y7 |( p2 V2 {
18. CString   s;
    ( Y$ D9 H* _1 ~: M4 a
19. 
    3 R1 n5 y1 ?6 a4 e6 w7 y
20. s.Format(_T( "%d "),   i);
    3 k6 C$ Q. j# `5 {  J
21. 
    
22. return   s;
    
23. }
    5 u+ ~( T; a! l+ m2 H9 t
24. 
    ; H( ]4 M6 P! a) [
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ' O6 l( x8 C  U5 D# ?1 A( d- k
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    . t1 R2 W. |: l6 V
29. 
    & T/ i; ^0 f+ P2 Q- b) W2 ^
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    8 |0 D( s. S' p, {3 b$ B
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    5 k( r; K8 |" N) N5 e2 t# b
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    4 W+ I) b: [- G% ?+ @& G: @8 Q
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    - [) N- g( A6 a
42. 
    " D$ z( _4 ?/ s
43. uint32   ip   =   inet_addr(CStringA(addr));
    2 L4 J4 ^* ~9 H
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;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    * r  ^) W2 W" F" r% H3 Q
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    7 O0 Q4 h$ ^2 m- q+ B
57. closesocket(s);
    2 [; Z9 L# c' ?' [2 i4 _
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    0 {. R- l& T  D7 I$ }+ V
59. if   (!rlen)   return   false;
    " G* r" H0 t* {9 {2 m0 g! T6 d" F
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    
65. 
    ' `- ^5 }0 {/ Z% v2 d+ C
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + Q  T3 `% _+ z% r! q) R: s
67. {
    
68. char   buffer[10240];
    
69. 
    5 p  _/ u- K" F6 h6 U3 b# V/ c
70. const   CStringA   sa(request);
    # W8 J' `, `& Y1 q4 s
71. int   length   =   sa.GetLength();
    " _  D' i6 ?+ U# {* s. o
72. strcpy(buffer,   (const   char*)sa);
    ' S! W7 W9 \! E
73. 
    7 T) \( `- r& O3 `) b0 A
74. struct   sockaddr_in   sockaddr;
    " A8 J$ u* i) p; t: E. i
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    - W: I9 X/ C; B# q, i
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    5 k$ x, A, ?3 L/ U
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    / m+ {7 m# r. M6 A+ v0 C! D; ^9 K
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    # A& U2 \: ^9 G. i$ X% a$ e
90. result.Delete(0,   pos);
    
91. 
    ) x* z* m3 V7 w, d# B3 ^. ?
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ! ?! S- m3 d) K7 }
94. status   =   status.Tokenize(_T( "   "),   pos);
    9 v; m  }7 f1 C5 I3 O% T
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    1 r! I: M/ k8 G, i7 |$ ?
100. {
     2 o7 l# o! [4 D4 z
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();
     * a8 b- g% O8 u! i% M
107. 
     3 @6 C: Y1 ^, s2 J! H
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     . m& e5 _' Q* M4 l3 @) Z
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     4 g: Q4 q! d' U4 N" a" ?
114. MyUPnP::MyUPnP()
     7 I0 \! D- ]' J
115. :   m_version(1)
       j! K! j+ i8 }+ b2 O  @6 p+ p
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     + F- z( ~2 P7 Q+ W! o5 o
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     9 I# e+ H. [4 C) r
125. while(pos){
     6 t) n, I4 u. V+ o" Z; F* P
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     8 j0 |8 p, w" x- k, C4 ~
128. }
     3 R4 t$ _. \( c' j
129. 
     & y! H. [: Z' O4 x; k$ F7 y2 G
130. m_Mappings.RemoveAll();
     % L& P/ K* V; B* Z7 {8 K, Q
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     # g3 p6 T8 i) @  A/ e7 Z, }
135. {
     ) d$ u2 k5 ~- b" K0 J0 f2 n
136. if(version <=0)version   =   1;
     & \+ e' V+ w5 N$ X3 p
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     0 l: g0 y/ ^( r0 ^" m0 Q4 T  R
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     3 z* Q5 Z7 d' Q2 ~4 u, D$ u
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     . U) K( n9 S/ k; K0 K
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     4 J! N% P& D" e4 l- I
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     " u) r% Z9 h4 {" N- c* t
152. if   (!(i%100))   {
     $ b/ y* ~0 k+ t
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     $ K3 ?6 p6 p5 h2 S; D" h/ W
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     . E- n& E5 O" q. ~/ {
159. }
     4 z7 p- S2 e. [/ Y
160. }
     * @9 ^6 y; y) s( Q
161. 
     
162. Sleep(10);
     9 i" m- H, F) S" a  O( W2 w
163. 
     9 y6 U6 ?; ]( ~
164. char   buffer[10240];
     & L% j) b5 m; y5 F; _
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     " t' n. I$ L7 R" }: y7 G2 t( |3 ~
168. 
     4 }& V" B7 F  k9 M; c- C
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
       A, F6 c$ E# |) ~0 r7 q; h
172. 
     $ t% [4 n! K' D4 z# j
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     # A! M* |- L$ y. A( M* W! A
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     1 L6 k: J; ?; _* B, M& }
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ; A2 E+ z& w+ E: f2 ^* L
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     # Q; \0 I3 I% _# g
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     . V7 p: Z' g" _7 Q0 Y
184. m_description.Trim();
     
185. return   GetDescription();
     - I& v5 v* l9 t: ?2 B; E$ S% P) g
186. }
       T8 H( n- U4 {, [2 B
187. }
     : J& d/ ~5 r* @" P- E
188. }
     
189. }
     8 Q' v: Q6 c: s( Q8 ?( ]$ t
190. }
     6 x: Y/ a  Z, }8 p
191. closesocket(s);
     . m: Z1 x4 g* \6 V8 W
192. 
     
193. return   false;
     9 d3 m; e" O. ]6 f  Z  ^
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
) V( e7 g6 H5 P0 E( j: W& F- U' N; ]
7 }, H3 g, ^4 T0 y& X
6 e7 L$ |( k7 Y) o///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
' G. W4 a/ R  F- x& F9 \& C

$ [5 p0 z& C$ s- ^+ ~) K+ t* O0 ~#pragma once
#include <exception>
: I: t- v5 Z; ~5 N5 Q2 l
8 [  o9 l( f3 I" F% |8 v
. n: t% C( n/ Z9 n$ X( {  enum TRISTATE{
        TRIS_FALSE,
" f1 V' Q" K1 ^' r6 U; T% G  o        TRIS_UNKNOWN,
" m& s1 D- t: [1 d0 x        TRIS_TRUE
+ X3 L% p% L* @$ h};

/ I. \- W5 H) W/ u7 r
& n0 f( e( u% P7 d1 c$ ?0 Aenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
. T+ m# T( @* D        UPNP_IMPL_NONE /*last*/
* W$ O! U2 b: m) s};
# O2 u+ B/ q1 H3 I' |



class CUPnPImpl
2 B* z0 p, t! @8 O5 s{
% u: d* E. o. v6 H) p: l7 s( V6 ]public:
# `; ]# u: S: a# r# ~3 P( S1 U. Y- o6 V        CUPnPImpl();
6 d; r! q$ ~4 b7 G/ D6 g) \        virtual ~CUPnPImpl();
* S9 f) ?! e1 B- Z8 z4 z4 E        struct UPnPError : std::exception {};
. P( B  a4 S3 V2 X+ G. f$ l        enum {
  @: @- p. ^' t! F8 q% T0 ]  T7 U                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
0 V5 `* r/ E: F1 n; d        };
8 A( ?# ~$ {! @3 ~" ~0 ~. Y3 e
: u3 o, z% W0 G1 V# `8 t
# o5 x4 m/ w' P. @        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
3 s/ \9 J9 \6 n) |- c! Y        virtual void        StopAsyncFind() = 0;
, u) W; X7 m- {        virtual void        DeletePorts() = 0;
7 j6 C5 e* F, F! a3 b        virtual bool        IsReady() = 0;
6 j. q$ D/ F& f9 a0 |% t" v        virtual int                GetImplementationID() = 0;
6 |0 H" c+ \2 [5 S) l. s9 z       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
5 [! T! r2 R  b: x        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

/ q9 h  r) [4 \2 Q' a
// Implementation
- L! S5 I. j; x; \protected:
: o( t2 k( t  u  l; T8 ]        volatile TRISTATE        m_bUPnPPortsForwarded;
" b$ N( k$ z  s/ h: R, @) u        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
3 q/ \  o) M1 |' \% [        uint16                                m_nTCPWebPort;
( i* t1 V+ h, r0 E* f        bool                                m_bCheckAndRefresh;
  ]+ Z* ^. f* r& g) x/ y) o
6 p. l# Q7 r# d4 K
- k7 w7 [0 v2 Y# ?4 Sprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
9 S% V, S3 n. ^5 P% {
% O7 e- x$ i+ C+ T
; O. b8 y$ k) s  _& e};

* n1 G& A  X, I8 ?
// Dummy Implementation to be used when no other implementation is available
5 J7 E8 N4 r* O# }" _* N! \class CUPnPImplNone: public CUPnPImpl
/ a" S( e8 n' W$ }* r! k" `  i{
# Q4 J# h' B- z4 \( j7 t# Fpublic:
  G; O( {9 P3 W- A! x/ B9 U/ T        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
+ L" |, O7 y* \% j9 k) ^5 c        virtual void        StopAsyncFind()                                                                                        { }
; L. B: N, n0 U  B        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
! v0 L- \- g3 _& z        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
8 U' y2 _- G2 D4 i9 B# E};
  J- C+ K* ^! {8 K$ ^
. L& {% F; _8 f
/ j4 D/ B2 S6 @& V/ O" @1 v/////////////////////////////////////
( i6 l% @. j& g! c5 G8 {  W//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
2 g3 w+ S# I+ y7 @#pragma warning( disable: 4355 )
) Y; d, O3 s( h) v, r! \) E

( X% D' f8 l  W2 y! i4 t#include "UPnPImpl.h"
#include <upnp.h>
9 y: Y) D+ r! _5 K3 ?- ~, }#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
% i1 {4 ?5 C+ C

+ D; `; A, a) `8 u1 M% }: {#include <vector>
3 j7 J5 {1 R; M; D3 H$ l# X#include <exception>
#include <functional>

( f/ [- o, x+ f9 B9 Q


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
8 U" n: w2 }8 y$ R& ^  G* W! `7 @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;
( u/ z5 M1 Z: n- J2 F
3 a. ^5 U1 B; H
typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
/ K1 G3 W  L6 y% i9 }  PDWORD pdwBestIfIndex
);
2 v: q' L: I5 H/ Z

3 l4 A& S; V9 T3 C3 ptypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
# o1 n" H9 Y2 A2 }5 Q: d2 n4 u  PULONG pdwSize,
  BOOL bOrder
);

, f3 @0 t2 P/ Q7 X2 P; h
3 j* i$ g* `/ z; M% h. ktypedef DWORD (WINAPI* TGetIfEntry) (
( R9 o3 M  K* [0 j  PMIB_IFROW pIfRow
8 V9 O# V  K# E  e& a* i& ~# D4 ?);

" L# V7 \: L3 C- E; c3 u
" s) D8 c$ n- t) UCString translateUPnPResult(HRESULT hr);
% f. ?$ u( j* y, {3 AHRESULT UPnPMessage(HRESULT hr);
" T- }2 q" S; |- D: o* p/ H

class CUPnPImplWinServ: public CUPnPImpl
1 U! L( [; q. h# ]{
" @3 T0 s1 h! ~; y- ]! a9 y! v/ Z        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
/ A9 R6 n+ x* T! I// Construction
! x0 M% F# L3 Y3 K8 U- k: fpublic:
        virtual ~CUPnPImplWinServ();
# L2 C  A- }" W" l" s, p$ Y6 b        CUPnPImplWinServ();
3 O0 L- }2 Q1 E5 X; Z& W0 D3 n

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
1 G; x9 F$ M& R2 |# e* D        virtual void        StopAsyncFind();
1 L7 M$ l0 K' [& U! w2 r        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
# i( K- ^) B, |! q

7 v$ q# c. q+ o; M. ~" K& ^. Y        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
( g; Q7 @, ]; Q3 ~        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
4 a( @# `! ]7 k! L* ~) E: _' E. H        virtual bool        CheckAndRefresh()                                                                                { return false; };
3 l: L' ^: c# `+ o) ~! R3 H+ U) |

protected:
% p- g* L4 z& L1 N! L! {1 Q. K        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
9 {& s+ v7 n! ?' q7 T9 Y        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
2 F% h9 L! p% m6 }4 y& p/ Z6 _: T        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
; W. m7 ?8 t5 G' i3 u- v, z

        inline bool IsAsyncFindRunning()
$ P8 l8 L, d; l        {
5 b' z/ w5 z/ @                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
$ l  P' L5 U, K" {, F                {
0 c) w! L1 \# M( a  e9 `3 P                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
' X2 F. M+ C$ |6 Z2 h                        m_bAsyncFindRunning = false;
                }
' X- p0 I5 B/ @                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
9 N, y* s- ]! j+ a) v% J/ T' ^        }


7 s6 w; h* K4 @4 B1 f: m0 ^+ e        TRISTATE                        m_bUPnPDeviceConnected;
! z4 I) U; q  ]' J6 w% v2 r
9 f  j; V. b# m; S( ]
// Implementation
1 O  T2 f' r6 b+ C        // API functions
* K+ F- \* S2 X% \' p% l        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
, I- M, `/ C* ?9 m        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
& E. L. z2 H& ]        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
2 _1 N8 {: o, m        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
/ L- P. T& \& b5 E3 N, X5 d- O        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

; i2 J) Z5 ]( |* ?- S' \9 i6 C
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
& k1 ~) G' y9 |4 O
; b: p6 t, S- j! o
) W( Z4 m  F8 ^' h7 B4 O& t        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
3 y" O1 J' M* q+ H        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
) u6 F5 X4 r$ p                {
0 b/ Y5 o( ]: n: x                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
& x' Z" y) k7 k6 W. }6 |; B                                return UPnPMessage( hr ), false;
" F' k, U6 O6 ]0 A: \

3 b4 A3 v+ D+ z* @8 b                        return wcscmp( deviceName.m_str, m_udn ) == 0;
% t% C7 e0 R  z3 `; K                }
                CComBSTR m_udn;
        };
) o# r' v8 G9 F, X1 S) ~+ {       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
, C$ h/ y! {8 \; w! x1 m        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
4 w, q% |" }; ?+ E! Y        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
  L. i; ^2 G' Y' w; t- n5 `" k        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
/ v; K1 Q4 Y0 o) W; ~* M5 U% H! r2 a        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
: J1 u1 O( J& {1 I* a3 c- U
/ N8 ^: R- o, e4 \$ [0 W% P+ m
        // Utility functions
( j! o/ R# y' |0 z$ }+ k        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
8 i/ R- _: `! s9 r, }/ g1 f        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
+ K" U. f( N- U, O& J        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
9 b  U( @. u3 K2 d        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
$ {4 |9 W4 }( M% p; c1 O+ o: X( o) Y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
) O4 Q3 o* q4 q4 T  J& K9 Y        CString        GetLocalRoutableIP(ServicePointer pService);
% B# k' a; }( H) f$ n: ^* X" Y

  H9 p8 A! _- f" {9 o// Private members
private:
- ~: D7 o1 ?+ W9 W        DWORD        m_tLastEvent;        // When the last event was received?
9 Q" F& N/ s* W( {# W. u        std::vector< DevicePointer >  m_pDevices;
% |5 x! p6 u6 \" e# T4 ?0 O        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
( I. ]& K4 Z2 z, C: @        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


. {$ ~9 S. n" @% L5 y8 r4 l        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
" ?9 E( t3 }: ?, z/ v/ Q3 `: M3 ~9 V        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
3 K; P# O: Z5 f9 Z0 }, w. }; s        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
* M' @+ c( K* C! F3 C8 O' Z0 t        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
/ ]. I* a8 `/ G( l* _/ z5 [3 C        bool        m_bDisableWANPPPSetup;
* c8 b0 ^5 @. a: x' ]

};
+ l2 P% m% V9 o5 R' ?
0 ?& T  U+ f& Z+ D
6 t2 i* t- k9 @* y. ^// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
* `* y+ s6 S  r9 \# O, ]$ upublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

  X' k! E4 P+ V; v$ \
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
9 |* b' ^  u. f/ w8 e   STDMETHODIMP_(ULONG) AddRef();
0 [: ^( Z  R' `4 d6 [1 D   STDMETHODIMP_(ULONG) Release();
1 w! s( Z7 u- |, ]

9 O9 {8 k2 Q# N( e( \! ?- V* k- J# ^. ?// implementation
. d5 z; A, [; ]4 ?& v9 \, o( Aprivate:
0 e& X5 q# X1 a' e+ w  g        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
. G5 |3 t* g- i; f, {. f        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
$ h, v) b* r( h& ]        HRESULT __stdcall SearchComplete(LONG nFindData);

0 x4 U' C( A" Y! c
private:
        CUPnPImplWinServ& m_instance;
4 m4 t! I7 |$ e! R7 P$ t: x        LONG m_lRefCount;
, V0 i5 L% e/ r7 L9 A, G' e: [- Y};

2 N# K) [) K/ M, G( ^6 G  T
1 i) R) w2 @8 s( U& o9 E: U7 k// Service Callback
class CServiceCallback
0 B5 g6 C- U! N5 A% x        : public IUPnPServiceCallback
{
' V% }' y; w9 e+ y- C3 `2 lpublic:
$ j5 _0 x9 C& a$ [        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
9 }+ q5 ]! H6 D1 n5 S' E8 Y) Y        { m_lRefCount = 0; }
   
' |; r) v" v- _, i0 M5 y" c! q   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
) Y6 @$ o$ x6 C! l, u8 p( C   STDMETHODIMP_(ULONG) AddRef();
# T: Q% T, D3 o4 v' s. n! B   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
5 G* W0 n0 N0 r        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

3 Z% l. `3 {) ~* ?& r, l
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

4 g( `: P0 p) S" y9 {. e
9 {& p5 W; `* w4 V/////////////////////////////////////////////////

; c5 H" i2 U0 I. r% y- V% U
使用时只需要使用抽象类的接口。
& F, Z4 [0 Z2 m5 m& ~CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
9 W2 P* K6 V  W  SCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
0 B0 p2 q! u' Q8 E  WCUPnPImpl::DeletePorts删除端口映射.