UPnP

天云上人

/*uPnP.h*/

1. 
   $ j  {6 N; |& X. s. z( V
2. #ifndef   MYUPNP_H_
   : C  Z9 _9 Y. J5 \, c" N# i5 K
3. 
   
4. #pragma   once
   
5. 
   $ R6 n9 l* d& F3 [
6. typedef   unsigned   long   ulong;
   4 h0 l& W& D9 W4 h$ P. D! _3 L, F# ?9 c
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    4 B; ?9 G0 S4 j' ?, d; ^' [8 \
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    0 T. x" F. \2 c* k& t( y2 Q
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    $ u; O2 d! L  B2 M( ]% d, H/ ?( i
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    . S) I/ @' x! [2 g' M3 v
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    ) r3 ]: a& q, ^4 U. Z, `4 v
18. 
    5 v% }5 v# Z% R; k0 [# d3 `
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    . O+ [2 c# R' B5 o
21. UNAT_UDP //   UDP   Protocol
      Y3 Z. q( e! ~: f, ~
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    4 W/ ]- M; F4 B  I8 v
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    - w/ k  _0 O+ e' w
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    ! i* U" S' l9 S; r6 H2 z
33. 
    $ ?; }/ ~/ t; f3 W9 \. {$ Q
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    3 M  }9 C3 ~" _7 L; k2 j0 J9 K
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    5 u% ~$ |( q+ d& B& z  l4 `
39. CString GetLocalIPStr();
    ! Y1 X, X1 ^9 \" d
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    - d. H8 d2 x7 R2 x; q
42. 
    " @; k+ T7 g2 ?! d6 T
43. protected:
    
44. void InitLocalIP();
    - e0 j" \# Z$ e& B0 N6 V, Y# L
45. void SetLastError(CString   error);
    + U  l8 h8 i! i
46. 
    0 S" {' i0 [# t1 }/ Z7 f6 R
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 z# }7 `6 ^- ]9 K  H6 z* j, p7 N$ e
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ) I% s4 a2 `( z+ z8 b( {* g
52. 
    
53. bool Search(int   version=1);
    . W+ ^4 j/ r" z% C  R9 ?4 ]) T
54. bool GetDescription();
    % A! x' O% M, D# m7 I' N" o; G5 o2 V
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    ) v( A5 X- V/ D% M3 o5 @
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    1 A4 }# |/ ], Z; I; F! _8 G: ^
60. CString m_devicename;
    
61. CString m_name;
    ) g1 N) z: o  |1 k+ r7 m
62. CString m_description;
    
63. CString m_baseurl;
    - y$ N! Q- O7 n: V5 p
64. CString m_controlurl;
    
65. CString m_friendlyname;
    ! x( _; g' ?$ K/ ?7 y+ q: V- I
66. CString m_modelname;
    
67. int m_version;
    " ~+ H8 y/ {; K5 Z
68. 
    3 D% `$ ]# t$ S7 s
69. private:
    5 g$ N9 y/ o) z9 j
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    8 P" t' Z$ ]. F5 k( z
72. CString m_slocalIP;
    ! Q4 U0 f7 k' }& U
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    * u, K( y7 B: b5 M
75. 
    
76. bool isSearched;
    ' ?" @" A# k- O& u0 o
77. };
    + X, I# x( A; C, p( }6 k; P
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   8 m. ^$ p) f" X. o: v& c4 x8 R9 |
3. 
   
4. #include   "upnp.h "
   5 ?: t8 [& ]( L' n  s
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   " J7 t- U0 I0 J9 y2 W$ ^
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   2 n( i/ i/ p9 x' J
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    % H" p, Q- @+ `- T" c( l
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ! J) Q6 _  g0 N% o7 b3 Y0 \
13. static   const   int UPNPPORT   =   1900;
    " W% s0 L: z3 j3 N4 [; b3 m
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    % F7 J; z! P' t, A/ z! Z1 D
15. 
    
16. const   CString   getString(int   i)
    7 S/ i6 m( }) H. L5 K6 a& O& y- i
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    . J& Q/ z& Y' ]! c
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    2 `- `) w2 U8 i9 [2 r( ]+ Q0 H
26. {
    2 T- I( D' {/ p- y
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    / A+ k5 E  H* Q) F
28. }
    0 h1 ]- s2 |  K; m$ I
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
      g: |; E/ U1 J% J
