UPnP

天云上人

/*uPnP.h*/

1. 
   9 G9 I3 l) H. v$ i, R3 I( T
2. #ifndef   MYUPNP_H_
   / e8 j# e# d, n
3. 
   2 j# c6 S4 G& b4 @. D# u1 B2 h! e
4. #pragma   once
   : U& ~0 _/ z) P. D$ p1 C
5. 
   3 X! G1 V* d8 E3 Y' Q' w
6. typedef   unsigned   long   ulong;
   
7. 
   . _. b2 \" f. s% |7 p) c7 C3 @5 B) i
8. class   MyUPnP
   " F/ C8 [; S+ Q4 Y+ @
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    9 D, x% q- ]" L+ a( {  D
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    # i9 k5 b; _( K+ j: V
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    # d+ p( i5 z2 K. E# W" O/ f
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    , R9 n, X* l7 g4 C
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    % Z5 D# @* j1 A# v! v
24. typedef   struct{
    & S" D" L( i: u) a+ C& D
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    2 v) R# y7 Q) Q. r
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    9 p, V: x) z" w0 Q
29. }   UPNPNAT_MAPPING;
    2 p/ T5 x' r4 O  H" t5 L9 Y
30. 
    - N  }) k) q. x, q
31. MyUPnP();
    / w, f6 Y& F/ ?. H
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    . c, F$ t0 L: B7 \4 a3 s: n
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    # C1 k; Z% J7 Y: M. |/ Z
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    * W9 I9 d2 b" W# u' T8 X* x
42. 
    
43. protected:
    4 O. |! o6 y' U/ ]
44. void InitLocalIP();
    0 h" y) M+ v/ P! E4 W( n3 C
45. void SetLastError(CString   error);
    
46. 
    * ?  ]" B7 K5 |" P$ i( c1 _/ ?
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ! A8 d( T1 r/ E3 ?; v" D) e
48.       const   CString&   descri,   const   CString&   type);
    1 n' y$ m1 Y# I" C* A; d# O6 w# R
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    2 H2 \% @% G! H8 e
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    3 x( n7 `% L! Y2 k0 H
53. bool Search(int   version=1);
    9 q* F& t5 A- b# z1 D
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    1 ~( P8 a  [  U) x0 G
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    # n* q; M3 \  U% U: K, z! r; j, a8 L
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    : Q7 ^+ p% g" L$ o0 z$ G
59. bool InternalSearch(int   version);
    3 W: s2 A, N6 w" ^
60. CString m_devicename;
    
61. CString m_name;
    . c$ I* f+ H, q3 A
62. CString m_description;
    
63. CString m_baseurl;
    0 k- w6 q3 @( C
64. CString m_controlurl;
    $ Y" K: [) m; p' n" J! X4 L
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    . W! Z/ T5 Q* a
68. 
    + Z8 Y- [" ]. e4 Q9 A' o4 \
69. private:
    1 {: ^0 c4 l1 @; D
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    4 D7 y1 [* G8 @
72. CString m_slocalIP;
    ' ?& t5 [" r$ ^! q7 p5 N
73. CString m_slastError;
    8 d6 ]4 x- \+ G3 }! `$ Y% o4 h
74. WORD m_uLocalIP;
    . l, ~1 K% [/ `& h( S( D$ W
75. 
    . ]2 d9 Y2 D& W
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   % e; {2 P% s3 g5 `+ R$ s, f. A
2. #include   "stdafx.h "
   0 \+ w' Q' F0 s; ^
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   $ I) y5 |' o; ~+ }
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   + ]; i& N$ H* c+ p! v. g6 C" C0 R$ K
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   1 _# X8 K1 v/ _0 L3 G$ T
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   . [& h' s" B4 ^) H  H) W
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ! p; g' u! b% B+ Y
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ' P1 P4 a; l4 s+ C7 w
13. static   const   int UPNPPORT   =   1900;
    5 A: ]$ r1 \0 b9 u3 E# `
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    
17. {
    3 a8 A5 @1 n! E9 }5 u. v4 Q! b! ?7 R
18. CString   s;
    
19. 
    * e' O: ~: L8 ?* p7 x3 M
20. s.Format(_T( "%d "),   i);
    
21. 
    3 e9 P! |- u2 Q
22. return   s;
    
23. }
    
