UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
     X+ e& Y, r5 {5 P+ s4 P; u
4. #pragma   once
   2 z: u2 U3 \$ a# v' t! x, y
5. 
   
6. typedef   unsigned   long   ulong;
   7 r) M1 }0 ^9 S% a6 R* b
7. 
   , i' N/ u0 G7 I2 w
8. class   MyUPnP
   
9. {
   
10. public:
    ' @/ C9 Z: E$ T5 z. A/ _% d
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
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
    2 L$ l* X; W% A: ~
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    2 J( O/ p% y% z: F5 W  a5 I
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    ( n; S' j2 Y4 P4 H' q+ A" \" G
17. }   UPNPNAT_RETURN;
    & ]- S0 I; K  s3 v. F
18. 
    
19. typedef   enum{
    $ l( k& l7 M( o& ]- ?) s
20. UNAT_TCP, //   TCP   Protocol
    0 X- ~! f  O* j7 Y! W/ O
21. UNAT_UDP //   UDP   Protocol
    & j7 Y! B, e( {1 N
22. }   UPNPNAT_PROTOCOL;
    
23. 
    ( ]7 x2 r$ {9 ]6 A
24. typedef   struct{
    ' H! q( d& K' f5 T2 x/ M; }
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    4 J" K8 t" c. e; Z- z
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    $ m  `/ b2 I3 B& x8 c- z
28. CString   description; //   Port   mapping   description
    ! l2 _% x5 p7 u$ W  H
29. }   UPNPNAT_MAPPING;
      r  q0 V0 F/ v; }- n+ H
30. 
    
31. MyUPnP();
    : c5 W1 x' b+ ^" H  K
32. ~MyUPnP();
    + s3 P, {7 x4 K# e1 x& Y
33. 
    ; T% l' w9 X8 }* J  u  W3 A
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    9 g4 ?& g! [8 }2 q4 \8 }
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ( P* M' `# I5 c* E) Y
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    " O( g% G, [: L2 x+ y
41. bool IsLANIP(WORD   nIP);
    
42. 
    * c6 |. a3 F* ^% r
43. protected:
    
44. void InitLocalIP();
    6 F) O, Q  o) e  ]
