UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   9 T( n2 ]: r' `4 J$ k
4. #pragma   once
   
5. 
   1 R' z1 f6 _; a1 O& q2 V
6. typedef   unsigned   long   ulong;
   
7. 
   9 s5 q* d) B  x( z5 |+ G
8. class   MyUPnP
   
9. {
   ' [! I/ y0 d9 }: t7 e3 G$ S
10. public:
    
11. typedef   enum{
    9 ?/ [9 F5 R; o. S8 |
12. UNAT_OK, //   Successfull
    ( ~2 m+ G& b" Y; w+ y: l+ @
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
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
    3 W: p: ]& S$ D; f# i% V" M
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    8 j6 w1 C- L. [) U# H9 F+ H% T
18. 
    / L" ~9 _5 i# C  E1 H2 G3 C
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    7 w6 o! g- H: ~) V/ F2 \" V2 z
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    2 B- Z( }, c! \1 A
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    ; I+ z7 N0 e) V: g( |! \% U
29. }   UPNPNAT_MAPPING;
    
30. 
    : E1 J  u1 x5 H+ t7 B7 c- |) G
31. MyUPnP();
    , d2 k3 J0 a1 z; b6 d" }# `# J
32. ~MyUPnP();
    3 ~* N( L: B/ e5 u
33. 
    9 x1 k  L+ b$ N, U" i5 Z
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ; g. f9 H" M# x9 K
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    3 m5 i4 o, W9 o' H! X* o5 ]
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    5 i7 G3 `( \$ g) g& U0 s% x1 N$ D
41. bool IsLANIP(WORD   nIP);
    4 Z* L8 ^: U7 @* Z; M
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    ' s+ [) d! q0 x3 ]5 x/ @+ C
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    % Z: Y8 l3 g8 N! B' I* D* P; Y
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    7 d8 x1 D' L% x
52. 
    
53. bool Search(int   version=1);
    5 \- l0 P/ r# M" `$ n2 g& A9 U5 I
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    1 j2 l0 `* P  U7 @$ n  X
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    4 F' ^. U& T0 r  e1 }; E" z2 N
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    7 Z9 d) Q* ~. M2 Q
60. CString m_devicename;
    
61. CString m_name;
    7 ^2 n+ i- d% k5 r$ `
62. CString m_description;
    ! K+ Y! }  R: `; A# G/ w3 F
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    ( \( k4 n0 s; L
74. WORD m_uLocalIP;
    8 i6 A& }7 M. d  J
75. 
    
76. bool isSearched;
    
77. };
    . [# I9 h' ]% z( n# h
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   / w) }0 V. W- E2 L& ]' u% P
2. #include   "stdafx.h "
   8 `( [4 k1 ~% b! ?0 r" A4 `
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   8 X( p: Y5 c3 U* g# Z: w
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   4 T; |% m; X+ ~5 A' `( B7 D
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    2 v0 o% o( m  M3 u6 Y
11. 
    6 x. S: g" i9 E4 Y4 S
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ) G. w$ U! n! h
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    ) u7 g- i/ C3 J5 I/ O# L0 |% {) x
15. 
    / s( U7 K2 }# F
16. const   CString   getString(int   i)
    ! l$ @+ @: w- B( s
17. {
    4 N) L; k( `6 p* N" f; V
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    " G/ W7 j+ m+ `3 F
23. }
    
24. 
    1 H" z) {, S9 h$ a5 P) e8 ^
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    1 L+ I: y) F2 s' D0 j
26. {
    / {6 y: k) x' u' U5 S9 s7 A
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    4 s- b$ T7 A, C& f3 i$ [
28. }
    
29. 
    2 K" }' u7 F9 x/ Y
30. const   CString   GetArgString(const   CString&   name,   int   value)
    4 S4 v3 S0 Q' b2 z5 T* q9 y& _
31. {
    - z7 M2 t; I% i, k# O
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    8 J# o) E; i& M3 F5 u  ~
33. }
    
34. 
    - b. L- q" h% W; A( k/ J. i
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    + ~9 o& |/ T1 \
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    ) w& s: ?) p% S. v4 |7 I# W4 u
40. int   length   =   sa.GetLength();
    - p* Q# ]* W: e6 T
41. strcpy(buffer,   (const   char*)sa);
    2 k4 o8 O* n! B9 }: C" {, f
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    % `1 z  i  F1 J( c4 ?1 `
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 u8 I1 `( z% a2 A- i" H: Q
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    9 ~% d/ v4 I& T: R  X3 E
50. u_long   lv   =   1;
    8 l' K4 L: E. ?9 k, G# O# m) d
51. ioctlsocket(s,   FIONBIO,   &lv);
    . g& s4 k, c5 _4 z: p2 u
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    $ A" M9 `9 M5 f4 U, y0 a
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    2 S. @1 o" u  J0 Q
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    4 c% G4 _2 O6 V, j, J
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    6 v1 k  b/ f4 g
59. if   (!rlen)   return   false;
    1 T7 E7 ~+ W  U& c7 s- @
60. 
    5 O/ X: i) ^* ^: w' n* ]