24. 
    - Z4 n3 @, J  q% f5 z& q# M
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    # S; @* z& W' ]( x
26. {
    / T* S( Z7 ~; }( M7 t& J
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
      @% D" H4 r5 l/ X5 r* B$ n
30. const   CString   GetArgString(const   CString&   name,   int   value)
    / ?+ P- s  d, c, K
31. {
    / q1 M$ N+ k! l3 c5 X
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 h% x8 g  |# ?
33. }
    , t% j5 r5 v/ W* R6 s
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    , }' r+ _% U  [9 C6 W
36. {
    % x( R- D; [( k2 [# m
37. char   buffer[10240];
    , |* w" Z& ~* l5 r7 _% F
38. 
    & _- [6 n# m" N$ d1 r
39. const   CStringA   sa(request);
    - M! h, k& ~" l5 E& B' P1 c5 Q  V
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    " O# e! @/ Q  Q; A/ W
44. struct   sockaddr_in   sockaddr;
    : a! Y' v! L9 z: G
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : }8 m' M# w7 ], [5 j. w4 j0 M) }
46. sockaddr.sin_family   =   AF_INET;
    9 b% T9 R0 E' m$ t
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 j- A4 s1 D* O' u" p
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    5 f5 }. g" c3 i7 y8 e, B
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    " A; @8 t* T8 l/ o2 a) k' {
53. Sleep(20);
    % b* @& Y! r1 \, V9 u/ E5 b
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    * T; W# P" ~/ U, |
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ) {& g2 V, T1 B8 Q  f% h
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    2 b1 j' C" H2 v9 q* P
63. return   true;
    
64. }
    
65. 
    7 x9 d9 \+ _4 N6 b/ Y, \1 t& w
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    / k. {" r  F9 ^9 Q% B
67. {
    
68. char   buffer[10240];
    
69. 
    
70. const   CStringA   sa(request);
    ( X+ e2 t* f$ v& k
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    7 }! n# S8 i& S+ X4 N% ]
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    0 L. x! j) ]+ _( x# e
84. {
    
85. int   pos   =   0;
    
86. 
    : s! Q" C, j9 E& V2 G0 p, j
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    3 F( P; |+ q/ Q
88. 
    8 e6 d0 C0 ]' b1 A' I- U
89. result   =   response;
    6 n) f8 ~% N% P. T& `: d
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    1 @6 w4 Z6 ?( z% ^( r7 a9 H4 X( [/ B
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    , P& t: s/ t  M4 y, y
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     % }6 M, v5 B4 z0 l9 c! j
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     5 w5 h& ~1 B, I& s3 B: o
106. if   (posStart <0)   return   CString();
     . a# s. \7 N0 J1 j" s/ y5 X& t
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     1 N* X; F, G' z7 u* I& p8 u& a
109. if   (posStart> =posEnd)   return   CString();
     ( a6 |) m; A1 ]0 r6 [* H# }
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     / i3 K0 B4 h; X# D: l3 ]% S
112. }
     * l% Z+ i! o* x* W
113. 
     ) u: _* Q  z4 v4 ~7 Q2 R
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     ) @+ O" j, k& x
118. isSearched   =   false;
     
119. }
     
120. 
     % K) @' |4 C! V
121. MyUPnP::~MyUPnP()
     
122. {
     ) C8 d5 K9 ~4 c" A5 x- x6 k
123. UPNPNAT_MAPPING   search;
     - J# F; l1 b) _4 a9 z% Y8 `
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ) n; g: M8 D* K( C3 ~% o
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     + d' o; B+ K( e0 V
128. }
     
129. 
     ( m8 c& X; w; N4 }9 W  o7 ]
130. m_Mappings.RemoveAll();
     
131. }
     6 a- j  S* b0 ~- R# Q$ M' A: _
132. 
     # [. b# W1 ^8 l5 @4 B  n
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     : [3 y* E2 V* U+ O& r
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     0 }  P1 p$ `6 G0 f
139. #define   NUMBEROFDEVICES 2
     9 c5 y! ^) v( [  R0 b. o: G
140. CString   devices[][2]   =   {
     3 w/ o; T- U/ b  B/ {
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ' x# K, B: p5 A8 c" Q* }" T& I" ~
144. };
     
145. 
     ! p  s* I- q" {4 t3 b! @
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     / w9 T& k4 s8 ?
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     3 r  d# I) s& @& k0 G1 O
150. int   rlen   =   0;
     3 t+ B; \) p( y! c
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ' j3 _" H6 \* C* ~& [
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     / Z' [5 a" ?( ^6 m
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 "),
     - W) r) u) t& {9 d, x) Q# Q- G2 m
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     7 u8 j$ C! i4 M; P
159. }
     + {$ p/ h( k5 F5 ?% B3 W
160. }
     
161. 
     
162. Sleep(10);
     . N3 o& S- p% o/ n4 w4 ?- `' ^: V
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ) B# E4 y! h, h$ S  ~1 D1 p8 i. Z
168. 
     8 o2 h$ A" E  G" P0 Q: k
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . @6 l1 N' t3 d* h( H  Q( F, T
172. 
     3 [  U* z' ]8 }' ?7 U
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);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     5 Q9 J, K& z( C5 i4 L2 R
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     3 y. F2 v, j- t0 |
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     " W* I( _* H' |
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     : O/ _6 l$ s6 F6 Y8 w7 i: F
183. m_description   =   line;
     
184. m_description.Trim();
     7 a8 L! W' T. j1 E' s! o
185. return   GetDescription();
     + R" Q5 P- {2 y1 G6 r, `
186. }
     
187. }
     
