UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   9 B5 p. ^$ @) ~. p2 S$ w' Q; g' R
7. 
   ) d! I* @! V; S0 Z" s
8. class   MyUPnP
   . R$ x1 H* [- c1 ~! A/ k
9. {
   6 [0 U* E5 D  k5 _# @2 S$ x# {
10. public:
    
11. typedef   enum{
    & ^/ v! _% a+ f) p& O1 H$ n
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    + F1 c0 {% Z3 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
    # y; {2 ?2 B  `5 K1 @6 K
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    5 H% a: m* w; @5 H- {9 _
18. 
    
19. typedef   enum{
    ( O( U& d1 O8 j+ @/ G6 p* ~
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    " D0 f: I/ ^. K$ a3 x' C
23. 
    * F% i$ f2 t1 R" u2 H0 w, G
24. typedef   struct{
    8 y0 B" r2 j+ [3 I
25. WORD   internalPort; //   Port   mapping   internal   port
    & h3 i  z3 F# C% ]. o) f+ w) @
26. WORD   externalPort; //   Port   mapping   external   port
      e2 z0 m$ N0 G5 \* T4 C
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    " Z/ U$ k5 J8 L" L  ^% I' z1 i
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    " ?9 m1 E; G* Z
32. ~MyUPnP();
    ! G) m1 r' F- \2 }! ?& i2 S) ~. Y7 X
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    5 h9 i9 E' Z/ c9 \' ^
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    $ `2 K# P! O  ?2 ?) ~
36. void   clearNATPortMapping();
    
37. 
    7 B$ {7 e1 e* O- w+ K( y& b
38. CString GetLastError();
    . ?  A6 y9 T9 n% w3 N
39. CString GetLocalIPStr();
    , i/ e0 h& O6 k3 i; g
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    . @: x6 z: w8 u- H9 d
42. 
    
43. protected:
    
44. void InitLocalIP();
    / E7 _4 w1 b. z- z$ J" Y( T
45. void SetLastError(CString   error);
    
46. 
    1 v" D. `6 O1 {+ j; @' K" f+ S
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    & t) m  c9 X; i- U8 s
49. bool   deletePortmap(int   eport,   const   CString&   type);
    4 X, E) X7 N% d3 \
50. 
    9 @. I/ ~* `$ a3 L5 t2 v1 m" Z+ \
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    ' d# W0 E4 }- b% R
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    $ J( E& P( \# e! C# h/ k4 `5 }
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    / {! |' W3 t3 v# t: s6 u
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    / u* \0 U1 Q0 H! W6 r
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    " w9 {' S# j2 {) `! h- o
63. CString m_baseurl;
    
64. CString m_controlurl;
    ; `; v4 {. z8 A8 W' g  J, c5 l4 D
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    * f  B. h5 E& r) j- [1 ~/ r& x1 u
69. private:
    ) I9 a  e% K* E" `3 m
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    % }& j+ ]* y1 B# f
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    $ i; |6 `" A4 U/ D6 U9 W' Z7 n. p) @; t
76. bool isSearched;
    ) s/ i5 B  p7 i; C
77. };
    : N4 h/ E5 L, l
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   & M' |; _8 I" D0 A
3. 
   / }3 {! Q" M( T, h
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   " I  d1 ]4 y! t  H
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    8 Y4 ~$ i& U; @3 @! q
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    , l. h! @" g, t( I# k
17. {
    ! K  }% N- }# s5 N
18. CString   s;
    - q4 W! J% I# F$ k/ H9 z8 e; x
19. 
    ) x4 ]$ w/ o: g) d. A9 w
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    " G- s2 a. y5 K+ D) b" a
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    $ O! o7 o0 Y! J# x) [" ]
26. {
    $ H7 Z7 w7 T3 u6 C( c
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; r" W5 b  }  V, q/ y) D! d1 m
31. {
    6 U4 s, D/ j( W: \5 C
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    + {7 z2 A" ], u# r* {7 j# H
33. }
    ! U% h6 Y2 |8 T2 D2 l
34. 
    / _. \: {% k. H/ j: N
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    4 C- c9 L) G" j( G3 }" A$ C
37. char   buffer[10240];
    : \! j; G  b" K5 L( D1 s% Z" {
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ' l9 @7 g3 d6 c$ p: x, B* T1 T
41. strcpy(buffer,   (const   char*)sa);
    9 F, c) E5 C( v  ]' x9 C
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    - ~7 m2 e/ G' |  `, R" L
44. struct   sockaddr_in   sockaddr;
    / t6 r/ H7 C& }: f
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    2 ^7 `! Z; @* [! R& Y
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ; B3 }# e' L& d- E# Z
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    1 K9 ]. `* Y, v; `
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 z% Q+ D4 W# o: d! e
53. Sleep(20);
    7 O! |) U0 Y: g, R" `; M
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ) Y4 p3 v2 j3 X6 c7 d; C
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    3 I" h8 {0 y6 C" x9 a% [
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    / i9 M# o, K' j; y( L7 T, r. ?
63. return   true;
    
64. }
    8 Q! ?6 w3 c$ {" e
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & g' y: p) h  e: G' I7 ^# a
67. {
    
68. char   buffer[10240];
    : M/ w! n. F8 Y* @
69. 
    . W. R: A; F3 s$ `
70. const   CStringA   sa(request);
    6 ^. |, t( ~9 k- `
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    2 h% {! Q$ v+ x3 i
73. 
    
74. struct   sockaddr_in   sockaddr;
    $ e+ d* ~& u5 Z$ |0 L9 t2 j
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    ) G( Y2 O* [/ a: R  b, o; C6 C6 X
76. sockaddr.sin_family   =   AF_INET;
    + H( z! ?6 N& H. C
77. sockaddr.sin_port   =   htons(port);
    4 l7 ?6 {: F7 z2 S
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & c: U" j& t$ g1 y
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    + ?, t7 J! a6 T+ N
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    ) m& [9 `; B+ u
85. int   pos   =   0;
    : X( Y) q$ \1 x
86. 
    & U1 c& `+ H' h
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ! U1 b8 n) D# [
89. result   =   response;
    
90. result.Delete(0,   pos);
    " z  g% ]9 g1 w/ c) N: @
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    , t5 e% \. y/ T, s
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    1 z( j- P1 _2 S. d' O6 x8 T3 C
97. }
    - K' D+ H4 ]- O. q( n0 N4 Y, X, q, i9 h8 \
98. 
    * S$ z* |8 ]" Z( Y: i
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    # @* H+ f# h/ T
100. {
     $ L3 G: z% X& d8 M  i
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     * c" m+ {0 m9 F0 b1 i
104. 
     ; \! V$ `2 ~1 I4 e5 }- ?; X
105. int   posStart   =   all.Find(startTag);
     4 W3 ?5 ~1 u5 G% V0 {/ Y
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     + J+ q- c- T* J- k2 _, V
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     7 Q& m3 G" n) _' H9 K+ }
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     " f) q& E5 h4 G( n/ Q  l2 \) F5 o
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     : @4 H/ q) {7 I- i' {( g( x
119. }
     6 L0 U9 i% G9 |) |/ B8 j
120. 
     
121. MyUPnP::~MyUPnP()
     * X0 M- V( A, H, \' p
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     3 j% x% A" b4 ?  {, I
129. 
     ' o7 T: z' j4 b2 Z% j4 K$ _
130. m_Mappings.RemoveAll();
     % z! g6 x* q7 Y' f
131. }
     
132. 
     6 ~2 C0 L0 }9 a
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     6 s$ m2 Z7 r' m5 Z3 X
136. if(version <=0)version   =   1;
     ; U, R/ w  U7 l: J2 ^7 p8 e
137. m_version   =   version;
     4 a  P& o6 k4 }
138. 
     5 r8 }# X5 U) j% ]( b6 @1 k: b
139. #define   NUMBEROFDEVICES 2
     " b( B5 k$ [6 P3 f' o
140. CString   devices[][2]   =   {
     9 B! b, f7 _& y9 |; x
141. {UPNPPORTMAP1,   _T( "service ")},
     / D% M% G. I4 Q, ]( k( ~
142. {UPNPPORTMAP0,   _T( "service ")},
     ' O$ K' o* r: U
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; ?/ T4 Y% L4 i- {0 s0 ]
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     6 X; s8 W2 j4 t) m" S6 _9 B
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
       p' B5 n# i$ L; y, Z
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     * P) |8 e0 @  a6 b: o
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     ; l9 i; o& }8 M5 Y
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 "),
     % U& W2 D1 ?; l7 X
157. 6,   m_name);
     % L% n4 ]0 B# V; \3 B- X0 r
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ! n4 e5 d' @, h5 n0 i+ n" I9 i) X
160. }
     , q( ?1 \8 ~7 K; N" @9 f9 {) Y1 v+ C! z
161. 
     & A1 s8 V" o! G! W" ?0 x: }5 J
162. Sleep(10);
     
163. 
     ; Q6 e2 T) {/ A$ L  s
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     6 E1 m' Z" u# W  l8 S& X
168. 
     / f0 _6 e" b  K' g2 v/ y
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     * f" L1 ~+ d' I5 c4 \, G
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     * W/ O+ V6 P6 {9 B1 d. v
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ! L3 I. j3 F  ~4 {* U9 o
178. if   (line.IsEmpty())   return   false;
     3 [; b$ B' i7 }1 l+ k
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     5 [6 M( D  p6 R
181. if   (name   ==   _T( "LOCATION: "))   {
     ' z- h+ ?1 m$ e' H% i
182. line.Delete(0,   9);
     ; E4 }: R4 m, U' ~0 Q: z
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     # x' n5 J" i5 G1 m, c% y
186. }
     
187. }
     
188. }
     
189. }
     + x- h8 ]* P' ?; ?
190. }
     
191. closesocket(s);
     
192. 
     ' C' @% h5 v  R8 Y, g: w
193. return   false;
     
194. }
     ( e$ M% N/ W( F6 N& t% o) t

复制代码

天云上人

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


- p3 A+ h5 I1 b/ P0 I7 ?$ k///////////////////////////////////////////
; S1 f! y* v" c6 _. M2 [" v//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
1 l- y# \5 A: o4 {& a* @$ u

#pragma once
#include <exception>

1 v. P, b$ m$ p! b/ U- Z; u
, a) n; H2 \- y4 V' ]9 ]( Q  enum TRISTATE{
) ]8 a& g7 ?( A% j% z) J        TRIS_FALSE,
        TRIS_UNKNOWN,
: B) y- A" t3 V$ E1 U$ H        TRIS_TRUE
* q8 }) n& `( ~; z$ C7 s% C3 M$ \4 F$ V* h};
- B9 I$ h3 ?# i7 B$ z

) d* I7 |3 y3 R6 f; M1 M( |enum UPNP_IMPLEMENTATION{
  y# \/ t; c  h5 Y  Q        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
1 Y% v! N# g; @9 g- m' n
9 {' s: a* O; B% H" N: h
8 o6 H8 Y, V% p* r: Z1 ~. |+ }
0 r3 W# M' s- n! b' J
4 ]8 T% B, i; t0 W1 Q2 Qclass CUPnPImpl
{
2 T* A3 U4 H5 L% F0 e) D! F* p! ypublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
1 g$ Q( ^$ m- p) ^5 c4 w+ V        struct UPnPError : std::exception {};
        enum {
- P9 n7 Y/ W* ~! l+ B- z                UPNP_OK,
2 _. |2 m. J% s# m7 @) @" |8 i                UPNP_FAILED,
  b  U# s, I: Q; u2 L9 \+ s7 F                UPNP_TIMEOUT
        };


4 X9 y% F2 K, p" P( v1 g        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
: ^$ w* c- G$ O) a        virtual bool        CheckAndRefresh() = 0;
$ A+ P1 x' @5 M        virtual void        StopAsyncFind() = 0;
3 y9 T: l7 B. p9 H5 }  B7 X4 |        virtual void        DeletePorts() = 0;
; B7 B7 b5 b8 u. n) [0 i9 y        virtual bool        IsReady() = 0;
) V# b& O( [8 k$ c- X: A        virtual int                GetImplementationID() = 0;
       
& x  g. b" I9 ]( N6 b  Q        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

. l/ n2 p1 ?  l+ Q
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
2 k% E" ]" b1 {1 N  ^        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
: Q- D- q' A' E0 g        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


! d. U/ I0 g7 C( T- i" \// Implementation
* Y4 ^; }3 d/ g! m+ dprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
4 d' J& T# `( {, l9 S% C/ `! t* M        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
* d+ s% O( V1 l$ u! @) d+ |+ u% y/ f7 n        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
2 \- y. l( k  c# u* q, d8 h3 b
1 t9 M. B: }3 x8 [5 f, \
private:
+ k% `# Y' n7 k# x: h; \. S+ r1 z4 P        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

6 N$ R% x: I' o+ }2 h1 F- A
" @1 E3 R  E- \) m5 A# Y};

$ Z9 V, ?: D5 d, v; h0 ^- m" ]
1 q# A% w1 j) }$ F- p8 R  z// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
/ C' k* V  r* X  }- d. Y{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
$ l& ^6 q! ~! D9 ^        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
' Z4 j$ X; V! h7 O8 n        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
* t) x3 C9 j: q8 q        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
$ {  ]4 W7 x& p
* u8 K+ q$ u  ]/ h1 j
' t  i: R1 W+ D' d, V) R( p/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

0 x% v) M8 X1 B* n
& v5 ?! {6 ]2 c& t: r#pragma once
#pragma warning( disable: 4355 )
& j; x' e5 W- W" X9 X; l( }
! I$ R4 R& @0 O
! A8 f4 b' I6 I' c" B5 g1 n, @#include "UPnPImpl.h"
" C  {; U/ f% d#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
' j/ s4 ^! M" A) |
' l2 T  a6 B) n
#include <vector>
#include <exception>
# o- ]9 r1 f  f3 L' r#include <functional>


, I! n2 _5 P$ s$ F! a* O& U7 t. R# Y

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
1 @( U" B5 v8 v6 t$ |; A6 ]! h( r1 X3 wtypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
" e0 ]; }6 a3 E$ {/ @typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
) J1 h$ h5 r  r6 Y9 }( jtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
! }: |, _- U0 Q' [% z0 ?typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
- ^% l/ ]# H" g) L
( c7 @9 @; \5 `( r/ T
2 c* m2 ~0 U) K$ N7 ltypedef DWORD (WINAPI* TGetBestInterface) (
& [' q/ \4 J+ ^) a2 g: h7 @2 f3 g5 q  IPAddr dwDestAddr,
0 C/ Z' j* x3 k# T3 p* F  PDWORD pdwBestIfIndex
7 x( K7 y3 a7 D# S( G);
' C- f4 ~2 W! t! B- W$ H

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
3 G2 A" H+ O% y1 l  BOOL bOrder
4 h9 N) H) V. ]$ g" b! A( C  n);

# C' q: i8 ?  r$ Q3 }
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


) `9 F4 ~, w0 hCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
. ]) K2 H% W! k6 ^* [
- @  u3 |( S! r/ L
class CUPnPImplWinServ: public CUPnPImpl
7 z8 p& F: i$ |/ P) k{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
" o. b0 H6 F! b) g3 mpublic:
        virtual ~CUPnPImplWinServ();
# ^* q; J/ X* h6 P; Q. v0 r, k# A        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
7 w, z( g! T3 O: s# w' r        virtual void        StopAsyncFind();
- u1 p2 c2 i/ Y1 M" U$ K. \2 X, T! j        virtual void        DeletePorts();
- s$ m0 p! @9 ?  U7 M        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

& o) X# X5 b. v! O: K
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
3 a) S9 T+ L9 j( @. A; k        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
( C" a) y) @: a: R$ X* @  q4 w
! W' J3 G/ B  o" e+ L/ v# P
- d; c- `- u+ D$ I; Jprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
* U- u( u( m1 k0 i( ^        void        RemoveDevice(CComBSTR bsUDN);
0 o) E) g* F- C7 Q* R! V        bool        OnSearchComplete();
) v$ g% x' g* W& Y! ]        void        Init();
& p. e0 B% D1 ]
: M8 _" I# O' F4 M
        inline bool IsAsyncFindRunning()
        {
& d% W) i7 e+ N. S% ]                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
$ m. o3 Q  k. V5 i4 ]) D                {
9 `) n: L( Y; x! M                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
  ?3 y* |6 [/ A$ p                        m_bAsyncFindRunning = false;
5 j  q2 k( Z& v                }
; \& |9 {- S6 E; O9 f                MSG msg;
& m4 N1 R  C4 u% N                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
" O' w8 e3 c$ W6 a& M. o" k# X                        TranslateMessage( &msg );
  P$ B8 |# w$ F# V) T                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
- K6 y- R* `- L. c0 V) k        }
) Z# |8 e  C; P# u! a! s% I
7 b  w+ n7 q' Q
& T, u# M: u0 B# f3 ~+ @) Y  e6 M2 B" z        TRISTATE                        m_bUPnPDeviceConnected;


