UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   ( ?9 n4 z" C; k
5. 
   
6. typedef   unsigned   long   ulong;
   , A' S" E. I8 z
7. 
   5 d' h, X; q: \( v
8. class   MyUPnP
   
9. {
   ; u; g) R- H& X, V" a! D
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    - d0 C: ]1 i0 r) H+ ^
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
    7 l7 D1 l" M9 k6 S2 E9 g& O; g
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    6 M4 E; @' O0 }' J$ D7 z
17. }   UPNPNAT_RETURN;
    0 o/ v$ Z; \* d! V
18. 
    / {$ _$ ^( J. x- _1 q7 ?! w
19. typedef   enum{
    3 N3 _# |, o& I" Q# B/ Z: M; K
20. UNAT_TCP, //   TCP   Protocol
    8 o  [# s8 m- V* e2 i  o; ?
21. UNAT_UDP //   UDP   Protocol
    ! R* r! R* Q, \' G( H
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ! |* b) ?2 q( @+ `" L- k
24. typedef   struct{
    8 S9 p6 K+ b* j
25. WORD   internalPort; //   Port   mapping   internal   port
    & E, {# D* J8 H* E2 R( F3 ^9 Y  g/ G
26. WORD   externalPort; //   Port   mapping   external   port
    ; E0 b& o( N# l- l3 t8 \! M
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ' T$ k9 K& D$ f) Y
29. }   UPNPNAT_MAPPING;
    
30. 
    3 \" e* d$ t8 Q5 L4 m
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    % `' K5 O& G' V; [% ~. r
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    $ n; S# ~3 L+ z9 ^- y0 q, e0 N2 U
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    + h( |  a; w0 }% D$ f+ w: a
40. WORD GetLocalIP();
    ( b+ K8 G& @5 g5 s. I' U( W' n/ H
41. bool IsLANIP(WORD   nIP);
    1 a$ \3 ?8 l' l. y# y5 Y
42. 
    
43. protected:
    6 A3 r2 ^1 y, c. H8 q1 I
44. void InitLocalIP();
    1 S; e4 L6 \+ Y7 ?  _5 h& m
45. void SetLastError(CString   error);
    
46. 
    9 p2 R" X% s- _, s: O) F( [/ ?
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    + f6 }3 c% H2 {4 f. T; s
48.       const   CString&   descri,   const   CString&   type);
    " r4 {, X6 Q: E# P
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    9 C  V$ }- K- R5 U8 v/ X9 \
52. 
    6 D. J/ q: ~. p$ G  |. \
53. bool Search(int   version=1);
    - A$ _$ U0 [) [
54. bool GetDescription();
    . _; {1 Q$ C- Y: Q6 d
55. CString GetProperty(const   CString&   name,   CString&   response);
    4 G6 L  u7 c# [8 i1 j* X5 y
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    8 Y  t+ z7 K" t5 p( ~# y
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    3 I! ]! a. I# D2 Q+ }1 `" I/ Z
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    ! K: r$ b1 ]( F. x
62. CString m_description;
    $ j% C$ a/ C2 Z1 Z4 h
63. CString m_baseurl;
    
64. CString m_controlurl;
    , Y9 j& v) L4 o$ V2 s9 h; W
65. CString m_friendlyname;
    
66. CString m_modelname;
    % y5 |) @# E; x! ^6 W. i
67. int m_version;
    6 w- n8 W2 Z8 p! F) s/ W5 f% t7 e
68. 
    * D% m- _7 C0 ~# w0 `. v8 w
69. private:
    6 {" N5 W1 @6 ^0 A
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    / b: o$ T* a/ ?( u% |0 A1 V
71. 
    : ]" R4 _; R( R$ j
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    : t- n/ {+ k& c
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ! \5 m6 O+ k% N- y
3. 
   9 h; k1 e7 H: v2 J! u0 w0 O9 w
4. #include   "upnp.h "
   # B2 o% y1 R" d; u1 A# a0 f: p. O
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   7 m; S: E9 Z  Q0 g9 ^2 P: G
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   / S6 E" q: G4 c* i9 X. j
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ; u& |" m3 ~0 [2 U4 u6 H
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    4 v  o9 Q" t6 B% ~& {6 v
11. 
    . W# w' q+ f3 m# L6 ^
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    % M" @) k- k5 Z7 Q
15. 
    , B3 r" J& x* L2 d
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
      S" g' y. I% r4 `
19. 
    9 g, ?3 s- r. j. D" b4 H
20. s.Format(_T( "%d "),   i);
    * M6 R* L/ i! f8 d5 r3 w, g) V! `2 Z
21. 
    7 m7 i& r4 V( _7 D* s
22. return   s;
    7 v% X0 l6 c" @, B' e
23. }
    
24. 
    : ]: f! g7 M! C- Y8 m6 ^2 q* _8 o
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    4 B, `( \3 ?4 c" h
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    + f) K) V9 {: c6 ^* m
29. 
    . f& u! ]9 \6 F: z9 ~
30. const   CString   GetArgString(const   CString&   name,   int   value)
      ~7 U& S/ m5 B1 R: y/ }
31. {
    " w; Q% \' w- q, s
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    5 S9 ?1 @& r( w; Z* h
37. char   buffer[10240];
    
38. 
    + B; z; g: v$ U1 c, q" E
39. const   CStringA   sa(request);
    5 e2 i+ d# r' _: \' S
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    " z- z7 W4 a2 l8 }, Z2 U7 F  q6 }
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - l7 h# _' J. S' `5 e, O
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    - U2 R- {2 b$ @
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    % ?4 m: {8 h, i+ L& I  [* `
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    , O" v* j4 m9 d) B
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % x$ V, E/ s8 |2 k6 Q6 O; x
53. Sleep(20);
    . V* Z( R6 C  u( i( p5 v
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    * D& _5 u/ ~& l/ f
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    : Q. y6 I) \# ^- g8 o/ P& ]
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    7 ]) D! h9 c, Z" n6 ?
59. if   (!rlen)   return   false;
    : s; p9 V& Q% F$ d: s4 `5 g& F  h
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    6 Y7 m! K* s  M% B' z; A- `+ R
62. 
    
63. return   true;
    
64. }
    ' a4 ~  [2 R1 o. @( O
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    ! [" q: z, u" w& j+ k8 S5 X
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
      ?1 }% t" c4 N8 V7 X* L+ J
77. sockaddr.sin_port   =   htons(port);
    0 h. ^, q& l: |* x3 M) e( u
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    0 _0 T+ h0 A0 ^6 G9 I3 X
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    % m/ j+ V" F  q' n' F
84. {
    ) y4 Q% Y3 u3 ]5 e
85. int   pos   =   0;
    7 W1 Z; M8 d! I8 t. w" b
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    8 P' v( L- d5 X) ?8 u% x! t
88. 
    
89. result   =   response;
    ! J3 x% {4 B: I- h! [  L, A1 m
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    6 c8 a" T1 m5 `' q
93. status.Tokenize(_T( "   "),   pos);
    % i4 \. z" L& c2 \3 _0 Y
94. status   =   status.Tokenize(_T( "   "),   pos);
    0 K/ V) d/ p4 `. }
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ; y! ]( G8 y3 m7 s6 J* ^( \
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     . @, t) G# P! R3 t/ B
101. CString   startTag   =   ' < '   +   name   +   '> ';
     & ]* }) \- Q4 c7 J0 N
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ' [) E9 Z3 S( N9 H! v  b- t" y8 [
103. CString   property;
     + r8 Z8 t9 [) Q0 z( e
104. 
     : v7 v. V0 D2 V# c
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     " x0 X" z1 g) }& Y
112. }
     & @) g% x0 v3 {- {
113. 
     
114. MyUPnP::MyUPnP()
     $ @6 V0 [! q! M4 J
115. :   m_version(1)
     
116. {
     " ^4 H& o" H  T* ]
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     # X' T" z; `" Q" V# o5 o: c- l
120. 
     : i5 {7 V  P; y& P7 N
121. MyUPnP::~MyUPnP()
     - T1 w0 t1 ?- @( g; b* T) Q
122. {
     # N9 F/ ~- _8 F5 Y0 U
123. UPNPNAT_MAPPING   search;
     ( H/ O* o# ?+ M" D" f
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ' `7 ]+ l- a1 a( E7 z# E
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     3 |& A" u1 Y7 \2 v4 P+ f
128. }
     
129. 
     # ~; y& \1 w: k' g
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     / ?3 W! z9 j  V6 y
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     6 u0 G. s4 t" `
136. if(version <=0)version   =   1;
     ' Y9 z5 N  [; A% n
137. m_version   =   version;
     
138. 
     5 q3 ]5 n* Q* W3 k
139. #define   NUMBEROFDEVICES 2
     1 r: Q7 m2 |& b: h  U" c2 K
140. CString   devices[][2]   =   {
     3 j6 G0 P4 l, C. |9 ~  G  e
141. {UPNPPORTMAP1,   _T( "service ")},
     ; m9 z7 H- f/ B8 e) b
142. {UPNPPORTMAP0,   _T( "service ")},
     2 p; V/ |1 \0 Q5 t  P) i
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     9 x' l2 l9 M% O) l/ H
144. };
     2 |  F4 A; h2 t+ ^- u8 |, T% i
145. 
     3 [6 |6 t$ ~1 d6 L' q( M; @2 s
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
       q5 ]* u, l& E$ l7 n4 D# n
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
       z! X& M5 \5 v4 V5 ^5 a! ?5 u
150. int   rlen   =   0;
     * W# c2 ]" f  F; I3 O# d% t
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     4 s# ?- ^/ Q0 ~. x0 x
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % k, G1 A% {# g7 B4 F" H
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     $ Y! c+ [+ C# B; d
163. 
     
164. char   buffer[10240];
     ! s" e! x4 s# l
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     / i/ \, o5 I: U% P1 A# l
168. 
     5 y1 K7 ?$ k5 u0 F+ |$ p  P6 R# W
169. CString   response   =   CString(CStringA(buffer,   rlen));
     4 X) W: [, P3 Y+ z% S
170. CString   result;
     $ v$ \! o4 Q* e( m) E
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     # Z7 A: h0 C  u3 j4 z2 d
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;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     , o& n% s1 F/ ^! A
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     0 N& H; o* m5 X
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     2 Y  X  E7 f. G
184. m_description.Trim();
     ! e5 K/ L" t6 a, I' a6 H
185. return   GetDescription();
     
186. }
     
187. }
     ! f$ p& r9 R, a/ v
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     
192. 
     5 k0 a& ]4 N2 C$ j) p
193. return   false;
       c2 L) j" j4 }4 p! U9 m, ]& n
194. }
     