188. }
     ! N8 l& a0 m+ n
189. }
     & j5 a: Q/ A* a$ \9 |: j3 V
190. }
     / D8 B" R/ q1 V7 b4 M% x
191. closesocket(s);
     
192. 
     4 w$ c( ^- L9 G( Y0 ^- z
193. return   false;
     
194. }
     

复制代码

天云上人

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


+ x$ X  D  O1 ^0 {# q+ B/ ?///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

( P5 L! F' b4 j+ g  B0 v" R
#pragma once
* _( T! k8 p8 [0 o( x( h2 Z% v8 G#include <exception>
1 L& V/ Y5 c& o) w( e* J- z
7 d! ?7 z/ r, M5 {" @
* e  F! C5 M. P+ s; D  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
$ I" X3 ]7 l7 F' e, n        TRIS_TRUE
};

# E! C1 S5 u, s  _( T- D0 E
" z6 ~: z+ h$ k* eenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
- r& c% `9 U/ N9 ?        UPNP_IMPL_NONE /*last*/
) c: M+ T6 L/ n( Q+ ]) H};

$ f+ C  S: C: f2 M2 }- B7 K' O$ Q
. h! W+ h5 v) R. E4 c
( w0 z) @( ]9 @" ~4 {
# b) ^' x0 g' e4 O1 U  o  J) kclass CUPnPImpl
{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
  N5 U  f. G" G& \, w. l        };

/ O6 j% f0 v) f" d! p
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
% t$ F! t; y! o$ M6 b6 k: l. Y& ~        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
* m% |0 p+ T! C; w) G. {        virtual bool        IsReady() = 0;
/ p3 O6 z: d- M9 ^' n' I  x        virtual int                GetImplementationID() = 0;
3 p7 V' ~9 h  O% N6 i( }       
" l3 W0 m8 z& D( B8 v        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

6 r: a# Y7 w0 R# H# b7 \
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
& y1 s6 p6 A  [9 z+ }$ H        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
! K5 O( I( q6 D6 w: h

// Implementation
- I& _( o, r; F/ U6 eprotected:
/ w2 N, \/ G6 f) y, W# {9 f        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
/ O* W2 }0 N  G7 o        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
$ k8 }, D8 Z0 q, S        uint16                                m_nTCPWebPort;
7 [" X: @3 u6 B$ C: Y( y        bool                                m_bCheckAndRefresh;
+ o! [& e8 y0 H2 s. P: y

( b2 }0 Q' R# N' Oprivate:
        HWND        m_hResultMessageWindow;
  C' v. i6 F/ F& i7 Y        UINT        m_nResultMessageID;

! x8 X1 G7 z% |" @: a
1 A/ l& F, X. W" c4 _- j( Y};
2 V+ o3 n# |  Y0 E4 m4 Z
# T& _' }. m8 o
" t  M1 J2 a& ~, S// Dummy Implementation to be used when no other implementation is available
( p2 i6 M4 d$ g: o! [0 `  Tclass CUPnPImplNone: public CUPnPImpl
. k! J) f& s/ u, [1 t{
6 e3 [% i& N" T( D0 ppublic:
8 o# t! }) l5 u' ~: x        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
2 `6 I# y8 C( i* g/ @1 H6 p0 b0 o        virtual bool        IsReady()                                                                                                { return false; }
9 U& V3 y& ?6 `" _        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
; t/ Y, v! V7 u# ]};
1 `" q1 `# a  `" f. F
( n9 L* `8 N3 a2 F+ L1 E
. A- M6 G* a4 e4 k) u( \/////////////////////////////////////
6 ^' i- d/ o6 @$ l//下面是使用windows操作系统自带的UPNP功能的子类


2 W5 Z2 Y3 F. ?* D$ o6 x#pragma once
- p3 t: E9 x- Q$ i! c#pragma warning( disable: 4355 )
; @4 I1 B, q7 O: `7 ^8 F/ ?
" E! {( J8 E$ h, C) c6 J
$ a# v% c9 T  B# ~+ Y1 b#include "UPnPImpl.h"
8 b) R& i* G) T# j4 B#include <upnp.h>
( x' k5 n' p3 h2 l8 X5 \% {0 Y#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
: }, ~/ Y% Z. b( q4 ?) y' E

#include <vector>
#include <exception>
#include <functional>
1 J7 Q$ J4 |* f8 R
3 d5 T7 @, Q$ G8 K  S
2 b$ F  Z+ b8 @4 G
- }3 @- C# l# W! K3 W
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
$ m# {1 i+ I% @% J8 btypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
  M: @. I( _/ J% J1 i" c: dtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
! n, U6 O8 W* [- ^5 y$ F; N" Ptypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
% {1 o4 C% Q: \' p' W- v: J! K2 C' |0 ~typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

9 t! g% w# g. k
5 Q% w9 h$ G; Q$ l* ntypedef DWORD (WINAPI* TGetBestInterface) (
6 u" p5 h9 F7 S6 q  j6 k8 t9 T  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
' r3 n0 {8 X' }% r7 |);

& e1 @3 H. c5 C
1 }( j+ k5 l/ q8 a( r) X, W$ U9 gtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
; T* ~6 ]/ t$ p3 u- s  BOOL bOrder
0 O3 g1 h& e. p" I& {' o* m- m9 k);
- b" A. L2 E7 E* [
; J- l8 h& e, t) n
typedef DWORD (WINAPI* TGetIfEntry) (
+ K3 B/ j  m) T' `0 p. C  PMIB_IFROW pIfRow
);
6 \2 H3 a2 s1 }! {; v6 D' v' ]
' T% P6 Y" M( C5 ]% ^, N
CString translateUPnPResult(HRESULT hr);
6 R% H# p+ Q: y/ _# ^; kHRESULT UPnPMessage(HRESULT hr);

8 A5 k& o5 {+ c) S- s1 Y1 ~
9 V& p  \1 o& ]class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
2 F8 }: l. s9 f% P9 cpublic:
6 n  b9 C* l; M( w        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
9 @' E" J1 V2 A; s6 U( D! J        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

" C' I  D5 F( C0 d/ }
3 P6 l* @; M% V6 S        // 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; };
6 W9 N, _  f' j3 C
8 f+ y! b, Q% S
. G5 W" V7 I& y- lprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
! D3 M( x* Q& U& g; j3 s# E        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


' D: R1 [( t% v. b0 V3 G/ G        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
5 H4 o2 N- M' Q( A                        m_bAsyncFindRunning = false;
                }
  M& j0 n0 c, S) p                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
8 u) _; m! }2 [% `1 h' c" x) N4 ^8 X                        DispatchMessage( &msg );
                }
+ D! A! A  V3 b; I3 T; `                return m_bAsyncFindRunning;
        }

+ y$ m7 N' R  v0 U
        TRISTATE                        m_bUPnPDeviceConnected;

% K; `/ x9 O3 I
4 V4 R- ?1 i( C* a- R// Implementation
% E& ~3 m/ a+ j3 V        // API functions
2 D1 q, `- d8 N, c% E$ g" S        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
$ Q# \( y0 L7 N' c8 ^! C1 g' P/ I# O        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
1 F" r: A7 B: W5 n2 B4 Y1 ?2 R        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
$ ^5 C- W1 h% a; F- n5 H5 `        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


6 `1 i$ n1 J9 u3 X        TGetBestInterface                m_pfGetBestInterface;
+ T- h0 T5 R* c        TGetIpAddrTable                        m_pfGetIpAddrTable;
% g8 `( M6 K( q2 \. |        TGetIfEntry                                m_pfGetIfEntry;

4 K" [- \# \; C# G0 ~& K
. H/ w! \% e/ n2 C        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
. @; }' v6 }" }                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

' ~( r$ x4 F$ y& C) d* Q
, x9 H5 O: h# A, H) P                        if ( FAILED( hr ) )
* P6 [4 A9 L8 X# C                                return UPnPMessage( hr ), false;


                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
6 V  @; A2 J1 z: V+ X8 I6 Q                CComBSTR m_udn;
- x4 V) {4 m  p% a; L& h: `        };
( ]9 Y- r& @* ~2 F7 `" t( J0 v       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
, r7 i* v% G* F7 _- s/ |        HRESULT        MapPort(const ServicePointer& service);
9 ?  Z5 p5 t0 D0 \$ b7 E        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
  q1 q; E8 R0 U5 r1 z& l- R/ |  p0 w        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
0 X) V4 d' F7 T6 x, k  f# h5 P

0 K8 A& t  Z7 z/ R        // Utility functions
6 {8 P; s/ ]5 F1 [        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
  O& c: Y( F- s" c; @2 Q        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
; N7 ?  R2 L/ J. z        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
& |  @  m5 f2 C1 f        CString        GetLocalRoutableIP(ServicePointer pService);


; r0 I( H4 A3 {// Private members
3 `3 X4 E- s0 `- {7 N% B; _private:
! D4 k0 G( r: D        DWORD        m_tLastEvent;        // When the last event was received?
) |3 Y* g1 ]) P. a        std::vector< DevicePointer >  m_pDevices;
4 ^% T1 w. p1 z4 e        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
1 E$ T4 |9 J6 M+ i6 O        ServiceCallback                        m_pServiceCallback;

! t$ y1 e# f* n  o$ q
" n% }; j' T1 [        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
  N- q+ }" F' b- a4 c. b4 p4 S        bool        m_bPortIsFree;
        CString m_sLocalIP;
1 {, c% ]; Z4 ^- t        CString m_sExternalIP;
) A6 t- [3 |% D2 ]! D        bool        m_bADSL;                // Is the device ADSL?
+ {0 \, ^: {5 y9 |7 U        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  W2 l0 m# _9 Z9 @* \        bool        m_bInited;
        bool        m_bAsyncFindRunning;
, }, g- t0 @5 J' Y" D9 `        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
7 f% r2 V- V; x. t3 u4 h: S/ t% b        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
! @# H+ \" k- c: N0 a
; W8 c1 K7 u1 S- w, Z% p9 P
) K2 W7 F" U0 Z5 a- V};
% ~% A9 @) z2 L) ~& k
; Z, \! X& F! Z$ T9 W! d4 B
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
0 H5 ?* s5 C' w) c  u{
3 d; L, M" Z# Npublic:
8 T" J! d3 s' q' N5 ^0 N) D! H        CDeviceFinderCallback(CUPnPImplWinServ& instance)
0 x/ E6 T! W8 n. I& a3 F, m                : m_instance( instance )
        { m_lRefCount = 0; }