/ Y$ G, k& W8 [// Implementation
        // 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);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
/ t, A$ c/ \0 t- g2 N6 h# a" S8 n1 C1 q2 u

1 V9 ?" I( ?/ M        TGetBestInterface                m_pfGetBestInterface;
  Q0 t& ]4 r5 D        TGetIpAddrTable                        m_pfGetIpAddrTable;
2 j, _' `$ J. y- D" z        TGetIfEntry                                m_pfGetIfEntry;
# N) z( Y3 w: @$ S
4 P) I  d# R  _: l* y9 O  n$ E
" D( z& {8 q5 j# G) ]' ?        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
4 X! M+ u6 ]9 @0 Z  s% d; q1 k) I/ O        {
. u- n( U6 _( l! _) e0 j) a                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
+ S  Z% y: D, j                        CComBSTR deviceName;
' a  Z; `4 z/ N8 S# j                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


; O. ]& ^3 i4 _3 e                        if ( FAILED( hr ) )
" J7 H9 X3 K4 V" i, E6 Q" H6 X. H                                return UPnPMessage( hr ), false;
1 `/ J0 w2 T4 L

3 E- C+ q6 t' K7 t                        return wcscmp( deviceName.m_str, m_udn ) == 0;
( [" U" M" x3 O' y8 C8 ?                }
                CComBSTR m_udn;
        };
       
( }/ D$ W5 B2 A5 X( C        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
, u: c4 l& ]+ k2 {2 ~6 l" i        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
3 _. }5 |3 P) e5 R, G: a        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
. B; Y1 y2 D' S  P        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
+ I( T& S, h/ s( z% X! M; J% I                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


9 M. y* @  A- r% C        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
& P8 I. @1 K! f5 U+ j        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# @5 w# _; ^7 @- z. B- u        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

% P* W% C; @+ \
* K9 S& y* A7 M8 D1 a" w) M3 k// Private members
private:
7 \* [& f' G, }& W4 V/ A5 R* b8 Z        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
, i7 V  x& |  \! z4 U, K2 b        std::vector< ServicePointer > m_pServices;
0 q' [, g# ~9 L# J$ b( B  d" ?        FinderPointer                        m_pDeviceFinder;
" _( c$ o* s3 }' u5 U        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;

# U" e1 W/ n9 S5 b
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
: e- g7 n" a( \& a        bool        m_bPortIsFree;
" c0 _/ t0 s# n        CString m_sLocalIP;
" ]3 T* R$ t! e7 N        CString m_sExternalIP;
/ v' q+ a6 e7 {5 d2 J+ s" ^) K" o        bool        m_bADSL;                // Is the device ADSL?
4 P$ d8 \2 K" A1 o        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  |/ O6 w- [+ O+ G, }& \$ R        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
2 r2 ?7 l- ?2 R( U        bool        m_bServiceStartedByEmule;
; J& g; y) }7 G8 b1 H# Q0 \        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

! E, E% c" s3 b& L7 @. ^  q
% x  Z: C6 `. V7 x8 F};

