UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   1 Q3 g& V' U5 I" P
3. 
   # s8 U5 ?8 U" ~5 C" F- y
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   1 b) w9 q3 a" J) d+ b* `0 g& q
9. {
   
10. public:
    
11. typedef   enum{
    3 |$ M( H" c2 q8 {4 w/ U1 ?; h& @
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 x" ?3 I! Q" Y0 D7 M* z0 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
    5 m; D; m2 G5 S6 W% `% e3 u/ O2 f
17. }   UPNPNAT_RETURN;
    . \1 p" y6 R1 H' d, m
18. 
    ' }! o( d6 o: ?, _& [( d
19. typedef   enum{
    5 c$ n' n( w2 W, O; I( B
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    , Z& X- E9 i$ \# }1 [* d' L
23. 
    5 H; K1 e$ H9 z/ J% `$ O5 J, R
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    $ _" C0 Q" t. |2 ]' |' d
26. WORD   externalPort; //   Port   mapping   external   port
    0 v4 n; k0 }. J: R& Y+ q4 W
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    0 u5 O/ c( c' F
28. CString   description; //   Port   mapping   description
    9 ?) R: R+ J* Y6 F* O6 O
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    0 y7 T( n; t$ X2 P& W5 b2 Q
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    5 I5 @0 M) ]3 f( G2 Y: F
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
      h# P% A! G0 ?! U
42. 
    . m9 j8 q0 S, u( f, C& `
43. protected:
      h7 A$ [, j0 a( w7 J9 z" V
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    2 M3 ~3 Q5 _# d' y
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    6 n" [6 H  k' L! y# S- J$ F
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    6 r2 E8 j' w; P' }8 M0 @& o
52. 
    
53. bool Search(int   version=1);
    5 \/ S  ]% s# n( C2 e0 x
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    % y) v! o* N( b: \/ A9 x
61. CString m_name;
    + ?" N, ]! v$ i& E+ R3 }$ A
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    & {0 q3 F! p9 V3 E; g
68. 
    & ^3 e3 L6 C, e+ K. j- D
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    . d% I' I5 U7 K( y$ o2 \) S" t
71. 
    6 M' e' |9 Z4 @; |# [: ]/ ~8 u
72. CString m_slocalIP;
    ! J% [3 @* d, F, u
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    4 i! r% g; f: y5 ~/ E8 U. P5 H+ P" W
75. 
    2 x2 B. g. R" C5 S
76. bool isSearched;
    & s, l" c+ j1 q8 J
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   * i! H( s5 {$ p8 D2 J% W' D$ o2 l
4. #include   "upnp.h "
   ; {9 @9 n" q8 V# B9 B7 B7 J$ Y5 r* v
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   3 v1 d6 c4 u" [
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' w6 H/ z8 R  ?  Z
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    " @( c# ~* d4 q1 B) [
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    4 W# ?) m  |4 q. Q( P
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    1 I" Q# j" _$ E3 T3 Q& r
16. const   CString   getString(int   i)
    - z& ^) u/ Z7 ]
17. {
    9 E* C4 D8 F+ Q' ?# G* c  k" Z
18. CString   s;
    ' \: @; l, r! M6 D9 x, S$ S* {$ N
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    0 i; L$ P9 B  N; i! i
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    . _/ _1 r: w  s! s" ]5 E8 ^( W
28. }
    4 Y; O- _( B: n. r  Y
29. 
    , `2 [, U" I3 K  `$ X. [$ _
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    4 l- ~3 D( L; R. |% Z
33. }
    7 P, \" y( k! Q  o; _7 J0 Q
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    # e- {" f: F/ ]' c6 R5 R( k
36. {
    
37. char   buffer[10240];
    
38. 
    , z$ d% L6 Y" |( u9 `3 d/ D
39. const   CStringA   sa(request);
    - U' A# K) V  O, I9 x
40. int   length   =   sa.GetLength();
    ! ?5 P2 I. x! M6 ~1 Z
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    8 s3 ]- E) V* x, _
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    / A9 ]" R- [1 s, x
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    0 K, |  ~1 \  o; R  p, n
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    : e: w+ y& R$ Y! Q, L4 l
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    - Q4 B0 c+ a; z6 V1 m" F
54. int   n   =   send(s,   buffer,   length,   0);
    3 z& {, a6 X3 T( K1 S
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    6 T1 S. _% l  e2 |  k# H
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    5 |' y+ V0 S% g# M! c8 M( B# y8 H
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    2 X% {+ a& k8 k: k6 R
63. return   true;
    * W: E, ?2 a/ \7 F+ w
64. }
    % Y# Q7 [. _' ~% h+ V6 w; P
65. 
    # Y# O( N+ g8 ?$ ~4 j! T
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    + T( f7 r; `0 F2 s
67. {
    
68. char   buffer[10240];
    : Q2 D+ i9 _6 Y7 X/ S4 L, M
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    : R# }6 ~2 F' K& H
72. strcpy(buffer,   (const   char*)sa);
      k0 ~0 j$ j( E! N3 F" k: D* @
73. 
    
74. struct   sockaddr_in   sockaddr;
    0 z+ Y" `/ {7 R3 A
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ F& t3 o# [# f! B% p
76. sockaddr.sin_family   =   AF_INET;
    9 U0 ~7 O; Q2 |# T  I4 K, m* c; B
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    0 I( Q! E2 X0 p. b- c1 [2 W  w* V3 E
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    # X' _5 m) B6 L4 m
82. 
    / U0 O' Q7 L+ X
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    / [8 d4 R) H! i" {. o6 X. a* }
84. {
    
85. int   pos   =   0;
    - ^8 H0 P6 W' Y
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    ; G, b2 ^. L& p1 O% Y' c
89. result   =   response;
    : L! s, H6 f/ c9 ]
90. result.Delete(0,   pos);
    # A1 U: S# k& B6 t# J* Z5 a; }* w
91. 
    + z3 V4 G2 [2 ^0 |+ j
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    9 b1 D( b9 q- `7 O6 t
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    . r  s8 @% O( V( ?% E& V
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     / A! X6 X' h6 F% W- a0 h* c9 k
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     3 V: @* q) Z% o$ n) n- b5 s  `8 F& @
105. int   posStart   =   all.Find(startTag);
     1 C  r" \1 r6 I4 {
106. if   (posStart <0)   return   CString();
     
107. 
     + k1 e9 Y9 g& r! s& [; L
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     0 k& ]0 Q- L0 m( L( G! R8 F' l: u/ y
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     : j  _4 `+ o$ O+ V
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     ' Z, G/ u  K$ h
115. :   m_version(1)
     0 M  c5 q+ r" x3 C/ K: q( b
116. {
     
117. m_uLocalIP   =   0;
     * y) a6 w! D# [; V3 u3 Z
118. isSearched   =   false;
     ! f) D( k' R6 Q% w0 D' x: V8 O
119. }
     1 K! G5 ~/ {. r& R- U2 G! l, p0 J
120. 
     : V+ Q" q& M. l* K) C
121. MyUPnP::~MyUPnP()
     
122. {
     ! ~" ^9 f, V$ h8 a; `
123. UPNPNAT_MAPPING   search;
     6 y# y& S' C/ S& h5 t) J3 Q
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     . ]* ]: a# S- j, ]5 w/ m4 M
125. while(pos){
     & B3 |5 O( M& k& X2 `
126. search   =   m_Mappings.GetNext(pos);
     8 r1 e8 \$ M% b! P/ |3 l
127. RemoveNATPortMapping(search,   false);
     
128. }
     3 N1 [  }7 s  B% E' j
129. 
     ( N5 [0 u" _, t* F: q- i. N
130. m_Mappings.RemoveAll();
     $ E# M% i+ X& h) ]. p  o
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     / Z7 Q  C9 o8 o
141. {UPNPPORTMAP1,   _T( "service ")},
     + |$ b# T" g7 C, I) C
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     0 {5 E! K% m% i& u. j
144. };
     
145. 
       V! B3 O/ u  F: x: B  _
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     / X& Y: l% U: Y) J
148. ioctlsocket(s,   FIONBIO,   &lv);
     - q6 ~1 ~; f4 B3 I
149. 
     ; M: ~4 s6 g. z# ~* g! Q
150. int   rlen   =   0;
     , l) a5 r0 g: L: G& @& k/ O
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     1 r4 A# a6 i$ R1 h
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);
     . g  ?  N4 s3 t' \
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     " }  _' m! }4 I  A6 q, L
159. }
     " _/ Q9 m* r! z
160. }
     & I" `& _7 h3 k5 t+ D
161. 
     : }5 _, n: T; ~: h1 k
162. Sleep(10);
     2 a' o. \' a5 Y9 g' z
163. 
     
164. char   buffer[10240];
     8 k2 b- ?2 [. |8 o: ^5 l! L
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     7 U9 W. r, m5 R, ~9 y0 c/ D+ W
166. if   (rlen   <=   0)   continue;
     + c; w1 Q3 b/ ~8 @# F8 P% M
167. closesocket(s);
     
168. 
     # \; c% i: w9 d, z1 ~# `
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     " z+ i  P9 d1 L% m# h
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ) R& f$ G- e) X8 f/ r
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     # M1 h  g9 R5 W% l, ?
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     ) i4 k  T) w& S7 l, x
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     * J& g  a$ D7 z1 G" _$ o2 w
182. line.Delete(0,   9);
     
183. m_description   =   line;
     7 U: r6 ?( P! @4 p1 ?  V( ^
184. m_description.Trim();
     . b) [( g& Z& a
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     ; O# s$ P9 Z3 P" X5 D9 R( M
189. }
     
190. }
     ) Q! W# ]; c- _% d5 Z# k
191. closesocket(s);
     
192. 
     
193. return   false;
     9 M! s. B; B  [& ~5 g6 }& U
194. }
     . ^, v. B  I. ~  x

复制代码

天云上人

以下有关upnp的接口来自emule，
: i' J0 N& ]% Y; j! q9 }2 o

  l9 J/ H* r7 E///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

! V) n! _  o$ }3 y
  G+ W5 T" Q, I#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
4 L$ o2 U6 c  B        TRIS_UNKNOWN,
/ ]7 p6 B& ]# r$ m        TRIS_TRUE
};


& h5 w: A: y  t7 x4 B9 _enum UPNP_IMPLEMENTATION{
6 G' l+ S/ ~/ N  q3 J' `& ]/ `) ]9 ~        UPNP_IMPL_WINDOWSERVICE = 0,
( i  Z: b1 p+ H9 v2 p8 N        UPNP_IMPL_MINIUPNPLIB,
& u$ @5 X( m8 `, N6 |4 m        UPNP_IMPL_NONE /*last*/
};
/ ^' N1 Y- M, l  Y+ T
6 {9 N' B8 Y, X  h

6 r, N  ^' P: P4 P; V: k+ f1 P: ]" B
, E/ r& n! J" [+ g' j( |class CUPnPImpl
- c) ^( f, B5 E6 z7 P; Y3 W  F{
6 P9 r- C3 Y# X6 S5 K# ipublic:
/ J! r+ I8 W  {        CUPnPImpl();
# ~/ z8 ]! V3 ~$ P) n4 g8 ]        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
' L# _/ q8 Z- G                UPNP_TIMEOUT
2 r4 N) c! F3 d" i! k- X  ?9 S        };


0 H: O7 `- x2 e        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
' }5 R. t7 a( v4 D5 b, W        virtual bool        CheckAndRefresh() = 0;
& r7 X& e0 X4 [' e, t) s        virtual void        StopAsyncFind() = 0;
) x+ e# k9 P4 A        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
1 X: w5 v) d; m& ~        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
+ h( t) T' J& L4 S        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
1 v) \8 F# N7 `9 v0 [; T        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
1 }8 K0 [6 O2 Y3 L3 Q
* o) U: r+ k3 l: o+ U" m6 y) U* @
// Implementation
& W# a# E+ H! m! K$ _protected:
6 _. y8 `% ?: D        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
* |; P, D9 E' z8 r; E' S$ U' {        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
4 L  a% ^# t! T, ^" {6 f        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
! H: J* j- ?3 O1 c1 M/ i

0 y% y! N- e5 i+ |. {$ n0 pprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;

; X) a1 L4 r$ ]6 {- D  x
};
" U, x( D1 Y. _7 V- M4 m8 }- v

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
8 t1 p& P, `! I, B' E{
2 a9 ]( ~& j" G! G6 zpublic:
: z9 P. {  O% E8 E7 r! q+ X" S# x        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
6 _% L! o0 _! i/ W, M  N        virtual void        StopAsyncFind()                                                                                        { }
  @0 w0 q, e* _        virtual void        DeletePorts()                                                                                        { }
+ A( h+ x6 B# e8 U/ Q        virtual bool        IsReady()                                                                                                { return false; }
1 i0 ]' q' g, B        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
  Z  f: a% x  T; b};


- F# S4 j9 @6 F; p/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类

) N+ u# m! b6 V0 `
#pragma once
. G( N% l( K2 v0 Y7 c#pragma warning( disable: 4355 )

3 P& X! M1 ?" }2 h2 }
#include "UPnPImpl.h"
1 z: L, s/ X4 {$ H#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


2 F! v( L( U9 `9 s#include <vector>
#include <exception>
6 I8 ?: O- D" P7 L#include <functional>
4 h0 c, P0 q9 @


: p- E! q5 L! `# h" Z2 U+ ?$ `2 z7 ^3 p
% d1 f/ V5 K9 Q" Xtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
5 A, @$ _0 q$ f( R- J& ztypedef _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;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
, a% x1 O2 l1 [* V- W

  }) k5 f+ C; Y& J6 otypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
/ r) A/ W# Q$ _9 `' E! ?);
9 z1 I2 C% ]; T2 q
, k$ a2 G) ~/ G/ X* S# E) g
: c0 e9 }. E/ e" `# v# Mtypedef DWORD (WINAPI* TGetIpAddrTable) (
7 `3 o" s8 E& z* [  PMIB_IPADDRTABLE pIpAddrTable,
9 y4 F$ W5 W, c; X# O  PULONG pdwSize,
6 T: q7 {9 K# w; N. f- \/ D" G  BOOL bOrder
5 u% _( Q9 z4 i; `( H);

' p; i6 V* P0 A8 e/ F9 k) b
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


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

3 z3 }( V4 {, ?# P% A3 V
5 t7 }7 a. s$ `class CUPnPImplWinServ: public CUPnPImpl
{
  {5 L5 ]( r" O+ A- j        friend class CDeviceFinderCallback;
2 {+ z" W: w4 Q        friend class CServiceCallback;
0 C  J1 o  E% h6 ~+ N  ^, o// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
4 r/ b2 l4 K9 V) y' `

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
; W  N1 H+ l( Z- ~4 V; k  m: ^        virtual void        StopAsyncFind();
3 `* {+ f% E' g+ D2 `% w8 P4 T4 N        virtual void        DeletePorts();
" L, Y6 b) w' n9 {7 }        virtual bool        IsReady();
7 h+ ^) r3 z( m! i        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
! T. f6 H7 s% l9 s) `

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
  x, \" a2 ?7 S6 d        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
$ b2 b6 x& l) r5 I        virtual bool        CheckAndRefresh()                                                                                { return false; };
% B% R  N( N' w2 F4 ~9 j$ l8 r
# i- F: h4 d1 f5 \" O
protected:
5 V7 a. F$ ^8 w2 V        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
: k( _7 B+ \  y. t        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
5 f! t. Q9 f9 ^  D4 ]8 C+ ?        bool        OnSearchComplete();
        void        Init();

2 f7 m( e6 i& n+ Z: h
        inline bool IsAsyncFindRunning()
' r/ B5 I' z% M9 j* p        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
2 `% c  n$ r0 F4 Q, X9 ~# g% _                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
                MSG msg;
) i$ p7 {' O3 w                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
8 S8 m8 k* R. e/ e! ^+ ^' T- ]1 N. E                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
' z# I( E' n, [3 ?/ H# x/ X& Q                }
                return m_bAsyncFindRunning;
3 J. p2 e2 a8 h  U3 O' k! G: p% k        }

' \  U" s; X, W) T' I# N
# x3 x5 m6 a. T8 O) Q9 r' c: E  f        TRISTATE                        m_bUPnPDeviceConnected;
2 d$ t7 [# e8 g/ f5 G1 o, o
8 j5 v* b$ o# f- P' O
# U6 W5 P% j8 o/ ]7 L; S3 t// Implementation
- }- A1 K$ g5 A5 U        // API functions
* d+ H2 V: B$ ~# J7 F# \. Q* F$ r        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
$ [" h' h' ?3 B4 Y" H5 X* U        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);

; Q& w" i. g7 w% t# b' [: G
" Q' p  E2 r9 y/ d& N+ f& t        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
1 C& h6 V; G4 Q) m& a4 t        TGetIfEntry                                m_pfGetIfEntry;
0 _$ ~6 g" k8 w! r- S. R: x+ _6 k  G1 [
3 F. {5 |. X4 t" Q/ ~7 |+ k  B6 k0 `
        static FinderPointer CreateFinderInstance();
+ |, t( w4 i% b3 s        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
% n+ H4 I2 Y9 L5 r2 W/ R7 k                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

5 Y/ V3 x* }# I; t  w( F6 \( ^0 K
2 v; U) H) Q2 E  h3 C  D                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
2 J/ ]: Y# v1 n& G
0 M1 i0 K* R2 k$ B3 o
) E' J3 v$ I" K7 I" F' c, R                        return wcscmp( deviceName.m_str, m_udn ) == 0;
! d4 `& P( e- }% s! s8 T7 o                }
6 A& Q5 g7 R5 f8 g- h                CComBSTR m_udn;
        };
4 q- l% h* L" l$ h  R       
! a9 ~0 Y1 M# h" q# k) U9 M+ Y        void        ProcessAsyncFind(CComBSTR bsSearchType);
+ x, t% r. G$ }7 |        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
7 b, U4 f% f% k2 e, ~( U! L        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
* U5 o; r6 ]/ F        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
, d7 U# W2 }  I$ G0 N) t        void        StopUPnPService();
/ n& o1 I& Y8 M" \6 o" R% k4 w

        // Utility functions
% _  w* ^0 }9 B8 l; S3 {" G        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
, R: e: e# g6 j4 W; T3 Z) Z+ O        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
" R) t. ~% g8 L        CString        GetLocalRoutableIP(ServicePointer pService);

) h, f( F$ M6 A) N) u5 e
7 H; t6 G, J, L) ~7 u// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
; J% T8 K# @3 N- D        std::vector< DevicePointer >  m_pDevices;
  j* h* m6 w6 z        std::vector< ServicePointer > m_pServices;
5 I) r+ _: T$ v  z        FinderPointer                        m_pDeviceFinder;
$ O* I6 i/ S! K        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
9 g$ [0 g. e5 e; ^' ]+ q

        LONG        m_nAsyncFindHandle;
$ n4 K) L/ Z& u) V" A! n. [        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
. O1 O9 C+ f+ |        CString m_sExternalIP;
3 ]# E+ X0 T& z: v8 z5 l        bool        m_bADSL;                // Is the device ADSL?
2 H/ J: }# S" n0 i        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
& [* w3 }0 ?: o& R        HMODULE        m_hIPHLPAPI_DLL;
" |, j: j( K) P* G$ z        bool        m_bSecondTry;
, Y: i) \/ {. B+ Y* H  c9 R        bool        m_bServiceStartedByEmule;
: F. s2 j- ?8 X" j        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
9 h: E. o" \1 q9 X
( M9 h" F  V3 \# t; {6 E
9 W" P# {: {3 w5 B};
$ J8 g: ~/ P3 ?

& ~7 i9 _' B& y8 \  K& T6 `7 {// DeviceFinder Callback
$ z: H  G. m! B2 W) B5 O8 O! C; |, Dclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
- W; y. b2 n0 \# `# k% O        CDeviceFinderCallback(CUPnPImplWinServ& instance)
" Z/ r0 q) D( K' i% |7 ?                : m_instance( instance )
+ {( ^7 a; M9 {        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
& B; t" U3 ^$ ]' f

// implementation
private:
9 P. s+ O& j" {! j) [9 C        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
# P. s0 M9 _4 d: n        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
: X) `: y& \$ Z* O5 ?' u3 _" n

/ T- L, A8 o' T2 }private:
        CUPnPImplWinServ& m_instance;
* `+ @+ l2 |' Z, ?+ c* o        LONG m_lRefCount;
};
. r7 Q- n* c9 p- E0 V! M& |4 y
7 |, p& Q; v( |4 p( @
// Service Callback
' W# Z( `5 X$ D$ o0 M2 cclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
& J; t& [2 i4 y" C        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
" E  h9 C! w; K# _' p8 n2 S. w   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
6 g7 ^9 y5 [% O  S   STDMETHODIMP_(ULONG) Release();
$ B. J0 h' C3 k, I
( o3 ~, w9 c4 M* J5 t+ B
6 y$ Z2 z% C7 s0 T// implementation
" u4 l, l8 ^' F" Cprivate:
: P8 G$ A, V+ Z5 L4 N7 ?        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
! [6 ~2 H- I/ B* v; v
, F. L! f$ E8 p; V: b
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
; e/ V# K1 L& e! o5 X7 ~0 a$ A( R};


- a- b  |1 R+ h5 G: m* ~( [* t/////////////////////////////////////////////////
+ _" Q8 ]) T0 O- {; u5 ^0 ?+ I
3 n. x) M! C! K. l$ i/ y
使用时只需要使用抽象类的接口。
( @0 k+ z5 D/ n& Q. ?: _( VCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
! T: F( m2 U. N0 E6 s& ACUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
. R: |  ]9 g8 v" d7 GCUPnPImpl::DeletePorts删除端口映射.