0 F6 V, b3 }- h$ H. w

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
4 w6 Y) q& Q$ z- g: o+ \: `   STDMETHODIMP_(ULONG) Release();

% r) U$ u& B& j" e2 S
// implementation
7 T: M/ ^1 {9 uprivate:
# I7 q* O& n5 j; Q2 K        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
. ^4 H- E1 H! D' x        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
' q8 v' E& }# k6 u' r( x        HRESULT __stdcall SearchComplete(LONG nFindData);
: s. \  D+ ^4 B- _: _# j

' c" V! {- e" R/ }; M+ ]* {private:
1 A- W, p( q+ k( J, X        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
' @8 V& A; Y8 `% @* T2 c3 t5 C! |8 d4 ?

// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
1 T8 W7 H& a  @/ a4 x7 N. tpublic:
; z( \" X5 E/ j' }: B& q5 M5 m0 B4 \; t        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
  o' @4 `5 X+ ~. G5 u        { m_lRefCount = 0; }
   
' k( o  l) A2 s& F$ S3 U! \   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
2 u/ `1 @) F1 a

// implementation
# j3 c9 X( c$ }( ~private:
8 e% I5 b0 D+ O        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
& f* B; x) h+ x1 H2 V# S9 f        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
0 ^, }# U  D  o; E. l& {  A        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
, W# M) z; A9 q9 ~# B};
$ l6 a4 Y, w  j) k; f, H0 y' w

7 u9 ^7 m+ D4 [. x' `; f& ~6 [( \1 I% H! Y/////////////////////////////////////////////////

( T2 ?0 G$ H- Q# Y6 B2 g
使用时只需要使用抽象类的接口。
( `" c2 D7 f5 S9 r) d0 W9 PCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
6 }# V* ]& w2 ^) A" G0 R  G& x7 O0 aCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
- I6 |1 K# v5 f) Q# M" jCUPnPImpl::DeletePorts删除端口映射.