61. response   =   CString(CStringA(buffer,   rlen));
      C0 M) b$ x5 p3 |# [
62. 
    
63. return   true;
    ! L( E' m& _6 T+ O! u# \1 U
64. }
    
65. 
      V  a" |, I! C8 s! `4 J  W( F4 Z8 M
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    , |7 A/ y, [( ~4 m+ L% \" v4 D! \
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
      R! p: C' G$ u) {! v0 z2 M! z
72. strcpy(buffer,   (const   char*)sa);
    % X8 A( m  t( ~6 u6 v+ Q& n+ W
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    # v1 w# [& v' m0 a; \7 o4 j
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    1 X' }' T! W+ f# ?% O: k8 |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ' |4 G! w1 o# B/ j  t
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    ! @% N4 A" ]5 U; h4 r% T% E6 N
85. int   pos   =   0;
    % Z. S. w9 \% W: F+ J! }7 I
86. 
    ! a2 {! B& J) ^+ v- f
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ' T# Q7 v" m# g/ Z$ M
88. 
    
89. result   =   response;
    0 z, W" B. ]* o7 z, @) o
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    - a) W! N% K: U2 K$ }7 P
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    5 k3 g& d* r# i) a: I+ [
97. }
    ; a. A, i1 `' C: c0 Q# K( M- f
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ; Y6 l7 D9 \+ B
100. {
       b! `. D% @  R/ z: S0 ?9 }
101. CString   startTag   =   ' < '   +   name   +   '> ';
     0 N0 o( C" z% j0 D% H4 h
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     4 N4 ]( C2 f( s+ I6 @. @
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     ' R6 X. m* D5 O: g( x
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     : |; C7 d  a; E! Y; z% u% n4 ~
109. if   (posStart> =posEnd)   return   CString();
     ' d% s, ^  W9 Q/ n6 l  C
110. 
     3 ?6 h2 H1 |7 W7 n
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     + m1 ?) W/ D' Z* J
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     1 B3 w2 J9 d. Z4 Z: P' Z$ o, x
121. MyUPnP::~MyUPnP()
     1 I0 u' ?5 G( l, }9 H
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ) I) n0 y4 x' S; @0 c5 }
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     5 h2 H+ _. T5 c
127. RemoveNATPortMapping(search,   false);
     6 s6 s7 b& I7 n) U3 w5 P
128. }
     : j) u& I8 x$ \' A7 P1 e6 N! b
129. 
       c$ E! k. k7 ]4 c% l
130. m_Mappings.RemoveAll();
     4 V; I$ ]% ~7 a  v+ A  ^
131. }
     