复制代码

天云上人

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


5 K: w0 f; o! G4 r2 s0 e+ r7 d- R+ n///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


- L- p( W- m9 P" I: `  e( B! h#pragma once
$ R; i: n% m5 l+ Q/ f* b2 \2 Q#include <exception>

: Q5 H2 j5 f9 }% b5 ?
  enum TRISTATE{
        TRIS_FALSE,
7 [) Q# o4 J1 z3 }0 R8 i        TRIS_UNKNOWN,
        TRIS_TRUE
};
2 [+ v2 k, a, Q4 s( J

/ ?8 K. ^; F  h, c$ ~: F3 D" Tenum UPNP_IMPLEMENTATION{
6 _- k' ^6 C  }3 E+ r  ?: F; |        UPNP_IMPL_WINDOWSERVICE = 0,
$ F0 a2 ^7 }# }% j* S4 ]. p        UPNP_IMPL_MINIUPNPLIB,
: k. [- |; w7 P' x2 {        UPNP_IMPL_NONE /*last*/
  j* h! i1 X& T4 ?8 T: X0 b};


$ n! ~( u% w7 y6 r5 e
3 v9 |: v2 ^4 v" [& n+ Z) l3 `* I
class CUPnPImpl
8 u1 d$ y- X9 Z  A- Z" ^( Y{
+ ?0 r0 v( R1 ]) c, Dpublic:
        CUPnPImpl();
2 m$ k+ O- O6 r; |6 ?        virtual ~CUPnPImpl();
( D8 P. {$ Y  N2 e, l, ?        struct UPnPError : std::exception {};
& L. d' j* j, @% `, G: x3 W        enum {
1 y: W" ?$ ~" X                UPNP_OK,
  t* j0 ~7 E% o                UPNP_FAILED,
                UPNP_TIMEOUT
        };
+ ^8 j2 |$ }: L- L2 [5 M9 M4 Z

9 ~) I; Y0 x3 u) E7 Q7 ^        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
) o7 v* J7 z- f7 \$ Z; U: K. L$ s( }/ c, v$ d        virtual void        StopAsyncFind() = 0;
) q5 K5 m1 t  Z  s: S  A  R* R& v        virtual void        DeletePorts() = 0;
" R9 X( `/ ~( W3 A5 n4 b        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
5 W! @" h/ r- O& J8 J7 f( u( h. u# v       
. Z+ Y3 k. j5 m. U$ q% b        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
+ @8 h% A9 Y2 n3 w        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

