UPnP

天云上人

/*uPnP.h*/

1. 
   , @, U# k" k* H
2. #ifndef   MYUPNP_H_
   # j1 n/ I$ Y' J) {+ S
3. 
   " ?. N4 H; P8 |) W# z( b  P; F
4. #pragma   once
   ' ~& L7 T" E: j+ ^
5. 
   # Q, Y3 C/ f3 p5 o- ^( S
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    9 H1 V. v& k5 }2 F. k
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    4 A8 i+ p- T2 K8 `! d& S
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( b) O8 u; K. s) h
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    / Q% E- ?1 a+ s) ~  F
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    5 k! x% B( c# H
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    $ v/ `" ]; q3 _+ g# E
18. 
    * _- \5 E7 B: E5 l: e  O
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : \; H! j% ^5 Y+ {* \- x
21. UNAT_UDP //   UDP   Protocol
    : H& G7 n. j3 d8 S8 V" G
22. }   UPNPNAT_PROTOCOL;
    - x1 ?0 t" Z6 p
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
      @- `  H! q. r9 t# t- U
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    7 q( D& `8 [$ {
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    % h+ \9 F& ?1 X! l9 U$ n; O  S/ Z" n7 b
36. void   clearNATPortMapping();
    7 s7 D; r0 m2 B
37. 
    
38. CString GetLastError();
    9 o" e+ [$ s5 ]2 I& g9 @& i$ \
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    9 P3 H9 c+ i  ]4 }( R, f$ f
41. bool IsLANIP(WORD   nIP);
    
42. 
    4 |; y3 Y% z* A; U: \5 Z" D
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    4 Q; q; d' _# [2 I' _! K
46. 
    + X. t; Q7 o+ ], n/ c5 c$ ^
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    5 T( n! A4 i; M, v6 z- P& H. }
49. bool   deletePortmap(int   eport,   const   CString&   type);
    $ D/ {+ Z! q, X$ i7 A/ }
50. 
    ) H  [/ B. N8 z& D2 h* ?
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ! I. b* z/ a% ^" }; w1 z, ?7 V
52. 
    6 ^. g0 S. u% @4 R- a
53. bool Search(int   version=1);
    * J# Z" K/ L# u3 B: ]$ q
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    2 `, F+ M; G# x$ N2 D  d/ N- Z1 y
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    , G  z- Z4 O+ D* W1 @
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    , P% V8 v" ~( b' ]/ |% \
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    5 M& D6 H0 a8 X
62. CString m_description;
    3 {! `4 Q6 X8 y- c1 l
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    7 Y; `) Y/ B8 G( t! R0 X
66. CString m_modelname;
    
67. int m_version;
    " A2 z" Z) E* C( }
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    6 B5 t: G+ e6 B
71. 
    # }8 U8 M- p& T/ d+ k8 A& w
72. CString m_slocalIP;
    ! k) Y. Q; y9 c, I
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    4 Y: e8 n! h+ e) ]- u
75. 
    7 V5 z& t* \8 y" _) C
76. bool isSearched;
    
77. };
    2 _+ \9 b3 _! p4 J; `, |; h1 q+ v
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   & h+ v1 H0 z8 G3 o
4. #include   "upnp.h "
   7 z8 [9 m4 i* o- V
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   9 v; x- m2 a' j
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   , Q- D) g& u" h# ~/ _0 f% e0 O, D, T, H
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    ! [& d5 A2 G2 z4 {- s
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ( V1 {; _' j3 H
15. 
    
16. const   CString   getString(int   i)
    
17. {
    3 r- l8 `9 ?& y# z9 L4 B
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    ! R4 F9 }/ G# ?' U
23. }
    * |( d) K( q: Y8 B7 Z, ~& `
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    4 q2 [6 H' }) Z/ V6 k; a
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    6 f+ u9 v+ |) A
31. {
    & o" Z# {$ w) b( g+ G
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ' L6 e4 n/ N8 Y/ m! M* q
34. 
    & r* ]9 s# D" I% P% k  W$ M) E
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    % l) |& \6 b' T- E( L4 C
37. char   buffer[10240];
    7 G3 E8 |6 v, i8 L$ L4 b
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    9 I- m  L. v, A% i6 Q: [
41. strcpy(buffer,   (const   char*)sa);
    9 u) d; E" n8 q- Y, w) Y
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    & R8 \0 q+ G1 }+ R. j4 \
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
      y! ?' ]8 L0 V# o& ?* V
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    9 f/ e+ H7 T9 W- g6 n0 f
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
      |, r* t" F3 ~9 h+ r
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    & n; e4 W. b- @: ?
57. closesocket(s);
    2 K' L1 f% x0 n# M2 O% {
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    * T* Z8 Q" t9 ^( I) W8 s8 V' m
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    % z& v1 Z- N$ Q. J
62. 
    " @: r' @# N1 c
63. return   true;
    * i+ M( C+ j: m. _% P
64. }
    
65. 
    3 F& E% U8 @' @5 v' D+ a- Y
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    2 I  F9 u9 u8 s4 d( S; G" H$ E; e
68. char   buffer[10240];
    ) n4 Q3 B" o8 G4 C+ k
69. 
    
70. const   CStringA   sa(request);
    / h) h: {2 u& ~2 n& K
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    1 z1 X/ O4 f  v" r2 Q+ N7 E7 K
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ r( u. N$ O+ u5 O8 d" C. l, W
76. sockaddr.sin_family   =   AF_INET;
    ) F7 {6 x0 O3 I5 Q6 M; ~5 C
77. sockaddr.sin_port   =   htons(port);
    ; L( h. D% s- U/ u% c) @% n: Y
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 Y9 h# e- [: f- m
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    : q& X; x' {1 E& g/ d
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
      `1 }3 J0 t4 B$ [3 }: f8 H2 |. ^0 ?
84. {
    
85. int   pos   =   0;
    
86. 
    ! S. D/ l2 \8 v5 H7 @" @+ J: E( r
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    6 a+ @" }1 V- u/ X; t& o9 b
90. result.Delete(0,   pos);
    3 U& |3 k  p9 B! L9 L2 ~
91. 
    % G0 ]. O* H6 X5 a4 ?/ _( E( t
92. pos   =   0;
    1 K- C3 p$ z3 t4 ]2 V3 D8 P6 o  P
93. status.Tokenize(_T( "   "),   pos);
    9 `- B1 o3 g6 }6 i
94. status   =   status.Tokenize(_T( "   "),   pos);
    & }; h6 x, j% v) @( ?  p
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    3 V" @. ?+ f( b; n. m
97. }
    & |* u% {1 ^8 D1 p* r, P
98. 
    9 w3 C, q0 {' m
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     : F7 j. A& _- \3 d8 {  j  O5 P
101. CString   startTag   =   ' < '   +   name   +   '> ';
     ' s8 ~; ~; |, |7 y7 |. h1 f% ~# d
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     3 _) L3 T1 M7 P' r6 M& N) ?6 J
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     . l$ Q, N, g3 \- ~
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     2 R! Y2 p6 s9 Y  I: B' u
110. 
     % r2 B2 x/ A; R* I
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     6 x1 A: w4 X) k8 N, u; I
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     ' k2 k, E- o% N: B
117. m_uLocalIP   =   0;
     ) d9 S5 A, Y4 J3 `& V
118. isSearched   =   false;
     9 @! H8 f; j$ w
119. }
     ) l7 F* w. u8 c( m) y) A& S
120. 
     8 }5 F% I: k3 X
121. MyUPnP::~MyUPnP()
     5 k7 A- D4 D" X# r+ P$ T  e
122. {
     5 [  E1 v2 ^% I8 }; D/ ~
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     " ?1 \8 q+ f5 v
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
       W# E3 l/ z' B# M% {9 J
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     6 E2 N4 c8 h( [0 _; E/ o; s
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     2 h- r( u4 E6 {5 `0 X" {2 t' V
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ! s' c7 v" t8 v' z# i$ |4 R( t
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     4 J/ M5 U* b$ _. f7 i/ x
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
       L5 s3 x2 e: _; _
149. 
     - N4 h, u3 w& a+ z$ u
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ( r& L2 t9 P: e5 e% `, [
152. if   (!(i%100))   {
     $ H# z. @6 s6 S: V0 o& G
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
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 "),
     7 J* P( ]+ _( U) B* A0 F6 m: m, Y
157. 6,   m_name);
     9 G. t, q! f- u9 c6 w7 r: G
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ; V+ [: s4 Z; `9 @7 T3 P3 k
159. }
     ! y0 T- N, p+ ]# m# \. q
160. }
     : {* J" W( O/ S& l  @& i$ A
161. 
     
162. Sleep(10);
     9 }# A! Q& }# l% `- c* S
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     5 I1 c0 ?/ P$ q& u; e1 p
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     - H6 d: e+ f9 l5 j$ E. h
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     , |8 l: L) j0 ~8 L: y7 D' V
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);
     8 {: g( I4 V' M* H- L
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     : U$ Q6 T5 W) t
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ( |. k* U8 `8 Z* o2 p0 z, t9 B
180. name.MakeUpper();
     5 u2 L% L3 K" x- P9 M
181. if   (name   ==   _T( "LOCATION: "))   {
       l' @, S9 k8 {# i
182. line.Delete(0,   9);
       {$ _6 z0 r3 t5 h8 K) F
183. m_description   =   line;
     
184. m_description.Trim();
     - l+ R# Q) n0 O: }1 J( r  N, x  D
185. return   GetDescription();
     
186. }
     2 B" _  k( m! ^
187. }
       C7 m! I6 @# m6 |8 ^# o" z5 X6 K
188. }
     
189. }
     
190. }
     ' \2 h' [1 L+ k2 p6 {$ {+ n2 u
191. closesocket(s);
     
192. 
     
193. return   false;
     ' u/ t7 d$ i' V
194. }
     

复制代码

天云上人

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


/ o6 M. F; y- v# ?///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

* o! r! E0 C! Z
#pragma once
; _0 l2 p3 A# N, T5 n3 g#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
5 A$ \& x7 G+ i) U: [# O" E; _7 Q        TRIS_UNKNOWN,
        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
' n( r; u! i$ q2 f, U0 L3 }/ I        UPNP_IMPL_WINDOWSERVICE = 0,
" Z, k, ^# T+ F* j- i3 R        UPNP_IMPL_MINIUPNPLIB,
& U7 L  x6 L7 x: \: `1 H$ d        UPNP_IMPL_NONE /*last*/
9 C4 j. D# _% G- I};
3 W: R5 h% ~- A. \5 N9 k3 c; S
& Z3 X' ?7 r! |. T7 q! g- Z

' C$ P, s( ^& l" A: W5 g( b  \
! b/ @+ B) k% J9 Fclass CUPnPImpl
/ @* V! _1 m( _* E0 N+ R9 C7 A, A{
public:
        CUPnPImpl();
+ E, k- V- O7 ?9 C4 N% t        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
: C9 ?* u* a: `% D. f                UPNP_OK,
* H9 ]! B& U0 p# X. l/ A! F) y2 d! p3 q                UPNP_FAILED,
# |, a' f9 S- u! w                UPNP_TIMEOUT
        };
: ?9 D3 O+ @) M2 C$ \* O" ]

$ f% Y% H8 L$ j: W" C" [# J/ f        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
! u2 I2 _, \" ?, [# Z        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
7 _: B* J( p) A# D( W. b        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


( q/ e0 n: D, w% a5 A+ E, J; r        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
$ P& c( }: v; Z4 X6 {        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
$ ]+ m5 ~8 [- L  i7 y0 o        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
. V+ [$ S/ |! ~+ b        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
/ h. m! x* X, T  B+ ]0 B
6 {9 E9 I$ i4 y1 y) l& e! C$ c
( t" b/ n, c4 o// Implementation
protected:
2 O2 N1 @; X1 c$ ~% k1 J# e3 K! E% m! `        volatile TRISTATE        m_bUPnPPortsForwarded;
* U  i- W* m) q# G; R        void                                SendResultMessage();
        uint16                                m_nUDPPort;
+ {5 b7 O6 @% s# I$ Y) p        uint16                                m_nTCPPort;
$ p# U9 H3 ^0 h. a/ w        uint16                                m_nTCPWebPort;
% X3 o. r8 D& N7 Y% R  m& J/ U        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
, O5 M  C4 l  \/ x$ l        UINT        m_nResultMessageID;

! r5 X8 `: ^8 s
6 U5 X2 m+ u9 I$ P% j};
$ [$ G2 G3 E+ x& [; P

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
1 D+ {1 m" B" [  r- B{
public:
* r; M5 B: L& m# h$ m4 L1 g        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
; S. ]" g8 V& n3 Y  s5 b* L  g( l        virtual void        StopAsyncFind()                                                                                        { }
3 H. v  o+ L/ ?) k, }        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
. Q. w; Z2 F. p5 r' t* \' W        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
1 S* P0 v  Z% e4 H4 }& C};
# q% ^. ^9 D! I6 Z3 J  O
% V' H$ Z/ W4 Q
( K0 @9 E* C" ?2 y/ j! _/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

& e3 \3 ~4 c/ Z! S( ~' L
- X: W! D: k' E0 V- N. j+ W#pragma once
8 g0 x& R: G9 k! o. ]' }) M#pragma warning( disable: 4355 )
0 Z4 ?0 v$ R- f8 g( U9 v
' E. M+ z& G8 m- w9 W* A
#include "UPnPImpl.h"
#include <upnp.h>
( N2 ^: X& g* j5 T8 i4 z! Y#include <iphlpapi.h>
( x9 H" E4 x8 B( o1 l#include <comdef.h>
#include <winsvc.h>
! U. A3 H! d6 _! d1 r% T* x

#include <vector>
( t( o% y* C6 w#include <exception>
#include <functional>

/ _% i8 O7 o7 [6 f: c4 X( ?


3 `) [  v8 ~! O1 ?typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
  T: P, v- \, @) r+ Ttypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
( e+ E$ q% v8 W8 }! e# Otypedef _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;
+ b# P5 m4 c- Htypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
. Y# {, [/ X2 _, _2 c# a# g* Y: v; `2 U  PDWORD pdwBestIfIndex
$ o0 c0 B7 d. R; o% p' O* @);

1 Z6 n4 b- N$ D3 }* a
typedef DWORD (WINAPI* TGetIpAddrTable) (
% Q0 l. h. X2 W  PMIB_IPADDRTABLE pIpAddrTable,
0 u. W8 P$ N6 R- o  PULONG pdwSize,
  BOOL bOrder
);
3 H; l' I" A8 h  j3 T; q

. x& P* h+ `' \. J  o2 Etypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
% v+ ?  J4 x. C) u8 h

CString translateUPnPResult(HRESULT hr);
: C+ C5 o6 {4 v5 Y; O3 s( B$ BHRESULT UPnPMessage(HRESULT hr);

# L: i& f( N4 w8 H
class CUPnPImplWinServ: public CUPnPImpl
8 ~8 U5 J2 v3 x) e7 H{
( G" @, ?7 q; i- O: s+ j% l  k5 a        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
' F) C4 m1 H/ Z. dpublic:
5 Y: I0 v: U9 u# x4 j        virtual ~CUPnPImplWinServ();
  A1 {- f$ s% _0 Z1 Y. W        CUPnPImplWinServ();
" g& e, Y* D* n- q/ n& I$ L- `
/ k7 A8 Y' [% \" M1 n* K
. T1 z" L: Q( B, |* M        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

: @& N4 _; A! s0 W3 k- m
        // 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
2 ]: G2 J* `! x* U% c        virtual bool        CheckAndRefresh()                                                                                { return false; };
% O1 z, D- l6 V
' x. s4 l4 B+ u0 s1 K$ S5 C
; i$ y- I4 d, a8 jprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
- n4 m: ^- N) c4 Y$ \/ E        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
' s, M0 D- p/ Q% ]; n

4 R: ^) Z0 _- F1 p7 s( N& u        inline bool IsAsyncFindRunning()
2 Q) B  F5 Y! O. w( m        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
; D. w4 N( G9 Q) V8 z# b                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
, x1 q- X# z5 [% J) X% u                }
0 y- g4 f3 H5 h$ ~5 b                return m_bAsyncFindRunning;
  N/ b# ?5 o1 d1 q, {& A        }
/ H& d3 s9 \4 P1 ^) b3 X. v0 r, h

; {+ d; E7 `; A3 c9 p, P        TRISTATE                        m_bUPnPDeviceConnected;

" m0 u3 u4 {+ x3 i: G
// Implementation
7 U4 p$ b* O4 B  D        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
/ ~( z2 G6 \  o9 L% s$ |+ E        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
- p' o! {( G' q+ y        TGetIfEntry                                m_pfGetIfEntry;


- @: W3 M% ]; e        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
; Z* C' p: P7 c, p) M                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
- Y5 l& k6 A" P                result_type operator()(argument_type device) const
                {
: b4 }6 A! B8 V* P1 m6 d6 v                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

: T% d: v  v+ l% E% @) Y
) |8 X+ y7 D7 X' T: @                        if ( FAILED( hr ) )
9 T7 [; d/ ~( v' p1 C6 ?' A                                return UPnPMessage( hr ), false;

7 l) H  X7 P$ ?- F5 w8 {
# L7 T) C8 j; h- Y8 s, E# p                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" O& u( D. r- O8 J, ~                }
                CComBSTR m_udn;
        };
/ n6 j/ |! T: o! B       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
+ O/ N6 S: `% a0 u6 |6 z5 H        HRESULT        GetDeviceServices(DevicePointer pDevice);
9 d9 Y. C. x  U. _- g        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
0 ^1 b/ m) r+ {+ l) ?! q# Z        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
, |0 E; w9 t9 M  ^% ~        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* v7 ?5 r% `3 j* l/ k9 m6 R7 S                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
3 v* F2 m5 o% h2 o5 y
/ i- T( E% B; D7 j  F
$ f  Y( n3 [5 j- y8 A        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
0 h6 }' R, n' I0 D/ ~, i1 D        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
) i/ D7 q% l0 |8 T; j1 P        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
* N% Q. x4 C; D% V        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
9 [0 \; t; ]& N) z        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
! \8 w0 E1 A  M/ W

// Private members
4 ?4 z* W) @5 v5 Jprivate:
        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;