% \+ ~4 Q# ], p' _% |8 U
! O; |$ n6 z8 O3 l  y6 q6 |. `// DeviceFinder Callback
class CDeviceFinderCallback
. i: N5 j4 ~7 `% B  ~/ g6 r3 S* z        : public IUPnPDeviceFinderCallback
9 `8 [" T! i& }0 C6 q- D{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
; F# |3 G: d4 _6 {' k        { m_lRefCount = 0; }

3 b; u$ \2 J# x, X7 J
3 ~3 [% {, x/ Y  M& Z   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 U6 k+ n: ]/ ^% W' O2 D. n8 |   STDMETHODIMP_(ULONG) AddRef();
& o- X$ g5 Y* b. w. Y, y   STDMETHODIMP_(ULONG) Release();
+ u+ w4 ?4 I, P- v

  N# h  C  s2 W7 M  p0 @5 R// implementation
1 D5 U# G5 v, H) dprivate:
2 L; f6 b/ \# J6 `5 ]7 z5 x        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
. [4 V; T' f- Q- q        HRESULT __stdcall SearchComplete(LONG nFindData);
. _+ |) H( _4 |" A! d1 p5 r3 n7 F: J# d
; ]! t, |: M& c* o! L0 x2 z& {
4 N! i/ M, Y; t& H: {) tprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
+ m2 A9 w4 I4 i  f" |; E" `( K  e};


// Service Callback
class CServiceCallback
2 J7 {- B+ U' k$ o8 M3 K# l8 ]( b        : public IUPnPServiceCallback
' b5 u4 ^3 Q* N, R: |5 g{
public:
( t: i' r% C- l2 u" U% L        CServiceCallback(CUPnPImplWinServ& instance)
# F' k8 K7 p1 p4 x- l# r                : m_instance( instance )
        { m_lRefCount = 0; }
   
: E! v" R' ?6 M  T   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& j/ p3 k: f6 [8 }, u& X  c   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

0 F3 y3 Y; @+ W' G; G! ]2 ^8 G
// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
+ C: r* {+ n$ m* j* D8 z) q        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


' W. e2 b( O0 f0 Q, S/ U; Vprivate:
# p. H7 q# b( {: E        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
7 B% _8 }  x, o2 I" W1 ~};


3 i/ B) o$ C  z5 f; W/////////////////////////////////////////////////
, k3 N( e4 i4 c$ ^1 [8 y( }5 z

使用时只需要使用抽象类的接口。
7 @3 F9 b; n1 p( uCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
) C" f7 |: D4 _  B2 Q4 zCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.