9 Y! n1 z2 P: \9 V
& S! s( F" }, s- W$ D* |$ t0 m& _) u5 o// Implementation
, G$ l. }# f" oprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
% `/ _# A8 D6 C: @1 h. W' a9 i        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
' O" L7 k( Y0 T- A9 T        bool                                m_bCheckAndRefresh;

# C! ^1 N5 z' T+ l" G5 [
private:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


};
; Z, j( A% W/ t. R9 I, n
8 P" t3 N7 ?% X2 d. c0 X" i
: j" H) n7 K; {# y. w8 a9 Y// Dummy Implementation to be used when no other implementation is available
) E0 K* R- k. `; j* j  N* Hclass CUPnPImplNone: public CUPnPImpl
3 x* q: f1 J$ _6 @# B4 [{
$ u4 ^) }5 v- R4 W, gpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
( P, f7 N. w+ Z: N        virtual void        StopAsyncFind()                                                                                        { }
7 ~) P6 E9 ~4 {6 ~" N" I2 Y8 ?        virtual void        DeletePorts()                                                                                        { }
' h3 c. n+ X2 `( P: E        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
8 {/ [  j# y8 ~0 r7 O. I};
1 N% Y2 g, ~7 w1 c
: B# u- N/ W8 _. k
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


; e- R7 c) |; P) s5 O#pragma once
: Y! H* Q& j# P; f4 g$ A#pragma warning( disable: 4355 )
+ C  ?& z7 z, E) C3 W) Y' u$ T: v