132. 
     & u  w) J, R: w8 j
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     6 T4 T5 Y! A6 i0 d; D/ u! N
135. {
     
136. if(version <=0)version   =   1;
     3 l6 M- |- X# Z2 B4 S( @
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     * ]2 I  ]% t" G' ~# }* N- w
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     1 u2 \( a& p. _& v  C2 b2 o  m( a
144. };
     5 Y! x$ I3 c" I
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
       B7 d  |1 y3 A3 r
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
       ~( ^5 W7 F2 R
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     8 t9 l4 E0 l! K' y# O. V5 d4 u
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     2 Y8 P) p: t& ~( H8 I" u  A
155. CString   request;
     " L2 q1 [! q  O" 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 "),
     9 ~% W) S: L1 n3 L# r  e6 w' J3 N
157. 6,   m_name);
     6 d( A" a$ |5 l$ I7 o' E5 s
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     # E1 {& F2 V/ q; C
162. Sleep(10);
     " E& Z9 _8 n- i6 l/ `. W1 a/ h
163. 
     8 `* z' d3 Q9 q6 k2 V
164. char   buffer[10240];
     5 m- [8 q  F- f6 o* h1 @8 Z  b
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     - t7 ^+ G* ?9 P  B* R+ E& N
166. if   (rlen   <=   0)   continue;
     ! k, i# H% G; `, B+ j
167. closesocket(s);
     : s& Z$ b7 k# L1 ]" J( Y( V
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ( `; [. ?2 z# a
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     6 |: z2 C/ W6 j+ [# K" t  w
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     1 j- j8 d8 {" P8 B
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ! r  f4 {: v, v$ @$ e; v
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     % ?) Q7 ^5 {) b/ y  c* D
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     * e, t4 I4 \2 h5 B' M, p6 ]
181. if   (name   ==   _T( "LOCATION: "))   {
     ; P& z  P7 R+ G( \$ R
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     : @0 z1 @7 I3 l2 [+ t5 S1 H* l8 s
185. return   GetDescription();
     
186. }
     4 V8 r" T5 f) B2 f: c
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     5 u1 B4 I& U7 h  \2 H  P! C
192. 
     & ~. P, c9 r3 B, p( P
193. return   false;
     . }" v9 l5 f9 c  E; O
194. }
     

复制代码

天云上人

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

2 ]2 J6 I0 [, p$ Q, @; }( e
0 f: p2 _2 K$ `# i///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
8 W& B& X3 p% c9 i& v* E: r% x
" P3 b$ _2 b8 `  g! E
#pragma once
, {' I$ P# s. w+ F6 I#include <exception>
, Z( i; `5 z8 N% [# y/ M$ o( Q) S
$ ^7 g9 x% K7 D0 `6 J: Y' M
  enum TRISTATE{
( c" W& t5 N* F7 t* u        TRIS_FALSE,
( [/ }: P6 }" @( o- b0 U        TRIS_UNKNOWN,
7 }+ Q  D+ D3 k& i        TRIS_TRUE
$ p$ E9 ?5 A$ h; U# D' a};
8 d! r. \! S4 c  {& M" X) n

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
" G* v. r" s4 C" J+ s% [        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
3 F& O3 k+ d) l" `5 z
2 }$ J, [6 {5 K8 L  w* @


class CUPnPImpl
8 M, \, f$ O" T( B{
public:
! v% b2 P$ f; r9 d5 [" }        CUPnPImpl();
+ `6 {9 U& D4 S& v, d        virtual ~CUPnPImpl();
2 C4 C) @' B" k0 N1 ^        struct UPnPError : std::exception {};
        enum {
! q0 k; \; F+ V( \2 L  c                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
3 `* h; @4 s9 _' f5 r        };


% L, _. _7 e0 p' u& U        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
  Z# c7 Z3 z: E. Y        virtual int                GetImplementationID() = 0;
8 Y( G7 t3 o% g8 [3 s( Q       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

/ O1 C- w6 z/ \& j+ s
" x, C. }& e" I& B0 C* W! x& D        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
3 `0 k$ w" [" h6 O& q5 v" ^! H2 a        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
* w# a" o+ i' E6 m! ]& g: p        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
, x# B! A+ q" }0 R        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
" V+ \& D; W& u: [8 d: ?% m
3 ?) w9 X9 k/ w: u' z6 I& r+ A
// Implementation
3 I( V  p; H) X. a+ yprotected:
" D$ s9 @1 q) D; v( G        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
+ m, z* m2 X% k- U$ T/ U: q        uint16                                m_nUDPPort;
' V& }! k2 Q) P0 v- l        uint16                                m_nTCPPort;
/ t- C3 E- l: B% ~: U$ y, p        uint16                                m_nTCPWebPort;
6 M% M9 F7 }. Y' u3 J) P5 H( S0 k        bool                                m_bCheckAndRefresh;

% Q+ c1 S- }: b+ R. k0 T
4 N! Z, M" f5 i  U9 @. Y9 Qprivate:
) z# e# k" z: }6 |% X        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


. d7 n+ j7 p  u; B4 A: Q9 O( {};


: e4 r: Y3 Y2 `+ d( V' _// Dummy Implementation to be used when no other implementation is available
; s/ f: H6 Q9 s+ |3 `class CUPnPImplNone: public CUPnPImpl
{
public:
! w$ y1 P9 o! O        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
; @  _' J% G! c% |6 ^* A. w        virtual bool        IsReady()                                                                                                { return false; }
* H' D0 K0 i. O0 [0 U        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
4 d( r4 U: q3 l, a, R; y2 E};

$ c& t6 d% J( Y
7 |+ D3 |+ }, x5 U  v/////////////////////////////////////
2 l+ s& `  s% B; G+ M//下面是使用windows操作系统自带的UPNP功能的子类
$ F, }7 Q' q( A! S* K2 @5 m
' a6 v" k: {' m8 n/ e9 _/ I8 N
6 q2 J/ W& R0 m#pragma once
5 i% v" Z; c& ^6 p#pragma warning( disable: 4355 )
! X/ e, S: h; m3 n" U& |8 t
- Z& q, E4 Y3 l8 D5 Z7 G& o' U
#include "UPnPImpl.h"
#include <upnp.h>
) S  I: Q( `3 ?8 V# l% d#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
4 S' b3 ^2 E6 W; M1 y# o
" R7 B& ]) Q. g* W% b1 U
- E# X9 C1 N& ~9 ]#include <vector>
. U* H  r2 x4 m  u8 V  @" b  R#include <exception>
#include <functional>
$ I- z3 u. h* g( _9 b6 G
2 g. R* T; Q; b4 ~1 H( `' x$ t% z

0 r" l* p$ h/ }# F
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;
- C0 ?% F9 ?1 L3 |  E" k" ftypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
6 ^% e2 i; u6 z+ Qtypedef        _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;


- P2 l0 h6 H  b) A( D7 Ctypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
: r. z0 U( Z  t# _! D$ D  P& D  PDWORD pdwBestIfIndex
; A" S, e+ M" ?0 X0 Y! c6 A);

+ O8 N) \5 C6 I  r7 B5 c, G: T
) a0 C6 F! R$ G5 _. dtypedef DWORD (WINAPI* TGetIpAddrTable) (
# @2 C% G8 [: J6 \  PMIB_IPADDRTABLE pIpAddrTable,
  L8 ?& }/ X, _: a( {# Q  PULONG pdwSize,
# }! c& V8 T3 K: d+ ~$ |" L2 j9 F  BOOL bOrder
2 v" m# n! U+ Q" J);

5 m8 \# ~5 D; R9 e1 U2 D! ~1 m
typedef DWORD (WINAPI* TGetIfEntry) (
6 q, h2 y0 V8 ^9 c  PMIB_IFROW pIfRow
2 l) [4 L, C3 w& o3 a8 [);
% p2 g' @$ c! N9 S$ O

1 |  q: q$ t* O8 G0 p$ q& L0 CCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);


1 T& R% W9 V7 w' y) Xclass CUPnPImplWinServ: public CUPnPImpl
. C5 ]& V* R/ t) w5 m! [{
* E- d- Y0 D2 ^% ]  R2 n        friend class CDeviceFinderCallback;
0 G( E/ O+ e2 a) j. t$ T        friend class CServiceCallback;
# O+ S9 Q( {( p// Construction
public:
, A9 `* M: a- C; l1 a6 q        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
7 E) q3 \/ n7 a# S* O
: o4 K8 s+ e: n  x
+ c0 Z: G, G9 c/ F8 d8 \        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
9 l3 O- G3 p3 v+ |        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
4 z/ U7 W5 i, f, v) c2 P        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
# h7 `& G8 {, p/ R
, L+ b: [4 @. }
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
8 ~, X/ H4 m+ D0 O8 S) F1 j! ?7 q        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
( u- B6 \- q) P) e7 [8 f9 q        virtual bool        CheckAndRefresh()                                                                                { return false; };
' ]  b6 N% R, \- |! }+ @, V7 M. E

8 F4 D/ j/ l/ n. q) i0 |" k6 e0 @protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


0 B/ N$ ^/ d  M        inline bool IsAsyncFindRunning()
        {
5 Q$ k1 O/ G+ F# @                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
! T: i; x" E6 @6 F1 T+ o1 V                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
. l# \7 \' M: R% u                }
% q4 P, {% l+ _                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
" z. b, }8 _2 ?' i: J) U                {
0 I0 L8 z; H' X9 e& U* A                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
5 y) K5 ]" b; ^/ V9 R                }
                return m_bAsyncFindRunning;
        }
2 b5 X5 O( Q, J$ M% J5 ~) ^: [

# S- [7 l# w9 ^. ^6 y        TRISTATE                        m_bUPnPDeviceConnected;


0 L& Q  r9 H5 \2 n- [7 _2 D// Implementation
- g) }+ R5 |! D. w$ s        // API functions
' Y+ ^0 V) d) R. O" I: D        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
- K# I* H; U6 w7 j7 c9 Z; I        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
* E1 @- s$ _" \( F: a* v# \6 D3 }& d- o        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
8 |- H5 ]  b' h. q        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 p: j+ B: K- H, x5 X* f- F! g$ W$ Q
& r1 ]' ^2 }  J$ p% n' |
        TGetBestInterface                m_pfGetBestInterface;
; t/ E# C! I/ s2 }1 w& I        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
$ y1 ^0 A6 A% {$ Y: _9 C0 Y+ q: g

& M  C! U3 y  W* u: B; W0 F  C4 x        static FinderPointer CreateFinderInstance();
: W" G$ _. ]1 H$ `! B        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
: L! b% L! a8 C8 \2 o3 V7 {                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
- o& H0 @6 s( F# V0 j6 x                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
- r9 y! h# r4 ~% a% k' p: E  m
; P' h5 I- r+ x, S# R
3 d' t& r! T- R' M2 U# O0 b                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


9 T2 t: j" r8 F; j1 P) ~' s; z                        return wcscmp( deviceName.m_str, m_udn ) == 0;
( N( x, t0 B( s                }
1 ^+ {2 o9 O  ?5 i, M2 i                CComBSTR m_udn;
        };
4 _3 u- G; o0 X% |" p       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
6 [8 g* O% s  G  ?, b        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
: i3 q  X( o6 U% H5 D1 x        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
  Z" S2 K' @8 `                LPCTSTR pszInArgString, CString& strResult);
! o+ y2 Z( v% R) U: Z7 Y# i3 R        void        StopUPnPService();
) k- u4 ~$ E+ Z% n* `

        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
1 f; b' p: D9 J- e0 N) ~+ s        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
. M. \' |3 I( Q9 B$ d( s: M        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
0 W" n( S' D4 s9 `) y        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
. q9 E' @$ a" H/ s& n        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
/ q5 z- `+ V0 w- y1 y        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
7 J1 h* _: H# N) O" t) Q+ W        CString        GetLocalRoutableIP(ServicePointer pService);
* I4 u& G+ o& k5 x0 d- T7 D  C
5 x4 L+ L7 {, G- n" x- J
// Private members
9 P# X. T1 ^4 T5 D$ Vprivate:
9 I" r+ X4 Z6 a1 g( S7 W        DWORD        m_tLastEvent;        // When the last event was received?
/ x- E( O9 p; I6 C4 M0 g        std::vector< DevicePointer >  m_pDevices;
3 ?5 H, S% z5 D" L: j5 ~2 j        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
" l/ Y6 e2 I* g6 t; j& p4 l        bool        m_bPortIsFree;
: u: x( t0 }8 }9 p        CString m_sLocalIP;
$ I9 t5 {5 W$ V$ l        CString m_sExternalIP;
1 M  O5 J1 H0 {        bool        m_bADSL;                // Is the device ADSL?
) c) G: G- H* ~        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
7 f/ {1 F9 x( {, @* c        bool        m_bInited;
        bool        m_bAsyncFindRunning;
9 p# y- d0 N5 j: G9 y! ]) U; j  L        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
$ D+ Z% k" r$ O5 f) X. s) ^        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
/ W( x2 ~$ c8 ?' E        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
' ^) x! [: ^% O# L4 @# p( Q

};
# K- `9 k/ a* w

// DeviceFinder Callback
/ e1 ]+ J, @1 ?class CDeviceFinderCallback
. ~, ^: U+ [  L- Z' D& X7 H        : public IUPnPDeviceFinderCallback
# G" C6 ^% r  c5 n5 n+ Q{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
  a8 E! u/ f$ g! k0 M

( O3 E3 d# M4 a4 [( s& n   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
& d+ R4 d# U; n( J$ s  W3 m   STDMETHODIMP_(ULONG) AddRef();
2 ]3 s1 g, l9 V4 b, ]+ R( j   STDMETHODIMP_(ULONG) Release();

( N* u( J* R4 e$ T2 {' h, s
// implementation
private:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);

3 i& o/ [  D3 E/ r4 X
. r9 w$ R: N) T1 X) V: H! Zprivate:
! y; d: I: @2 y9 L! s- m        CUPnPImplWinServ& m_instance;
: O- }& ]. a7 B. i& @# z        LONG m_lRefCount;
# v2 j+ b) _" B- w' U};
9 ]0 `  z) a# @# C# m& o. ^4 [

// Service Callback
/ V; R) B# K" S  cclass CServiceCallback
  s; W0 y  v$ x6 v7 S. B        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
" }( D. t* [4 ?3 n* I7 F! M% l                : m_instance( instance )
7 L4 c6 o1 o9 g5 N8 `& n, n, S9 W. B        { m_lRefCount = 0; }
% t5 l) W. A/ Y9 h6 A. P. l7 J   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
4 W9 V" G$ N  m2 ^( s* e   STDMETHODIMP_(ULONG) Release();
& T7 E( v- W. Q  t- }8 p. k! Q
' ^$ e* F& }( [. O5 p! e- [3 f# Q% `& M
; x0 R  r2 g# E# K// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
7 Q5 P+ w7 X' [1 Y0 d& O        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


0 b% {7 e, X+ Yprivate:
5 s( v3 ]- y0 L$ C; Z        CUPnPImplWinServ& m_instance;
% \6 Q- I5 F( K, a6 ^5 s5 R) S) j        LONG m_lRefCount;
. V4 K! x& c  d+ D3 ~/ G# O};
; `& R7 Y7 R- Y& B; t' l8 f

/////////////////////////////////////////////////
/ K# H! R3 z+ A! b% u0 D$ z
$ }+ P: W$ R. L( W2 Z' X3 i
使用时只需要使用抽象类的接口。
& f$ N% s7 \! U' q9 dCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.