UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   7 a! A" k; E; n/ U) ?0 D. r
5. 
   0 X- c* a, r1 I8 E6 K
6. typedef   unsigned   long   ulong;
   & s( {) C8 O/ D0 J% Z5 F6 C: B
7. 
   
8. class   MyUPnP
   5 p1 A/ S9 C  B0 _$ A
9. {
   
10. public:
    ' B$ I$ X( y' \) }9 G5 ?
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    7 r* V; q5 ]8 P. @
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    9 Z9 {3 Z; n3 c& l6 \) [
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    6 ?) J" l3 N$ g
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    6 E6 t3 P/ L4 ]$ F5 ?
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    / a5 |4 p/ ~5 z* r2 \7 X- V
17. }   UPNPNAT_RETURN;
    ! C: V. W$ f3 w
18. 
    
19. typedef   enum{
    6 }# d" V; L) |" O  z
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
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
    1 I" r4 s) k9 _5 @1 M% z2 Z; E4 q
29. }   UPNPNAT_MAPPING;
    
30. 
    2 y/ X2 S9 z( ^" ^% _
31. MyUPnP();
    
32. ~MyUPnP();
    $ G" o: F1 u4 h0 ?  a3 X
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    9 N" p' b9 j  t6 ?& E: z
37. 
    & k2 W4 L6 I  J! ], l+ m
38. CString GetLastError();
    9 u' a$ r2 R) d& I* k& ?6 ]
39. CString GetLocalIPStr();
    ' ?, m: J$ E8 G& V
40. WORD GetLocalIP();
    1 S0 q) D1 U" P0 }5 q6 |
41. bool IsLANIP(WORD   nIP);
    
42. 
    . K& n  y/ `- P3 J
43. protected:
    
44. void InitLocalIP();
    - U; ~) g0 t" V6 Q: R
45. void SetLastError(CString   error);
    4 `( ?4 I" S. U0 r
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    . C  }# x3 m9 ?' s6 I1 z
54. bool GetDescription();
    + r4 ?# `# Z* x5 v% a  Y0 z6 r6 [
55. CString GetProperty(const   CString&   name,   CString&   response);
    ) Q; y6 g5 `5 c
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    " e! Q4 q. W7 l" M+ i' c% X3 N
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    - s3 H( V. w" I# Y, I$ i
59. bool InternalSearch(int   version);
    % W2 C+ q: I/ F+ v. _
60. CString m_devicename;
    0 l* {  b; N6 U5 y* k% n
61. CString m_name;
    # Z! s- }  M. Z7 m- b
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ) B3 |# f5 I! \* q  P% I
66. CString m_modelname;
    # m* d8 i+ q0 H
67. int m_version;
    4 d. ]7 p7 Z0 H) u0 j/ |
68. 
    - y9 }' C9 Y. J& k  e
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    2 y; h. v7 q( h/ M7 r0 D! F
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    ; ?: P2 O9 `! U
75. 
    7 A$ f. z3 k, m
76. bool isSearched;
    