( n: M( E- P- l2 ^& ^        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
; ~9 R4 f# {/ M7 \8 P        bool        m_bCOM;
# ^2 d) e0 n, j; k$ I9 y& B        bool        m_bPortIsFree;
! _( u& H6 A* q  ]        CString m_sLocalIP;
; n! a' J0 ?5 d/ P- H' ?        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
# ]& Q% G5 V7 d" J4 T        HMODULE m_hADVAPI32_DLL;
; n# x; ?* R6 z        HMODULE        m_hIPHLPAPI_DLL;
; G& K9 f0 q/ q5 i2 \/ ~! v1 g1 U        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
: T0 r' A* T5 `6 Y; R+ k        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


5 s% P# h. k, o) A1 r8 {};

8 f, V& i# D. M, B8 D
// DeviceFinder Callback
class CDeviceFinderCallback
, F  K- Z7 k( _8 [- w5 T9 l( _        : public IUPnPDeviceFinderCallback
{
public:
$ _" G9 D  O7 f. H0 t+ C        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }

- s/ l% I- L, V/ K
6 m& W/ j4 l: g+ D9 p+ Z( Y   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
2 M' j' V8 [  K' w+ C2 Y   STDMETHODIMP_(ULONG) Release();
, R2 @, n! c8 O, K5 A1 b

& M9 l& }& L* J- E4 R. @+ p// implementation
private:
# {! ^  |( Y4 K' b4 C) o        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
4 {( N, v* X* i0 {        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

: M/ L. x3 V  c% L$ g
1 M8 `$ P% v0 U! Yprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
& f" d) C& C- M0 }/ ^/ ^6 h};
' j8 A0 U( g+ W0 P3 N' g1 ~1 \
( ]0 D! h( X6 V
1 E% q4 N7 j2 n: a// Service Callback
" ]. P- T- J" V5 C4 L' a+ d8 mclass CServiceCallback
6 L7 E) L) O. g  K8 C8 d% }4 l7 I        : public IUPnPServiceCallback
9 e' `0 _/ L# i5 D{
public:
. z7 j: j$ f, f        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
. D7 X0 p# r4 y9 z1 @7 h        { m_lRefCount = 0; }
. }; N. I* ~+ w' v   
4 y* S! g- d7 T   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
" U3 Z$ g& d$ _( a0 \8 A   STDMETHODIMP_(ULONG) Release();
" F) K: x$ |& k# p
5 N, F" K6 M8 U6 A7 _
0 g2 o8 U! }' p( Q9 Q// implementation
) U" i; s9 z& \7 L7 Nprivate:
& g4 R$ R1 \0 w        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
; q! Y2 k3 m8 f! u        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
1 U5 s) ]  P5 S. g$ [6 L

private:
        CUPnPImplWinServ& m_instance;
* N5 w5 y- x7 b$ ]* |( l        LONG m_lRefCount;
4 w) \# D' h7 t1 r. f};

9 f* Q/ [! Y! j6 ?$ |7 Y
# M8 a6 V1 x( ~. s2 B- ^0 Q4 c/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
/ P/ a+ G6 f' w* ^, s* I6 I) |CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
2 C5 `, X% e8 E3 d& D' OCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.