31. {
    , A5 a8 t# l. z/ a
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    / T% o- K# s. U
34. 
    + x6 t% U# E- [: Z4 c) c. \/ `
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    ( e1 H+ k! P' Z# [+ s( X  S0 \% _
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    + t% D6 }1 K' S6 G. \. e, X3 q
40. int   length   =   sa.GetLength();
    4 G5 Y3 o5 E4 {# C! F2 x7 A: q3 M! Z; B
41. strcpy(buffer,   (const   char*)sa);
    & G% _  O) |5 s( H  p+ L6 P& U: C: g
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    % s3 q# f/ F: V
44. struct   sockaddr_in   sockaddr;
    ( v" O/ |: {. k
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    4 J4 h% R. J' o! a1 ^! U8 w
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    0 C2 k6 a& w' ~
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    ) E+ a- Y2 ?' r
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    - L: {: Y6 z% [+ N4 d" p1 u1 ^8 T
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " X4 _; U" e( b' G! M0 ~
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    6 `6 O+ [3 {/ ]; P+ W" t
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - w$ p6 c$ a% s
57. closesocket(s);
    6 a+ `6 p; X- D% \
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    3 Q+ F" A6 R% i: y- w
63. return   true;
    , d9 \, A: r' C3 k9 D
64. }
    & f) o- q/ E3 W! w0 o$ l5 `
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    4 L$ ]& }( Q' l! B) A
68. char   buffer[10240];
    
69. 
    5 b" W4 I) }9 X* t8 {
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    1 V- z# g5 v2 R8 j' W- b
72. strcpy(buffer,   (const   char*)sa);
    + ]: _  }. b' D9 }
73. 
    
74. struct   sockaddr_in   sockaddr;
    1 V. f% m# X& g4 f* y' R
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    + n9 s8 u: \/ ~. i0 O
76. sockaddr.sin_family   =   AF_INET;
    - e' Y5 m" f/ [2 \9 J
77. sockaddr.sin_port   =   htons(port);
    7 g/ k7 ~- B0 c5 F6 G( N
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    - w/ N: c- k4 b. `! J6 a' d
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    2 o+ J4 d1 j+ d. Y& {. ^. K
81. }
    ( Y# k) H) b1 p
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ! S1 U6 C9 \6 J/ i# D
84. {
    . W: s% E- _. G
85. int   pos   =   0;
    5 G1 K' @2 A/ l$ ?5 l" ^6 D
86. 
    0 ?% q; R! ?* e% B8 I- `+ i
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    / o- X" |2 l, h" F) s. l% k* w
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    8 ^4 Q' F0 S1 w' O& t+ K0 }
92. pos   =   0;
    + ?( e% B  n% T/ y  Q. J1 v5 R
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
      L7 U: x7 a& \2 I
96. return   true;
    - Z9 p) Z8 h+ X
97. }
    
98. 
    9 |9 I3 G! m5 g- D
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     6 ]# U' b' \9 n5 J1 H$ w" C
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();
     
107. 
     8 U# N7 p) k7 _( X
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     8 p0 x% k5 k& H& q+ |6 }% s4 K
110. 
     8 K5 H1 S; T( L/ d5 L
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     & ?5 g" o5 i8 g- Y- |9 [" C; K
112. }
     
113. 
     - L" P- f$ W1 b3 D1 _! {5 p) k
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     " O( p6 X' W, M% ~
116. {
     
117. m_uLocalIP   =   0;
     - `7 h' B6 ?4 H
118. isSearched   =   false;
     
119. }
     
120. 
     4 W* \/ q3 P: K$ C
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     # F& M7 ]( @& n& y' {& @9 I
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     # c- _5 k+ T9 ^. H
126. search   =   m_Mappings.GetNext(pos);
     - g5 |* h: f- R. O
127. RemoveNATPortMapping(search,   false);
     
128. }
     - ~% \$ [5 V" S2 s$ {4 _! j/ h
129. 
       H% D  @5 K" v/ f
130. m_Mappings.RemoveAll();
     8 G4 ~/ x( i/ X+ h0 A+ V# d
131. }
     8 |# t& P) V5 i( ~4 }3 B
132. 
     8 {, W2 C- i, t/ P
133. 
     0 G( b* D( L# O8 H/ e- q8 r
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     + d8 r9 ~. k- `7 r
139. #define   NUMBEROFDEVICES 2
     7 x' [2 `! c( v2 Z% f3 i& v
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     8 z% H4 \2 Q, |9 y/ i" G
144. };
       ^' O* S! r, g. p2 j' C
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     + O$ c- y$ r$ J7 u# y) ?
147. u_long   lv   =   1;
     / @) I$ [! Q5 f
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     + q5 m* a9 ]9 `' g
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     . J+ Y3 g! j2 ^* W& s& h0 t
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;
     3 @) Q) G1 A( A1 ^9 \, R9 {% Y0 p
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);
     ; ]# }3 e: {( K$ S# A
159. }
     
160. }
     
161. 
     # Y6 Y5 E6 ~0 c3 m% u
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     ) Z( H9 Y8 W$ m( \) q
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     / O) F3 R4 W$ y) T( c7 G( B
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ! L/ v; m! q' I5 R0 g0 [3 C$ @
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     , T, J0 ^" d: x' ^! o
172. 
     
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);
     . |% b" }7 C4 o* K2 i- U6 M
175. if   (result.Find(m_name)   > =   0)   {
     5 p. w6 y5 M4 i: H% I; B# M
176. for   (int   pos   =   0;;)   {
     0 @( T' S5 ?' s5 \6 r: Q
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     , G7 N, v. j% _# J( ]; F" m! Q
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     * V; v# ^  L# H$ h8 W2 S
182. line.Delete(0,   9);
     + u3 `# a6 l' R9 z6 x4 P4 P1 `
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     ) q* d8 w7 |- m( l1 \& o" R
186. }
     6 B8 W$ [3 W/ B0 Z; J
187. }
     
188. }
     
189. }
     0 g  K' @1 m5 J. Z) L
190. }
     3 a: k, s1 p- N9 d4 r: F
191. closesocket(s);
     
192. 
     ! j; ]9 B7 g) v