77. };
    / `) `3 D- B. d
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   3 x% a8 h+ K) D+ a# `7 K
2. #include   "stdafx.h "
   ( n% v6 `! O5 Y
3. 
   ' `2 Q/ K) Q, N4 H
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   1 |9 Q) ]2 n- X3 h6 ?5 _2 O
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   ' m. R  o4 f0 j: {# ?/ P4 e
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    8 Y$ [2 V/ q: k0 c" c
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    8 {+ C9 k- u, u
13. static   const   int UPNPPORT   =   1900;
    $ `8 H, f2 }- y9 g& F) X9 T8 |
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ! t8 J+ }' o& u/ @* _* g2 r& I7 o
15. 
    ' A2 Y. i7 y' O! `3 I! c
16. const   CString   getString(int   i)
    " K/ @8 Q# T( Q5 H0 h
17. {
    1 r$ o& w( z* C( `! L- ?4 P
18. CString   s;
    
19. 
    7 k  O; N  j6 m+ o. u! H# R
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    0 d! n1 v: v* N+ j4 l
23. }
    
24. 
    ' o* F  z9 ^  P8 H2 \
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    ( `! K8 S. L& C. _& d: K# n
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    4 C; M. u5 p) H: O
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    3 f& z- |6 o: M2 T, Q4 F
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    3 L6 @+ D' ^8 U/ r$ U
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    2 \7 q/ F" l9 n% a1 D' g
40. int   length   =   sa.GetLength();
    7 ~5 D' P% V/ X
41. strcpy(buffer,   (const   char*)sa);
    9 s" i# ]5 V4 Y4 w
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    8 {9 {* U) K6 _1 {6 K- h5 `
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : l8 T: w1 `0 K$ d; [
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    0 K  i& j8 {; Z( b* c2 l
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    + C! M# `0 B( g" e. E5 j. a
51. ioctlsocket(s,   FIONBIO,   &lv);
    0 v7 _& H: Z0 ^' `; |
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ) B9 E8 d3 L( [
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    6 `- T9 j. x+ [
55. Sleep(100);
    & P4 E1 D  C$ P# n1 e
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    1 K2 b& L# U* j% q
57. closesocket(s);
    3 k6 C( o" V: L
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    8 C6 ]% a2 m# t8 Q
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    0 o. O. }; s" F. b8 v
62. 
    # X/ P6 r( o3 b0 S
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    , d8 ^4 N  B$ D+ a
70. const   CStringA   sa(request);
    5 R' E) T5 V1 I0 ?0 }
71. int   length   =   sa.GetLength();
    ; L0 q; N- t! h- b
72. strcpy(buffer,   (const   char*)sa);
    1 _: `1 S3 K. }+ l- ?& ^
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    # [6 M0 I( l9 A( L2 y
76. sockaddr.sin_family   =   AF_INET;
    + s5 o  l: U7 _* ?
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
      J7 X( G2 b& i# d; i6 h0 V4 o
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    4 k7 o; E3 v& p
84. {
    
85. int   pos   =   0;
    ( y1 A8 o- D$ B9 r% s
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    & ~  r* m. Y1 E! \
89. result   =   response;
    . X9 \" |6 L8 U6 F" m
90. result.Delete(0,   pos);
    
91. 
    ! Q1 y' a, O% [3 d4 P6 f
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    5 M+ `: \6 K* B, z0 _5 d  x
94. status   =   status.Tokenize(_T( "   "),   pos);
    - q1 L3 g4 }; ]# X
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    " @' D/ p# [/ f: E
96. return   true;
    $ J/ H9 R( O9 T0 y
97. }
    # W8 Y% W! q. |. j" A) o
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    % Q' ^4 C$ ]$ G) x1 e/ i
100. {
     % ~. R0 x1 l8 p- L8 o
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     6 v$ c2 k8 F0 E. T: [3 B
103. CString   property;
     0 X+ |3 Z' x  i% n+ A. J! _
104. 
     
105. int   posStart   =   all.Find(startTag);
     * P9 r: m8 }0 L3 F- v, Z
106. if   (posStart <0)   return   CString();
     % W' |/ p. V9 w; C
107. 
     # n# I- T: A$ j1 Y8 \: p( [$ E
108. int   posEnd   =   all.Find(endTag,   posStart);
     / r6 s6 V  T" i% J3 D4 {! _
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     / ^: H# U, n' e
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     3 L6 J8 N7 ^8 o4 j# K% A  v1 N
112. }
     ! Z* U" M5 ]6 Y: [. l
113. 
     
114. MyUPnP::MyUPnP()
     & G% b, z2 V& b& l" F$ S
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     + Z; ~' ?& _) V
118. isSearched   =   false;
     . A5 ~% g) {: `! V( f
119. }
     
120. 
       a7 d- K& `" q
121. MyUPnP::~MyUPnP()
     ! ~5 D5 d( l* q3 }+ C" Y3 x
122. {
     
123. UPNPNAT_MAPPING   search;
     2 j0 E! k- H3 ]+ H0 \7 d) G4 M
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     / F( n3 a) S; H4 C6 }  S
130. m_Mappings.RemoveAll();
     : Q( l; \! L5 V) U0 I. n
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     & d, L# ?7 ~% g! ?1 \" b
135. {
     
136. if(version <=0)version   =   1;
     3 X8 v  h1 v' v4 I2 a
137. m_version   =   version;
     
138. 
     ' _( x+ n" j4 X9 j& r; F
139. #define   NUMBEROFDEVICES 2
     / W+ ^8 h& r: O# r0 Y! W
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     / |) D) k+ a" k
144. };
     7 e% x3 ~" n1 R
145. 
       J  x8 r" h( T  z
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     8 j. {( @) `4 x! `  t* F) x9 T0 w
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     . w+ t) B& T4 x% j- g9 `( \
149. 
     ' }9 [: b1 r- Y/ G7 I% }
150. int   rlen   =   0;
     * d( {# d+ h& d
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ! l7 q: ~3 ?  R
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     + ]) Q1 o4 G* e5 t2 l
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     1 a) d; @" f+ G& i& ]) b( L8 N
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);
     - E* m, o9 P' g) g! k0 \; \5 o1 G
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     # ~) L0 f$ g- U# w9 ~
159. }
     
160. }
     9 F. J7 m0 k" r' m! e
161. 
     9 D4 \! m3 U" }5 D
162. Sleep(10);
     
163. 
     4 B+ u/ t5 \5 v+ k, W! ?
164. char   buffer[10240];
     9 m6 [/ \; g" x* l8 q6 p' q' Y5 r
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     4 {, ?; w% r. N4 h% M
166. if   (rlen   <=   0)   continue;
     # s  e" M: w1 R( B. v5 t
167. closesocket(s);
     7 ]: \6 O; a* ~9 p( J
168. 
     6 h6 o7 q! b( S$ v
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     2 R4 B! t) c" h/ |9 Y
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ) k- D2 i3 l3 X
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     - `3 P* A  |2 w2 r/ ?
176. for   (int   pos   =   0;;)   {
     & ~- I( Z3 J! P1 @. o1 N  p. q
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     " H4 C1 b) I2 ~+ Z  ?( r4 g  O& p9 Z+ y
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ' Q$ N; x) F7 i! {! o+ ?# q9 Z
184. m_description.Trim();
     
185. return   GetDescription();
     . C8 W8 @9 b, R2 E) W
186. }
     % c8 W5 {2 r$ @( }( A5 s6 D- @: G
187. }
     ( [/ U, n/ Z. j9 o4 e9 f6 n" B
188. }
     
189. }
     / N- B* k9 ?- e6 X, c
190. }
     - \# {3 ~, R5 B  l7 V
191. closesocket(s);
     1 H/ O. m! p0 u& U% A4 u$ f5 a8 k
192. 
     
193. return   false;
     + N' G# v: R5 w) s( }! H0 ~
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
3 U' p8 ^; X$ x2 Q+ C4 ?* b: I- P$ X
0 k8 ]1 J  V* O* l
5 g3 W2 m: w: B) y+ S///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

, M- f3 @  M: z/ Z( Q; j
/ r2 X. ]( \' ~! ]' T#pragma once
) r% x# f# U5 }7 D2 o/ c' b#include <exception>
- v, J! A+ v- r

7 K4 v0 ^0 i3 w/ v( G- s  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
% j4 R. L0 u3 i$ T; f! T        TRIS_TRUE
};
+ V. K' Y0 M" l7 B& f; m8 u
8 V0 h; L: J" {
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
" p7 E) N- ^' y* e1 H' @        UPNP_IMPL_NONE /*last*/
8 U: _4 I6 P/ |+ a. R};




class CUPnPImpl
- f# y  H  p; W{
4 y7 X6 L) h# h0 @0 Spublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
# E' c; n3 f6 m        struct UPnPError : std::exception {};
        enum {
0 E% L& y1 I: ^2 Y                UPNP_OK,
9 Y  d# ^+ Y! e# D" v( l                UPNP_FAILED,
9 }8 e3 x) q# `2 }! N! }                UPNP_TIMEOUT
        };
3 f# _* g1 H. h

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
, G2 |- D- M2 Y# P( W- Y        virtual bool        CheckAndRefresh() = 0;
: F! g+ K. i  }4 s, L7 w4 _9 k        virtual void        StopAsyncFind() = 0;
) j$ F/ [# s7 ?! h        virtual void        DeletePorts() = 0;
/ T1 |& H, b$ N- G% ~) a# N7 E        virtual bool        IsReady() = 0;
; h/ {: Y: X" p9 s1 ~& Y        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
( V+ l! \7 ^, t6 Q* F5 ?
* [& A! v5 C1 i( r7 s2 E( ~
- ^3 a5 R3 R0 f3 w6 J' o        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
3 c0 \1 L7 a5 n2 y5 I% S7 ~        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
0 @' ]( n& C- }9 @* C
+ i# @9 j1 V) T
# K; j9 j. v: p6 [4 I( {0 b! Q// Implementation
protected:
& T9 ^1 {$ b3 o8 J& y% C# y        volatile TRISTATE        m_bUPnPPortsForwarded;
$ z( y& L' T* K( E9 |        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
0 G7 K4 a" g: ^0 ^" I+ K        uint16                                m_nTCPWebPort;
( R1 |( i/ n6 B- w& l7 B, K2 L5 Q        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
/ N; e0 i4 u, u/ {2 T

8 V; {; d* X3 w8 v5 p" `6 ~+ [};
8 l, v0 S/ E- j* }
% f& B$ N0 L' Y) j; N! s9 n9 y9 T
// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
. L2 J2 s' k* ?% w! ~{
2 E4 f: K- i/ P/ q+ xpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
; T) ~( c& R4 _! k, r0 P: I        virtual bool        CheckAndRefresh()                                                                                { return false; }
4 `" A+ D& J6 F* w        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
+ m# T& Z: ~5 n  S        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
' W, D! b& @9 d* [! ~; J

  U2 s2 M- B: {/ T/////////////////////////////////////
9 w" Y5 x: E; n; u+ E* {. ^//下面是使用windows操作系统自带的UPNP功能的子类
8 {: [4 w: A  v6 q

9 H3 }$ ?4 b/ f/ ]) I#pragma once
#pragma warning( disable: 4355 )

+ W. t. \! v" W+ L) M2 g! H3 D
#include "UPnPImpl.h"
#include <upnp.h>
# K! I  C9 M4 M#include <iphlpapi.h>
#include <comdef.h>
6 [3 w6 m" P/ Y4 N; I#include <winsvc.h>
( L5 \2 ]. O$ O: s
: Q3 Y3 t( E& [/ W) b2 [# z7 {: v
#include <vector>
( ~4 _& T& v9 A5 e#include <exception>
$ X+ e) S9 b! _- b" w/ P#include <functional>

, F  d1 Q2 \+ H

! l6 R  e' j  Q: S
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
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;
* |6 e3 ~& p" Rtypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
1 }/ l* n( B5 {+ P; k/ L
0 }7 G- Q9 r  @" W* e
. k" m- \/ ]; D; jtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
- E% s. j" l  m/ m6 W- n+ N! k  PDWORD pdwBestIfIndex
);
2 {4 L) p+ Z! U9 {( j. ~
% r3 _1 v2 f  g3 ~) m# B! ?
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
) O5 d/ I% Z1 q, I6 j  PULONG pdwSize,
/ g1 v/ Q7 b; Z! L3 m- m5 K3 c  BOOL bOrder
3 g5 n5 T' Z8 b$ z9 P2 W);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
) _" u0 Z' B, ~/ _2 |2 e2 d% {" y
8 m% _8 Q/ n; f- l, C4 U2 _! ~
CString translateUPnPResult(HRESULT hr);
/ @* S) U9 {0 e5 aHRESULT UPnPMessage(HRESULT hr);


( c7 A! B5 L$ @3 uclass CUPnPImplWinServ: public CUPnPImpl
/ t4 r3 z- P) D" O{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
, o, E( k! O9 m5 v// Construction
  M9 [* B: L( a# K- ]- lpublic:
        virtual ~CUPnPImplWinServ();
; N' c; X* c( q        CUPnPImplWinServ();
& H. Y* M; w, P4 I1 P/ ?1 ~; h

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
+ j- }# v1 m/ q5 c        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
$ F( E& Q" `) `5 ]; m; a        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

5 y  \" k( E8 `+ s% o+ l
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
) o) q# h0 ]9 e2 [& D
  h$ [- l* S3 x- H7 D& o
! `( t& b9 ?1 C3 l6 Zprotected:
3 [1 @8 p; L# T& L        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
) }" I" @1 a4 m9 N$ H0 G5 L8 f        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
* ^# G5 x9 Y% y. T$ {0 E* Q4 _        void        Init();


$ N" u/ q# e3 ?0 U) C3 z        inline bool IsAsyncFindRunning()
        {
- X% f1 H2 \% J                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
) D* b/ v6 v! s: n2 @7 B( p                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
  W9 F! ^( ?7 |8 X* d" h                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
6 d9 M% y4 S2 }- _                }
                return m_bAsyncFindRunning;
        }

: K& C6 H$ t* r  U- {5 s0 a
        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
: h4 q# U# O, y# l  j        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
0 a+ u; T" z/ x0 v  p" v) ?        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
# n) b# D: X: |+ t        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
% c* u& l$ b! F5 t        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
: _: J6 n1 b+ ~. f1 j9 y        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
- j" i6 j+ }+ m# t7 Q0 @$ v        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

1 |4 b- J+ F; B4 z3 |
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
, a* k' O4 P" a$ M9 ?& |/ O( H        TGetIfEntry                                m_pfGetIfEntry;

! `' I3 Y* k* z) v+ E
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
8 r6 h; A& c: M0 q        {
1 z( g7 i, O5 ~6 N; A( K* n- `: d                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
" T/ v. B1 S" y+ J                result_type operator()(argument_type device) const
3 l1 G! w, D) a' G6 T, z" S2 m                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

2 C# }3 V' s+ X& @9 q% J1 u3 ]  ]
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

  o# e. ~2 r/ F
3 g; Z( Z8 ^' b* X1 z7 d8 |+ I                        return wcscmp( deviceName.m_str, m_udn ) == 0;
' u# X3 j1 n  k4 A                }
                CComBSTR m_udn;
        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
1 A5 T9 }) M7 K        void        DeleteExistingPortMappings(ServicePointer pService);
4 }# X8 e1 t( S7 D        void        CreatePortMappings(ServicePointer pService);
+ O4 R) N/ J0 X+ O        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
2 S' G2 h% {: V2 `                LPCTSTR pszInArgString, CString& strResult);
1 q: Z  o4 S1 o. l' l# c; Q0 G        void        StopUPnPService();
, {& u+ c* {, y, ~- Q2 V8 E
% ]- L, `+ Z* c
        // Utility functions
. |% @) p/ q. P% Q5 U        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
* k- z6 ]: V! L2 Y9 [7 V0 s: T8 C        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
/ S- N+ l! Z# l* c9 P        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
+ B: l$ L/ F* Z        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
# C0 t) `; @1 ]0 e7 M

' l0 j& W, u3 I' Z% u& Z, ]* A// Private members
6 K, k& H& c! |  y% sprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
+ U' j, u& n5 X9 Z, v        FinderPointer                        m_pDeviceFinder;
0 \, N8 n% ^  ?" D2 J        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
; g8 F+ |2 \, i9 p* Y
6 Y. t0 O, m; i) _, t* _# m. C- C
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
. b% n9 ~; W0 s' ~7 R5 d* v6 I        bool        m_bPortIsFree;
* |0 @2 h4 \1 X# |        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
" }( g- o! t) y+ Q9 C$ }: A& R- P5 m        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;
        bool        m_bSecondTry;
/ W# `  F7 S6 [* c( r. H& c        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

& v7 T2 \1 y" C# L! f4 N
};
; P4 X0 R* m+ Q6 H- W
0 n  q5 |- ^. |
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
/ B% o- e2 X; j4 R8 l/ s{
. H1 R0 W8 P3 i6 }  Bpublic:
; r* o7 Z& {0 c8 U1 e( k        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
; V5 L. k3 M% x$ P# L' V! L        { m_lRefCount = 0; }
( {6 ]+ f! ]: \& c8 ^
2 a4 P' [% r0 x; h" n3 w4 v( Y; C
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 r5 a1 D  H# N1 b1 {- y
' ?5 Z. s+ f6 m/ o' f9 X0 o' q
9 L3 p+ K; n' H( [$ h/ ^// implementation
3 H' C- N" u3 |1 }private:
" F* G. j8 R+ g& z% B  ~  E: V        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
) U$ ~+ T7 o7 ?        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
& I, v. \. v% R- ^1 h        HRESULT __stdcall SearchComplete(LONG nFindData);
3 a. k% i( y, R( p, u$ U1 `

private:
, g' R$ i; q) N! \$ H        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
$ c! P' R* [4 n  U( v4 ~' j2 y3 C};

, _6 @3 N" o. T+ T2 k# J
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
! o5 i" A) r+ `! Z% g$ @{
* z9 p# _) c8 j" _6 e5 i. `public:
4 I* R! P! w0 v4 m        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
" ?% ~" r2 H& @" s9 Z; u. x        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. ~5 _! ?+ U6 w   STDMETHODIMP_(ULONG) AddRef();
# ]5 @% Y/ M, t7 a# Q   STDMETHODIMP_(ULONG) Release();


. E3 n3 L% i" L1 P) E/ B9 C/ |// implementation
5 s6 R) g5 [) [" v+ aprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
+ q1 z* M3 J2 z, V. ?% v; D8 Y' [        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

+ Q" c, w% j/ H2 x+ N" P; c
private:
! I& @3 a- Z& {3 _! u! K        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
) p  M# m) m* K! F6 I6 S};


/////////////////////////////////////////////////
5 d3 e; F3 B; J; ^1 [
+ h& U1 j6 a6 j6 m
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
7 F8 v3 n# g* f( GCUPnPImpl::StopAsyncFind停止设备查找.
. N$ {2 s& C9 T/ @# m& Q' oCUPnPImpl::DeletePorts删除端口映射.