#include "UPnPImpl.h"
#include <upnp.h>
( d5 X1 Z8 r$ a( ~" j# V#include <iphlpapi.h>
4 \  ^2 J5 k- [2 A6 D#include <comdef.h>
#include <winsvc.h>
, Q2 d0 F! r- Q! [; G8 c% j6 h% ?

) h% S  Q. M/ C4 A; M. y; q- ~/ l#include <vector>
3 o; g7 P, Y5 N0 [# @#include <exception>
#include <functional>
; l4 L3 V2 @6 R0 D, p


. c( f- }; M7 h" b1 M
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;
. G0 b5 b% ]( xtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
/ z# x" W3 J+ s6 n0 f& V/ Stypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
5 s9 W$ n) j- M  V) Q
) V+ p+ G$ Y1 r" a
% |2 x$ Q8 ]1 G+ c, r3 e5 otypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
+ t2 H2 ]1 s/ R4 G+ L  PMIB_IPADDRTABLE pIpAddrTable,
. ~7 G- ~* B5 P6 V# d$ `9 \  PULONG pdwSize,
  BOOL bOrder
0 o1 X+ ]2 i1 h) F4 w1 Y* z" U);

/ Q& z+ _" I& E
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
6 P1 |1 p5 C9 t, X);
7 Y0 E/ Y6 O% o" `/ H- q  D& u& h