45. void SetLastError(CString   error);
    9 ?! ^# |( \) z3 U4 E! v& g
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    4 Q1 |4 h; m; }, [5 v1 j9 {1 {, m
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    9 s! o! f8 R5 F
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    0 ^* _: G: V& Q2 ~, j" r  g
52. 
    / k2 {% a# ^2 ]5 ?
53. bool Search(int   version=1);
    
54. bool GetDescription();
    ) {5 s8 b/ w8 Y" D- @
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    + {6 F) n- s; u- `8 _4 l
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    6 U4 o3 n9 P/ u8 }5 |( d% u" Z! t" E
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    / q8 M* _+ t- K$ `% h
62. CString m_description;
    , g3 _8 m: P: h' `; z+ f
63. CString m_baseurl;
    3 f3 q8 P& u, C' h
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    ) p. Y* E2 P( e+ A
69. private:
    ! o5 U1 U! F7 I
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    3 g7 n9 x4 j7 d! M9 |& ?/ w
71. 
    1 \# Q9 S& t" Q/ c1 E, e5 O
72. CString m_slocalIP;
    & Y7 q) m% j& q9 l: f9 O. B
73. CString m_slastError;
    ( k- E* x; d( g$ V3 a1 b
74. WORD m_uLocalIP;
    
75. 
    $ o! B/ \& U4 c5 L2 n  ?
76. bool isSearched;
    & f0 k. n, n0 Y9 o- T% I  y' u
77. };
    ! ~2 v2 v! J5 g
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   7 U* R# j; y/ u/ m. d, N6 u
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   5 ]2 s$ e2 O2 q1 e
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   ( a, {' U* y4 Q) {: p
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    # V" R( A  b- W. V' m1 e
15. 
    # @6 ]2 }  d( e5 D7 H: z8 n" B8 e+ z. |
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    : Z6 L2 B2 x: [+ Y1 G) u
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    ) r3 a  ^" [  Z2 r; O
23. }
    
24. 
    ( f, \* V/ P0 S5 D! r1 Z
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    5 Z, }$ X% e2 F6 o1 D) L! h. v- x- q
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ( J, P" z1 q5 f) y6 ^
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    6 \1 h; |/ ^' D0 D4 L  x2 B
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    : I+ s/ _# h2 I" ~0 ]
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    8 f' q  o( h3 P- ^+ u2 `
37. char   buffer[10240];
    . U8 S& U" ?9 d
38. 
    1 _- @% t) w, k6 p
39. const   CStringA   sa(request);
    8 \6 I# b' g% @% {9 f5 H6 }' P
40. int   length   =   sa.GetLength();
    * p( N* a! {  M& K1 Z2 l3 W
41. strcpy(buffer,   (const   char*)sa);
    & G  T+ g6 ?9 F% a7 n; J( s- Y" u
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    7 C" |! ]- U( }
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    : x' Y9 V2 w- v: t' A& p
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    + C( p. F! h$ X
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    ) d0 i9 k9 S3 r
51. ioctlsocket(s,   FIONBIO,   &lv);
    6 d5 I" n; u2 d4 R8 S
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ' U" H, n& j4 J% ?5 c
54. int   n   =   send(s,   buffer,   length,   0);
    ' W6 Q, ?1 ?" E3 [6 G
55. Sleep(100);
    : n* N8 {  ]! S9 p* l+ [3 j
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ( s+ h$ t  @1 _' m% w" P
57. closesocket(s);
    ) u0 s$ o) b8 c. f: Y7 h0 c0 F
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    8 [$ v& u- z+ k8 P+ B6 p3 W& J
60. 
    ! r! }/ M3 o  J- a& n2 e" T
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    8 [* m: H- r) b! Y3 _! l8 Y
63. return   true;
    7 B3 y1 X( P  {! d$ _
64. }
    
65. 
    # i$ M9 D$ X% `; z" Z: w  `/ ]
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    % g7 W/ @! b. j( R: B
69. 
    " q/ y( y/ t. ?
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    : |$ i. v% e8 x' K/ G4 U+ t
73. 
    
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. 
    . V( D, `2 ?& A6 t- y& T
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
      \1 L/ e& p) p9 F' F5 T
81. }
    
82. 
    0 P- W- j( e3 `$ N; [% ]9 z0 [7 p
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    * e  [7 [6 g9 l, _7 s, ^0 \3 A0 v0 v
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    0 B. b. s& h9 }) ~& f6 w
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    2 b( l2 i* e# p3 k1 j. k
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    % Y  w" L) l6 x& p/ R4 {8 H# C
96. return   true;
    # {0 ]/ ?3 Y9 ]  D4 z
97. }
    4 A. D$ V( w2 z+ @' b7 c  f* u5 ?
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    $ o. g, m( j: o5 z3 R* U4 q
100. {
     ' o; p' _4 f4 h/ w: f
101. CString   startTag   =   ' < '   +   name   +   '> ';
     5 r7 `/ d" d4 V  _2 k. q& \
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     & b) y' O" x# b/ R( c6 D
103. CString   property;
     2 O0 C5 a4 d" n1 [
104. 
     ' j$ b2 j! F7 i
105. int   posStart   =   all.Find(startTag);
     , x# @4 j& ?% }0 H: d0 l; _: d
106. if   (posStart <0)   return   CString();
     
107. 
     " q4 }* y" Z* P: B- r2 D
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     & r! Z1 l7 O. p* u
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     8 F3 L# I: V8 \, ^/ e
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
       I  i# N0 [# x, Z. o
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     1 W2 T& w+ ~- n8 B
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     ! |; x( ]) v+ E* T
125. while(pos){
     ' B) V4 e. M6 ?
126. search   =   m_Mappings.GetNext(pos);
     8 l4 z+ T& C% h$ ]+ ^+ G" d
127. RemoveNATPortMapping(search,   false);
     
128. }
     
129. 
     0 U& n; @9 W' Z! u: w  D
130. m_Mappings.RemoveAll();
     
131. }
     # D7 N" K, Y  A! }3 C& _) {
132. 
     
133. 
     ) z  Y  l% z, d
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     1 ?7 g9 V. p3 G4 [+ T6 @9 W7 o
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     8 X1 E! o7 u! ^* P# i* N+ a' Q
139. #define   NUMBEROFDEVICES 2
     7 c; a) @& n5 K; l" H" }' g
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     / o: I" A+ O: x1 O* v7 u  O3 d
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     5 e/ O7 y, \( Q, d+ r+ ^
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     ! i( m- W- X- Q% h  q7 L" `
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     2 j8 x5 H$ w7 h4 ?4 z2 ^# l
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     ! V. g& Z4 `6 L- ?( D
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ) o+ g5 {( s. T! w- k. ?
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     - _0 d+ c7 r8 ]8 ~7 B2 J, m3 r
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     ( g# e3 N5 N" N6 [
168. 
     ) P$ ?) p9 |5 \1 k" _
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ' J6 G0 L2 d; e! D2 f8 t* O  P
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . A0 u  Y' U# x# O
172. 
     6 k& `! z3 e$ H* a! S. X
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " V# a2 i# A% n. y4 Q7 E
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     9 y4 S) N( F3 A. b
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     9 U4 G9 ]! A7 F  P3 U' l
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     ) |& O- \$ ?; ]9 y1 Y
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     / O1 d  q. _0 ~8 T! B8 [9 |
182. line.Delete(0,   9);
       j. }. \4 ~& @
183. m_description   =   line;
     ! t. E# e! P3 q9 w9 T
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     2 ]& H* u; A, y7 S" L7 }4 k, Z
188. }
     
189. }
     
190. }
     ' P/ A" T. A" P0 I: Z* U
191. closesocket(s);
     
192. 
     
193. return   false;
     0 s' v; I; z; e. _# j
194. }
     1 ?, a$ J+ x0 x7 u

复制代码

天云上人

以下有关upnp的接口来自emule，
% E+ H& `) i4 M, b0 E7 ~* r  a

/ a8 A5 S5 \3 J: Q% V///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
# E; a: k+ X. c6 w2 C
4 q, O- x0 w' A1 s+ Z# b4 ~
#pragma once
. f; j$ E6 A2 W7 A: @/ [' X#include <exception>
3 d/ s4 Z& B4 v4 o/ h
5 w6 |0 y7 J3 X2 M2 D3 c3 _; C
, r$ Y) l& ~6 m; _* G6 w  enum TRISTATE{
/ D* |. I+ d* j- t* A        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
};
- ^7 z' L7 l/ o" {( T+ d

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
4 _5 Y8 j0 B: r        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
5 ?' I7 a. ?- v- [$ _/ G};




class CUPnPImpl
- C7 G6 I8 Z' H" N) Z7 Z0 ^1 ?{
7 a, t" [" e. F6 Lpublic:
" h0 e3 f8 c1 u- I' {% |        CUPnPImpl();
        virtual ~CUPnPImpl();
! C+ Q9 k, @( d( {9 U* Q2 w3 K        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
: f/ `7 p& k9 e% D6 M  K" A8 h, B                UPNP_FAILED,
# G2 n/ {, z+ w1 `7 }                UPNP_TIMEOUT
        };

; t+ Y% {& ~1 r8 U
' r% o: j: ?+ W* m4 b  Z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
5 f8 K/ m6 ?+ ^2 O7 h- b/ l2 f/ n, ^        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
8 r! i) S; M* t% m9 F! {        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


( w8 {* W* @6 z6 h2 [' u1 E8 z! l        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
, N' N) X0 p% b6 x        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
3 x3 M3 v5 Z7 i        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
! {' C- J4 [7 \; N$ A# A" J
% X! A  u2 F2 N
// Implementation
" S& P6 v# b$ n. h1 [protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
5 @) b6 N! w" w/ ?        uint16                                m_nTCPPort;
0 c; z- ?; K$ t; H7 s* B* ]        uint16                                m_nTCPWebPort;
4 C8 k+ o% s2 g, u        bool                                m_bCheckAndRefresh;

$ I& Z( y* o. H7 U2 E( M  i
private:
) N( P  }, [" |9 m' T% U        HWND        m_hResultMessageWindow;
" y0 {" }% ^) ]+ h0 q9 d% u5 q+ z        UINT        m_nResultMessageID;

; L! N  i% s/ X1 _8 H) Q
};

! d, T/ [( t: x4 m+ c1 h
, v3 f) X2 i: j/ K// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
- i2 u0 @2 C3 `9 }3 E1 x, G        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


0 Z7 A1 r- G4 d4 w/////////////////////////////////////
% U  s- Y! \  @6 c( m//下面是使用windows操作系统自带的UPNP功能的子类

' N, K! L$ k  P% Y; ?
/ A: H( n, p9 J0 N: G#pragma once
8 j4 H1 s" o5 V3 P#pragma warning( disable: 4355 )

* {2 i8 r7 u0 W
- a, v1 l) W( e. }+ M#include "UPnPImpl.h"
#include <upnp.h>
+ C. f% @8 O! p6 _1 m% y8 T#include <iphlpapi.h>
' F8 R2 t- Z- k7 \' J8 p5 W#include <comdef.h>
7 J% ?# q0 S3 L#include <winsvc.h>

' Q6 y8 J* G" @$ l1 N
& U2 W0 |3 j( E#include <vector>
: O3 h% _( x) l$ M#include <exception>
#include <functional>
3 m- p4 A+ x& E4 e$ u" b2 A; X/ o/ S

3 }, x6 I9 Z/ j0 O

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
3 W8 o; E* Y9 Xtypedef _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;
$ s, j- b: p3 q: ~* L6 l( htypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
2 E& {& h! g' E- z/ ]typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
0 S7 o. a1 Q6 y7 I' k, V$ Ntypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

$ s% C3 b5 q! t  M( y; e
typedef DWORD (WINAPI* TGetBestInterface) (
7 T( y% h0 t) y! U7 u  IPAddr dwDestAddr,
, u3 t. u9 A4 `1 O  PDWORD pdwBestIfIndex
  q7 t5 K, w9 W, _# R1 w);


typedef DWORD (WINAPI* TGetIpAddrTable) (
$ H% F% R; H9 c  PMIB_IPADDRTABLE pIpAddrTable,
' H; ^7 p9 S9 m' z, r2 x, Y# W  PULONG pdwSize,
  BOOL bOrder
$ Y  I+ ^+ X" B. e0 G% t% h);

7 H0 D1 C: B5 C& a
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


7 A  e" r5 g2 g' U* i8 y) f" uCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
& k" y8 k: M% z  V- C9 b5 g

class CUPnPImplWinServ: public CUPnPImpl
/ o( g: z! H. a# V# q$ p{
        friend class CDeviceFinderCallback;
; A9 u- F5 P  @- i+ i! m: v        friend class CServiceCallback;
' I4 f2 m/ S% @# N: ]$ R# s$ H1 O6 u// Construction
1 |( T( N3 ]1 g& e: {& d" F/ j+ _public:
        virtual ~CUPnPImplWinServ();
" J6 N+ j! X1 E' {        CUPnPImplWinServ();

+ n# B4 Q: R( |2 H$ }# o) h
* M$ E* B% R  \+ S7 ~        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
6 M  r" A" H9 t3 @; [+ a        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

# ~& \5 B1 N; ~! Z) F; a+ Z
5 a1 {, `" F: w  m% y  t( e        // 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; };
) _2 [6 i/ _. a1 H6 f
# S& H  ~7 [6 Q% y
" D1 n" C1 o/ r6 \3 bprotected:
' N/ O' g& v/ E        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
4 [" l, J9 ?8 L# v% z, l! i2 c        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
9 J; K7 J) M' G4 m/ }        void        Init();

3 T( k5 i" W0 U: D3 ?* K' I/ W
        inline bool IsAsyncFindRunning()
) E2 f9 ~, z: r# O- f        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
$ ?4 M7 N6 {! ?& E0 R: }" v                {
5 C5 h6 Y1 Z" D) x! c0 M                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
  H5 L8 J! Z" R) P& \8 J* V$ j# O7 a                        m_bAsyncFindRunning = false;
8 I3 U* d1 U1 k6 B1 g! J7 w                }
6 W& X0 x# i( J( U                MSG msg;
7 D: }) b$ V' Q- q                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
. ]$ f) p8 }/ S% K  O( ^# S                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
& z% t$ K4 N/ A! U6 C        }
4 X7 K) X+ G4 H: R/ ~3 k

  V8 x; N, j" f* b( }        TRISTATE                        m_bUPnPDeviceConnected;
) ?9 Z* H- W$ v7 |$ {8 W) {/ t
9 y; b; T$ }  }% u* ~
' ~' C2 k: y6 D) S// Implementation
, C6 L3 B( I  {- \' S* L        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
& n# y9 _( M8 n" }% e8 e! u$ h        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
3 ]9 v9 N! ~% s$ F- K; B; H        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
0 N6 ]. L3 a: A$ \1 d/ z/ D        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
4 j7 a2 y. @! K+ R9 F5 C
6 y! p8 T9 V7 a3 K. A6 Z
        TGetBestInterface                m_pfGetBestInterface;
+ B, `+ L3 N2 y# Q) K: j        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
- q* B* v( C3 ^9 c
  X" V- b1 |& M: v* f2 ]# x5 N
* v  \' n. u5 d. X        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 ^( Y$ L5 X) Z; I3 j                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
4 z6 e8 Q# a/ P3 z/ ^! j+ Q                result_type operator()(argument_type device) const
: Y4 g8 g) n2 S  u2 I6 t                {
9 S" h2 g, R8 h6 V# f5 ~- x                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
$ ~3 L/ S8 e) m$ U* ^; @
/ _- G( g' U& J8 _) K6 w, S
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


9 ~7 b- r% _+ T9 D- m2 F# T# c* c                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
  ~3 z5 L! |7 ^0 z3 Z: A        };
6 M$ X: T$ O! W: @1 s# s& f       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
: _2 f( p+ r  k, M/ S# P" ?        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
( S+ d: l7 g* ~0 Q        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

- ~  g8 a# L$ d
4 ]6 @# Z7 j- ~9 Z: t1 l  N        // Utility functions
/ U% S1 ?' D5 u        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
5 R- T( O% w( U& t5 r, i+ ?) q* c        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
+ R- M" L* G4 r; F7 `+ ]6 y3 k        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
6 p9 n) u1 [, H' T3 w        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

( t9 ~. f$ K; ?7 z; r* v8 N0 \3 h
6 v/ k3 `+ s& m  X// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
; \& M# u  I) C2 M        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
1 ]/ j  }5 @' W& Q, d$ R7 X4 x9 u        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
8 _6 h. o2 q7 }6 V" @3 f        ServiceCallback                        m_pServiceCallback;

+ m% A/ o; j) C- I
        LONG        m_nAsyncFindHandle;
) S7 z' c5 d+ q0 R+ \% B+ k: n        bool        m_bCOM;
0 M! w- K( ?/ [4 `8 D2 ^) ^        bool        m_bPortIsFree;
) i2 n5 d7 Q9 |$ D        CString m_sLocalIP;
        CString m_sExternalIP;
% o, A- S! D' X, Z9 x/ E, w        bool        m_bADSL;                // Is the device ADSL?
  W2 A9 V  b8 t/ u' T0 }        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
6 g4 z3 ^! z* p' f; Z5 m        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

3 h$ Q- ]. x& y) |) l2 l. n; ^
};

7 p+ j0 _: }+ p5 H9 C* |6 }
6 H" I9 W' J9 P* Q# Y. f// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
( u0 Y6 ^7 I# z! k0 x. V# V        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) \* A" v1 n# F; y( t6 d- l1 y1 z                : m_instance( instance )
* l4 a9 E" N5 }/ u8 e        { m_lRefCount = 0; }
# t  R* A! G( P/ C

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
. T5 v+ U" P  }$ ]: D   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
7 ^. D) d6 Z$ _  Y
& _# T; H8 s1 G# U
+ R, h7 |  G2 r( n& `. \$ H// implementation
% @4 W+ q. H# z( t" xprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
& f' k! G+ M' j4 z
4 E7 B( D- U  D6 y3 C
. z+ r- q3 {$ r7 Aprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
3 R9 {. i% r5 i" ^. M' ?" q# [};


// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
# P) V5 @( m8 i0 p  u$ e9 Tpublic:
        CServiceCallback(CUPnPImplWinServ& instance)
9 v1 X* }; n4 G1 C, o' q0 ]                : m_instance( instance )
' k+ o  K7 V! Z. _* J. M        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
! [- I" M6 {& v  q, s+ C   STDMETHODIMP_(ULONG) AddRef();
$ @5 h; D& ^5 z3 A. e$ |. {, J) m   STDMETHODIMP_(ULONG) Release();


// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
6 _1 y% x+ r+ @+ u: e, s        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
* R# {- `) M  K5 R- v        LONG m_lRefCount;
9 O* v9 D3 X7 x$ f- k' V};
4 Z" a  G$ |/ y/ B

/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
" y+ Z2 Z6 w( Y' s8 Y% yCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.