193. return   false;
     
194. }
     

复制代码

天云上人

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

* h+ k, p9 d' l$ s4 [
///////////////////////////////////////////
& C5 `: M  ~/ V! ]' L/ v4 P//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

% m. V" Z' t: _
#pragma once
#include <exception>
/ f$ k) y0 v+ X
  {5 z4 f2 ~9 s8 g7 ^6 s1 U6 g7 [
& U% P8 w; k3 C6 s3 b' N  enum TRISTATE{
! q1 r2 _% }6 |# m7 {* O9 \% f& C        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
  F$ i4 k0 t# l6 `  {; g' D};


enum UPNP_IMPLEMENTATION{
" D/ ~& g, T6 L* [- ]: d/ X0 S        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

% T  H1 e  W1 t
: g3 Z" B+ D& t
- t( K8 }( r' ?1 q. V) i
class CUPnPImpl
, S$ \! D, v2 n7 O- q- x% F{
. R" j5 {; D: B) gpublic:
0 r( j7 k* e! c& i6 E+ ~        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
) w$ j" v; }) _3 n. N: K! R. y                UPNP_OK,
: ^: a' m5 ~/ M; L- Y5 O                UPNP_FAILED,
: o6 Q, }! t2 Q, w0 i6 j" M                UPNP_TIMEOUT
6 a; I: o. i1 k5 `) }' ?        };
1 r; h' a- |! v; Q. \. V& t1 r3 B: t
6 Y: v% |8 s% w  g, L1 M# E0 y5 d
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
+ X2 ~5 S4 m6 }" ?        virtual bool        CheckAndRefresh() = 0;
6 P4 G8 S* T8 w% f        virtual void        StopAsyncFind() = 0;
$ ~& V1 ^- e- B* k! M# U        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
+ v# ?8 j% C# _" N3 `1 W        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
$ r0 L/ Z1 P& F        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


// Implementation
. ?: T7 e! Z5 o0 c$ T0 C3 Hprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
1 w/ E; U: O1 b1 p3 x4 M        void                                SendResultMessage();
        uint16                                m_nUDPPort;
- U" p1 S0 I3 O0 o/ a* k. Q1 [        uint16                                m_nTCPPort;
/ n; i5 A% N4 z& B        uint16                                m_nTCPWebPort;
: ]& O/ _% M; i# k1 u2 n2 W        bool                                m_bCheckAndRefresh;

! \- {1 Q* u" x5 T7 m% X5 w3 r
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


9 ]( k6 O5 o, i' Y};


( |; f# o( g6 H- e$ z' N  W// Dummy Implementation to be used when no other implementation is available
5 \/ M& C0 K* z* _  k  {class CUPnPImplNone: public CUPnPImpl
9 J; y% A3 u/ s: p{
( f/ g+ O& t4 |) ]1 npublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
0 W( v7 ^  N0 @6 G        virtual void        StopAsyncFind()                                                                                        { }
! L! P0 i5 ?& C7 h* b# s  i# e        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

: o* g) ^, I$ q2 \, i1 R
$ s4 H& _4 A# I4 r, e5 l/////////////////////////////////////
" ^$ @1 g$ H8 O6 A+ I' @! x//下面是使用windows操作系统自带的UPNP功能的子类
& U# Z! Q- Q  l% b

/ S* o/ n/ `, W#pragma once
#pragma warning( disable: 4355 )

3 g0 |- N& G9 R6 G. {; w
#include "UPnPImpl.h"
" q& X  T3 M! K- z! w) T+ y#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
#include <exception>
#include <functional>

  I, d! B8 ?  D* d8 t
! |; H" @9 |8 t+ {& ]
$ n* t( E( P6 a3 t5 U$ u
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;
: C' ~! c0 N, Wtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
( T7 v! M$ {7 Z/ Y# ~7 X: ctypedef        _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;


typedef DWORD (WINAPI* TGetBestInterface) (
3 k2 \$ X5 v( i9 r3 \8 c  IPAddr dwDestAddr,
% g( N3 c! |! U2 M2 G* r& f8 b  w  PDWORD pdwBestIfIndex
);
9 e: ?: F( {  ~( Q$ ~
' x5 I" o5 H* U+ k! h
" U. m8 J* S1 _% X: n9 xtypedef DWORD (WINAPI* TGetIpAddrTable) (
2 `8 J* Q) @0 j+ O1 l  PMIB_IPADDRTABLE pIpAddrTable,
9 s1 W. m# j$ Z7 h+ I  PULONG pdwSize,
* n5 t$ h" X% _  BOOL bOrder
: J0 l9 W% H& {5 _( |7 w" g4 P);
! L- q# w: I  i) b
& t. T4 F; X. S3 i+ X
; [" |! N( m$ F% R! [0 V: {; E! dtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
/ x4 L" p* u" b6 v+ E7 @! b);
1 ?- y/ x  H2 O8 s5 ~* D2 g

CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


class CUPnPImplWinServ: public CUPnPImpl
+ f" s* e7 l5 @5 j{
( B. x5 o1 W  V$ i0 h& y. ^        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
2 F* [$ @6 p6 C. c/ M' q// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
: {; v' J& L+ ~0 V$ D0 \
% E9 d: U3 Z- P" G! l% x
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
5 r# y2 O7 u- O0 D        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
; d0 \& l+ [, W        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
- f) a. F: a# Q3 a- d$ h; k
" f; |  M% h2 u7 y' c6 R. S
9 \3 ~6 i; P, r( n, P        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
" i" U' p+ J1 S3 G7 `+ {" e! `0 J        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
3 x" j# E8 Y! f
$ P! K8 N2 n/ M% p2 \! u/ M
/ \) k: a  v* n  S- N6 {9 o* Bprotected:
6 f: K% a! V2 c6 `0 D7 k        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
- c# X, r; u/ W* l. q% S8 C0 G        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
8 ^& p% q2 C+ j/ t- n. [) ?3 M        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
; Z, Y' {/ v: a+ B1 W6 f6 W# m6 v        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
. u" \' Y3 y. u6 d                        m_bAsyncFindRunning = false;
" M8 O" C( }4 Q+ v                }
' j# l9 p; i2 g                MSG msg;
- `& ]4 f5 D! ?0 \6 G6 H8 \! b2 D                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
6 L& B+ B: ~  g                {
& C# M. @' d7 J* ]( A% b5 U                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
- a- ^, X1 P; E1 H) ^: _* r2 S                return m_bAsyncFindRunning;
        }
+ J, ~0 [0 a/ w: O4 M  ?

0 Z3 T* u6 r2 r5 m' G6 X2 [0 g        TRISTATE                        m_bUPnPDeviceConnected;
0 `. N  Z8 O# U. j8 {$ x9 ?; G! r
6 M2 u4 Z* e( z$ c$ C  y
// Implementation
1 ?* o$ k* L3 v5 e& P        // API functions
: I4 M! V; N7 u1 Y4 g+ E' A        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
: y$ _+ L/ s+ r2 q+ a1 }        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
2 C1 D' b; {- \' L        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);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

+ W7 c+ }1 V6 x
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
2 n7 Q5 P( d. ?6 P0 s; `                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
% o- Q+ X* o! ]+ U0 J% |7 f5 ~& M                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


' [) m) A* f, z5 C                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
8 k& ?5 e- h+ q* [) n5 a        };
       
3 t) W8 n" J: r+ S, ?) H        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
/ S3 D8 X. g4 ?+ ^" x        void        StartPortMapping();
5 p4 @* k, C' ?3 E5 T: `        HRESULT        MapPort(const ServicePointer& service);
/ W( y: Y& B, x, {" g) p3 E: r        void        DeleteExistingPortMappings(ServicePointer pService);
: @2 R# S) c$ l0 K% l, F        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
! ]* _" w9 v3 C3 I7 V6 m) B        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
& d, i: n: x+ J7 }                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
5 y+ {* c1 h! F9 e" \
* E( ~+ F/ O$ R
1 z8 Y; w  H$ b. A        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
: @0 m0 n% j; G( N  N% a4 }        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
: [- j+ E: R8 J4 B" R, r$ z" p  q' G        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
2 E( m2 p# W5 j& t0 L* V. N. F5 C        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
* w/ O: u' \& @
( M" `5 Z, F4 }& j
// Private members
private:
( j- T6 B# w  K+ Z; v1 c        DWORD        m_tLastEvent;        // When the last event was received?
5 O+ i) Y6 T5 R        std::vector< DevicePointer >  m_pDevices;
8 T% ~/ Z/ S9 @$ X7 U& j        std::vector< ServicePointer > m_pServices;
& W& L% I% S+ E2 |. C% K4 ~        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


6 u" c$ I  h% L3 z        LONG        m_nAsyncFindHandle;
: t. m1 g; y7 ~8 V2 p+ [5 b        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
% ?) |: f9 {3 `5 E- }" ]; T0 m        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
' {) F* Q8 M" @! P% g2 p        bool        m_bServiceStartedByEmule;
- @/ s; K1 q8 f; G. f        bool        m_bDisableWANIPSetup;
" ?: e2 A- n3 Q        bool        m_bDisableWANPPPSetup;
$ ~% h: L$ f6 h& F, u% i9 B; L: z  c
8 u2 \) W$ v6 i: c8 `# [
};


8 _  Y* h1 p  A// DeviceFinder Callback
9 Q% e1 ?9 K; f/ ^, h" ?( fclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
+ y1 m. ]9 v, U% c3 L3 s; g        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
( c8 W. g. S. X& P# }3 u  @) ^3 x
+ g% X( o! x5 S$ n, M
, d! ?: R- T$ G: c2 {0 \: P. l# I   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
, m5 f! @; x6 }* U8 F
+ Q. a  }" n8 R  o1 \! E
// implementation
9 L, a( p, n1 J- D. ]# Sprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
& L9 n9 G2 U" l- m        HRESULT __stdcall SearchComplete(LONG nFindData);
3 N4 F- O6 j# L: y
. U8 ~: a3 A3 I
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


// Service Callback
class CServiceCallback
- k( R/ O0 t! @/ d6 {. G        : public IUPnPServiceCallback
+ s  l4 k7 H5 P: c; o8 |9 ?{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
( W$ M% R. p3 S                : m_instance( instance )
        { m_lRefCount = 0; }
0 B' [; [( E/ [& S$ u+ S   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
1 y$ a  D: C) i  M) P& i   STDMETHODIMP_(ULONG) Release();
' ^: _& v, O) Y! t9 Z

! @8 y2 G! M! Y, x5 M// implementation
5 t  E) k$ \7 f% B5 yprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
  K- j% s  \4 z/ I* A        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 @- Q+ M% \* ^
6 p' l( Z2 @0 ?private:
2 R  y1 p5 c5 ?7 ?$ n5 A4 y0 B; {        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
% t: W3 t" _0 ?/ ^" k. y  @1 ?8 `};
4 Y7 ~5 t* K$ M; e3 W4 O( Q. V9 h

0 k" e) ~& N+ {1 Z  u! V! H/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
6 ?9 u$ j+ y& _8 P4 q; ~CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.