$ O0 ~- }( L( G$ CCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
' }2 [% h( M+ `5 F" i  c- u7 S# }

8 G0 }7 O! I8 d4 S5 c8 W% Uclass CUPnPImplWinServ: public CUPnPImpl
! S; K6 F% n3 s4 ]1 H{
+ v" j( X0 |( S# A  K/ W        friend class CDeviceFinderCallback;
. y- U  C% f! C# g; a% q/ f        friend class CServiceCallback;
// Construction
public:
: K6 x) U7 [; k- Z! B; G1 G( r        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
) z/ Z1 I. R8 ]2 z$ ]3 L
7 M; l* a* b0 M# z
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
; @+ i* o! q0 X8 L6 h) Z        virtual void        StopAsyncFind();
( j4 |, F) l# ]0 R6 `# S5 J        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


+ E2 [7 q! L7 T3 P        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
1 F; s8 ?! |7 `        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
2 O$ V6 x, {" i        virtual bool        CheckAndRefresh()                                                                                { return false; };
" x( p% n/ F6 C; I5 ~. |

, D" C; o9 Y1 x, ?2 wprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
. F1 u% [4 h) x3 f9 V* L9 e        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
! S! x: y4 W! H6 @1 C& J        void        RemoveDevice(CComBSTR bsUDN);
( B. b' s6 ^5 P$ W% u. V0 K        bool        OnSearchComplete();
$ `+ l/ c! R( _1 V5 w        void        Init();


6 p) [8 y/ t5 u/ T: A( h1 p        inline bool IsAsyncFindRunning()
  H) k' j+ N' x        {
% O4 E) v: M/ N' A+ @1 `                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
8 `7 Y/ d0 p; k$ [8 @                {
2 m! m& n# _/ [. T) c                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
9 x$ z' J! W1 O6 M$ M% q                        m_bAsyncFindRunning = false;
                }
                MSG msg;
. f4 D& `6 }" _1 c  [                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
  |) v% B2 }+ g                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
* t! L: U4 I2 w3 c5 c4 d0 [* B        }

2 Z# l" k+ x' J' Z+ \* i
7 j& d2 `3 g8 v, c" S) T- P        TRISTATE                        m_bUPnPDeviceConnected;
/ M1 U/ C& |2 d, ]0 X; Y4 [

! u; Y2 r) w! @  f+ z- U& C% U// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
7 d* j) L9 V! w, R  T  h        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);
0 }/ h6 k: E$ M+ D7 _* [
) K+ f% B2 ^5 V5 a7 A( n$ V+ K
! i' N, B/ `8 a        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
, }; q" o5 s- y+ X4 `
7 B& E0 k0 b$ ^# ~
0 z2 @( v# b& x: a9 y$ O! B        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
1 q8 Y! m5 i) |( c4 s        {
/ C1 R  q7 \6 n1 X: W* h2 T3 X2 Y! A                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
. C" W' z3 T8 D$ X1 `; \" Z                result_type operator()(argument_type device) const
: s$ z8 Q2 f( o2 w                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

1 g4 r* f  R( v2 _5 [9 i
% A* s1 A1 w! G                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
3 S8 B2 [3 F. k4 t

8 l2 R$ I7 _0 S                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  O- O& ^  X: ]7 ]9 |                }
                CComBSTR m_udn;
        };
/ a9 O3 O6 T$ r       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
6 b# w; |& _& K; m        void        DeleteExistingPortMappings(ServicePointer pService);
0 B5 }$ D( _2 o# V+ {        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
* N( u) p2 a3 G# U6 q                LPCTSTR pszInArgString, CString& strResult);
' A% O* g: Z: S. A2 }8 B        void        StopUPnPService();
2 q6 o) ~' g; u% ?+ {

7 Q0 N6 g' P- J! ~        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
; e2 n0 l: Y& |5 i, ]( _- W        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
+ _, i8 U) g. p" q        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
; `# V* M5 w  x% g9 E; v# H! Q) z        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
( c% l' a1 i! D, ~" N4 t* Q        CString        GetLocalRoutableIP(ServicePointer pService);
* B& Z% Y  ^0 Y" V- S
. x- a  f; u0 j9 e- X2 T) m
+ W3 F3 B5 ^1 u8 S$ X9 b8 }* W$ [// Private members
private:
' U# g% ?; F1 t0 w3 t* [. S        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
  i3 b- _1 W- E( ~- f5 t        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
* H& d+ h9 H9 X+ ^
0 c" M0 ~) D- m# m
- A: c4 X# }" m' g# ]( z        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
- _$ _/ K8 x: F1 T$ s1 j        bool        m_bPortIsFree;
: L$ ]4 F; V/ Y: H) P; b        CString m_sLocalIP;
        CString m_sExternalIP;
- z/ F+ m/ p% y7 P) {* O, b        bool        m_bADSL;                // Is the device ADSL?
# z& w  q% u+ N( m2 ~, N        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
5 Y  |, Q- Q" |5 }) @( L        bool        m_bAsyncFindRunning;
6 _: B8 K/ X+ W+ J4 d& S2 f$ F6 i        HMODULE m_hADVAPI32_DLL;
1 i+ q& K: n3 z* |        HMODULE        m_hIPHLPAPI_DLL;
2 ?; \: \/ [& j1 s4 g        bool        m_bSecondTry;
6 S3 [0 m% v. H5 ?1 \/ \        bool        m_bServiceStartedByEmule;
7 z" F$ |! I8 h( _" d9 ~) D        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
1 K2 a/ B) \9 L( k9 X; d9 G- E
2 X1 T7 O7 B+ ^( d) f7 m
};
0 U2 c7 G+ `! m; ?1 N5 O1 ?7 D

// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
. m6 g1 j) D: C) l% H" O  [{
/ R* \7 g3 n  D" L: A0 Fpublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
9 M: p7 h! F1 o  V6 [9 ~. {

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
2 w8 N9 A5 i. l8 y% q, s   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
) C3 o7 D3 Q" B5 W8 O) _. ^
# f, S) L. F) B
// implementation
2 m6 l5 c5 i, H- b& ]private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  B* [2 A6 U: C! r' s7 t8 s        HRESULT __stdcall SearchComplete(LONG nFindData);


3 c! K1 M2 r" p0 h5 E- p$ Tprivate:
        CUPnPImplWinServ& m_instance;
9 G9 ^3 w( p$ M" W% d: D( Z, u6 r& w        LONG m_lRefCount;
+ j3 x) s! [: X' \) ?7 l! H, o};

. l/ {# F0 f& o, ^1 V
: g& y0 c* Z! G5 Y6 g$ ]// Service Callback
class CServiceCallback
7 S; L" w) i1 M7 C! V/ x+ l        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
. ?- R3 ?% M" p. a: x& C1 `' g                : m_instance( instance )
        { m_lRefCount = 0; }
; b* F) t$ U4 F   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
( F+ K5 S! o8 y6 U   STDMETHODIMP_(ULONG) Release();
/ F# Y! w3 m. @

// implementation
) c& ?- E" p; m: k0 Dprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
+ e3 i6 S: m, E( g, h& y        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
7 A% h- U6 \0 [5 Y
  m9 q) U: ~- M: J/ @) T
' f! X  i3 M# n* j! cprivate:
( e2 Y0 H; ~( K: Y$ d5 L* E% Y/ d' \" `        CUPnPImplWinServ& m_instance;
' @/ x- ?/ T; g' }        LONG m_lRefCount;
};
8 ]& c7 u0 g' J: |6 v
; |- Y9 a( G! C- j" j3 W, K  ^
/////////////////////////////////////////////////
" q8 _5 L9 E, J6 v$ U* n

使用时只需要使用抽象类的接口。
$ F2 ]" B+ d( s3 YCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
& {& g* \3 P: M" P* w/ v4 O* RCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
8 u$ v8 D( J" O& Z! hCUPnPImpl::StopAsyncFind停止设备查找.
! Q" _6 U4 U4 c0 m! ?! n; F2 f6 I- KCUPnPImpl::DeletePorts删除端口映射.