UPnP

天云上人

/*uPnP.h*/

1. 
   ( I5 [/ {) I( j3 j1 k
2. #ifndef   MYUPNP_H_
   
3. 
   + D1 p, |& W3 ^$ h
4. #pragma   once
   
5. 
   ! [& b6 D8 m; Q. ~' w
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   3 \; m: ^7 x0 f/ _
10. public:
    1 [. d7 s" H& N3 `9 ^5 w9 z
11. typedef   enum{
      T. b9 t1 j# ^+ l; H0 q
12. UNAT_OK, //   Successfull
    ; W  W( F+ u' A, ~* U' Q7 t' B
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    8 v- }# T3 u/ D! _
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
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    7 _7 b/ P) E8 C, w& U! S' y
21. UNAT_UDP //   UDP   Protocol
    9 X( U' x; O2 _, W  S
22. }   UPNPNAT_PROTOCOL;
    7 ?7 V# m, L1 T5 t# I8 s
23. 
    7 v2 h" T+ ~* D
24. typedef   struct{
    ; R4 p' H- z/ W/ q+ Y7 `2 ]
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    ! ^! B9 e- ]: B3 ]/ u) y% `
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    8 C6 f3 v: ~* i) i& ?+ G. q: Y
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    , Q* N5 U+ j7 k7 M
30. 
    
31. MyUPnP();
    : C2 c0 H/ _( j  M
32. ~MyUPnP();
    
33. 
    ' r  T9 t2 z# F- x. j: p
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    + e) N) G  J$ x1 M6 F
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " k7 Z6 ~8 ?3 G6 _+ s
36. void   clearNATPortMapping();
    # ~9 u6 k3 D* g7 V; b: I
37. 
    0 f; k/ ^1 V, R+ ~
38. CString GetLastError();
    6 g$ H: t4 A& ?& }
39. CString GetLocalIPStr();
    ' I# T  e) `% u9 S
40. WORD GetLocalIP();
    ; v- s# k9 ?3 r/ `; r
41. bool IsLANIP(WORD   nIP);
    3 `7 g( y8 N( D/ m7 `8 B4 L% W
42. 
    . [* W9 p0 m. i0 C' w
43. protected:
    , Q) ?% S( m$ g/ O
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    - [- l6 l# \7 L
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    + i- J5 y$ e, O6 q
48.       const   CString&   descri,   const   CString&   type);
    + @' U5 X: c3 T
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ) B3 ^" _; i& P7 R! V
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    . j. r6 |- Q8 V! m% Z+ j" p% V" P0 I
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    : c! A# G% }5 |: w4 U
57. 
    & v3 ~, J7 |' v! n+ @0 v
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    # Y% q& M& b0 P2 i* c7 o, I, r
60. CString m_devicename;
    
61. CString m_name;
    3 j" m, w" S6 i2 x9 y! R4 S
62. CString m_description;
    
63. CString m_baseurl;
    7 Y4 u* g# O; Q, }
64. CString m_controlurl;
    % D, O- D# T# Q% q, ?% W0 Q
65. CString m_friendlyname;
    4 T8 M4 B; L, s6 [9 y
66. CString m_modelname;
    4 L$ S; H, q. X0 V7 T  v
67. int m_version;
    
68. 
      p* Q) t$ U# L8 L5 q8 ], S
69. private:
    2 z, i3 u3 e2 ]5 J$ {
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    / C, f/ I8 D0 R/ b
71. 
    7 g7 I& @) n* L
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   $ d2 h$ h! C6 [! C) q8 {
2. #include   "stdafx.h "
   5 H! e; |) l' K7 \1 l: F1 S
3. 
   / y, b$ s" u, }% b; N
4. #include   "upnp.h "
   
5. 
   * e0 s# u6 j, }8 l$ \6 b
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   " L9 @- r: K1 e6 t
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   8 m, n6 c+ _9 h0 D$ \
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   0 b4 E' D# ^5 O  s9 O$ i
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    " p6 _& u: s, d4 Y1 }* b) k+ t' _
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    " n$ n$ I, r1 G, W
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    * n2 D6 c! \" S. X
22. return   s;
    : m7 ?3 I' ?( o& B
23. }
    # S! L# P# F- e! U+ Q. R* w' F& w
24. 
    & U- t, M: ~" C: E$ r: M' H2 U
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    4 z9 z1 k9 h+ G; G
26. {
    3 V. W1 k( f6 N
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
      p* J) z7 E, I
28. }
    # @/ r4 Y9 g3 |. m0 x( N9 C
29. 
    4 F- p7 _+ u2 K$ F
30. const   CString   GetArgString(const   CString&   name,   int   value)
    . L4 i' g8 v1 _  F1 s8 A* p2 s: f" N
31. {
    * Z$ _' `# T; T8 A/ W
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    9 ?& d1 Q& ^; [
34. 
    1 g4 a$ |. P4 O. }
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    8 p. \5 z8 l& ^# \: Z
36. {
    
37. char   buffer[10240];
    
38. 
    
39. const   CStringA   sa(request);
    , X/ X# ?( a( u& y6 p1 `0 @: R3 i
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ( e7 Z7 a4 Z% \; ]; y+ f% ^
42. 
    ; a/ \6 {3 D$ D+ c; O
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    ( N3 b: H6 B4 l' p! p9 k' I) b% p
47. sockaddr.sin_port   =   htons(port);
    & [' W0 {* m5 o2 m. c& x
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    8 h) Y; c; N$ G3 P
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    1 h3 x/ a3 c& Z* c4 R, d5 E' k
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    ( Y3 Q2 V- T1 l1 L$ A
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - ~2 ?5 O! l* Q4 o5 b
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    4 V7 B8 I1 I* G- k+ `
60. 
    4 Y. c+ s7 f+ G' t0 m% ^3 e/ C1 u
61. response   =   CString(CStringA(buffer,   rlen));
    0 _# P8 x( U9 N; C
62. 
    
63. return   true;
    
64. }
    * V. k2 p8 T+ f
65. 
    5 Q# ~! k5 K1 ?; [* x
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    & x2 c3 E( l& K8 a. }! G1 I( b& X
67. {
    
68. char   buffer[10240];
    
69. 
    6 H' p/ I+ M2 @1 V8 Y: q$ w$ x
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    8 Y6 I2 h: M. ^2 \; T. P# V0 [
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    1 P% E7 W! s4 q( B* d" {* g$ }
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    5 Q5 l, ~8 I2 h' K  B. x. N; ~
79. 
    ( }3 n3 [1 d5 c/ D
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    7 D" ^2 ]( B8 t5 Z$ `
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    / q4 e4 \9 N. ]4 R
85. int   pos   =   0;
    + M* q& V( n! `
86. 
    : Z# R+ @8 M  R) ?5 [. v; d( j/ q! j
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    3 Z+ u0 X" t* k% ]% S# T/ ]
90. result.Delete(0,   pos);
    ; E9 @+ A6 o# z- K7 w9 J! |
91. 
    . D5 j; _) r) j' V) `) N4 Z
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    % |( t- u( u0 n6 y' P. n* F
94. status   =   status.Tokenize(_T( "   "),   pos);
    " l; q- e# L- {  x& q6 u6 `- X
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ' w, b4 l& n2 _) {- ^
96. return   true;
    
97. }
    ) q3 Q4 F! c6 C6 j
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 U) p) ?5 z; J; i1 _  a2 L
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
       s; G5 S2 l# i) T. P
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     2 f7 m- S. H3 Y4 i, R# h) _
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     4 ~5 i) o& C3 p
106. if   (posStart <0)   return   CString();
     3 b4 g' G) e4 j4 e
107. 
     4 L! `. b  B& P2 F( B" }4 t0 q
108. int   posEnd   =   all.Find(endTag,   posStart);
     2 n) L; E( C, ~9 h
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ' t- k  B3 r% {6 H- o" H2 Q/ o  n' P
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     * S3 ^: p. k) Z' O  y/ T9 t
115. :   m_version(1)
     
116. {
     6 y. V7 p2 l+ P5 O
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     # W2 f! t" u( N0 _) E# j
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     5 r2 B" L- O  u8 T% d) p5 O
126. search   =   m_Mappings.GetNext(pos);
     % O! J' \# e* U/ @& i2 d1 f
127. RemoveNATPortMapping(search,   false);
     
128. }
     " X' l% {: B* P, Y" K4 e
129. 
     
130. m_Mappings.RemoveAll();
     0 ^% G  n5 G1 _, z
131. }
     
132. 
     5 {! ~7 x+ u5 w/ _$ @6 Z6 c: D
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
       W. k9 S7 `7 ?! F& a1 D, h) ^
135. {
     
136. if(version <=0)version   =   1;
     / E( x) ^/ C) Q5 [- f! k
137. m_version   =   version;
     
138. 
     ; e- |( h" e% n# w3 H
139. #define   NUMBEROFDEVICES 2
     9 G7 _# u- p: O
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     . ]/ D1 n& t2 s& Z8 E1 F0 [
142. {UPNPPORTMAP0,   _T( "service ")},
     1 _& N+ R) Q* y2 r
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     ; S# b( C! `# {4 ?* |; J- X/ Z) R
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     " m! L4 P/ h* M% @
147. u_long   lv   =   1;
     * s5 T2 X) p* R* W4 I9 O$ f7 X% Q% L! L! \
148. ioctlsocket(s,   FIONBIO,   &lv);
     1 E1 x+ [. D: R2 X
149. 
     * R! B+ ~3 r" A5 U0 d9 ]: R
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     3 B8 w+ Y( D, ~( h+ m
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);
     . c) w1 p6 X* p" M7 x' Y* d* k
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 "),
     ; V2 J( e3 V) c# Q5 w4 X& T
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     * l( e3 _/ H* s7 `( j1 X
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     ) v. m, k% I+ Q2 R
167. closesocket(s);
     " O; h* B" E9 J' `+ J* W6 {7 Y
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     " G2 ^/ f; Z# y+ n& T% W0 }
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     ' H& r9 Y9 a2 k8 ?/ N6 q& G9 _& E
175. if   (result.Find(m_name)   > =   0)   {
     / }- X3 Y( L. e% y! J
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     1 `0 }6 J! U: o2 T; u$ G$ T
180. name.MakeUpper();
     4 z7 F% L8 J# x  c5 m$ p
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     & J1 }; T7 R! t$ z+ g1 R8 l
183. m_description   =   line;
     
184. m_description.Trim();
     4 z/ U/ U6 V4 M% l& V, F
185. return   GetDescription();
     
186. }
     . _/ K' t1 b! C. a! U0 I) Y+ U0 c
187. }
     
188. }
     7 k3 a4 D) y) i; P) N* x
189. }
     ; k6 x, ~5 l& M. e& v. V3 L# \" S
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
8 T, {+ N+ q) r& o, k2 W+ v+ P' O

///////////////////////////////////////////
8 `' Q, \. f! B$ H  @+ l/ q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

# ?! g: p: S$ b, l. f
#pragma once
, {5 n$ I$ H5 ]/ r" e#include <exception>


  enum TRISTATE{
2 M  N6 v  ~% p2 z, I& l        TRIS_FALSE,
) a8 r* J% a) B( H4 @        TRIS_UNKNOWN,
        TRIS_TRUE
8 L( _6 m8 {" X! I};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

! K+ z! `% y  E' U3 u; l/ |
  D& x4 G& Y- h

; k1 N/ v8 |2 L( v4 \class CUPnPImpl
- W. a" }/ t! n4 U4 a, q" y{
public:
        CUPnPImpl();
: h$ H, E" ]* T5 H) }2 x" M        virtual ~CUPnPImpl();
( e: C# q5 B2 U  g) \% f        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
" x6 C" \& C$ z4 l' o                UPNP_TIMEOUT
        };
" D/ ^5 D" E3 N7 Z; B; w  R  l) c

5 f8 p% ?0 W  N! b, d$ A        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
: ~: r1 m& v# S+ `" `7 s: f        virtual bool        CheckAndRefresh() = 0;
2 _; f( U1 q6 B# x  F: E        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
; l; T6 s5 B3 P$ z! O. L$ {        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
9 O1 H/ [6 F$ R: B7 e8 u. Y9 c        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
- j; G7 A: l/ F) {        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
. [* J- `7 m" m! j% u# q+ H4 r

% p% |' {4 B: _8 V// Implementation
2 h' m" q" y" ^- ~1 O6 r. y% \protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
  b" R* S+ M/ T/ h  U; O( }  F        uint16                                m_nTCPWebPort;
& v) q* w6 t3 q8 m" D/ R6 H4 t        bool                                m_bCheckAndRefresh;

2 z: S% S0 c- i! ^0 X# ~1 D/ b
private:
        HWND        m_hResultMessageWindow;
! G* N& a& _, J1 R% `        UINT        m_nResultMessageID;


};


) q4 ?4 k. V1 O9 g// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
5 G+ ?2 x* j5 z9 h7 v. v& E6 Y{
/ m; X' y: p4 Y' @% _% mpublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
1 T2 d( X6 P0 C! `        virtual void        StopAsyncFind()                                                                                        { }
) @5 |( |2 x' [7 t        virtual void        DeletePorts()                                                                                        { }
8 @' ]/ s# v" b! _        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

  F1 J9 H! w1 W% d
/////////////////////////////////////
6 N9 M3 f  O: [) X* j  V//下面是使用windows操作系统自带的UPNP功能的子类

0 _# S8 O, G- a. p2 e! o
* p' l( u' l  a) y* o. J- N#pragma once
#pragma warning( disable: 4355 )
" S; j' X6 Z/ x  H* T$ f
% H* f6 H- Y6 H3 u5 U9 f
9 m" v- ]7 `+ t. v: x  W/ l#include "UPnPImpl.h"
3 V' D  t  X2 ^6 j0 \! T$ J% _3 l#include <upnp.h>
) r% E: H3 }/ Y. }#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
# a' M8 i& y3 M

#include <vector>
4 D( `/ Y) M- f* [" t#include <exception>
#include <functional>

0 R/ Q& O9 ?# ~* |- Z! Z


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
5 ]4 H, e# d- d4 B% q6 Ctypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
& Z% ?, t% g& @, R5 s# d' T, ]typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
4 J. A# i) Z+ Ctypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
& k: x- l/ ?' K! \! ptypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

- T8 |. K9 @; ^% l3 ^
- H+ M3 E2 n2 ^. z3 Xtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
. `! W* c) Y4 `9 i( _$ n  PDWORD pdwBestIfIndex
2 |) j* K) i8 ^: O);


typedef DWORD (WINAPI* TGetIpAddrTable) (
# e3 G$ V" M' Y; R. {  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
9 b- L; G1 |& a  BOOL bOrder
9 X$ e9 ^; |6 x);
) v1 I) I' h0 \& M* C1 w
2 h0 t: W5 q* A2 `9 _; a9 M
typedef DWORD (WINAPI* TGetIfEntry) (
& U7 M% Z7 D1 }$ R5 ^: m  PMIB_IFROW pIfRow
);
% E% z! f5 A) j$ [" [0 j5 g% ?. I
/ O, h7 P' \# @1 f& l, A* t
CString translateUPnPResult(HRESULT hr);
* x9 r1 t8 u" Y% \. O7 ^. `HRESULT UPnPMessage(HRESULT hr);


  X8 Z% T8 e" Vclass CUPnPImplWinServ: public CUPnPImpl
- o; Z# \/ W2 ?$ Z{
        friend class CDeviceFinderCallback;
) ?0 I8 ?# p) o5 K6 L( G  V        friend class CServiceCallback;
// Construction
" G+ g  ^1 s( T( Mpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
$ \3 E. k' G5 l9 V, `3 |

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
* d0 n8 I6 a" v9 J/ f7 B5 k        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
" T' [6 `% B! L+ a        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // 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
2 u" D. \2 b! ?% E        virtual bool        CheckAndRefresh()                                                                                { return false; };


: }: k9 Q7 L. ~, g3 I, s7 I; i9 J) T, Iprotected:
8 B' C0 Q  O2 B# P        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
, r( ]  @; H! X) c        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
5 [/ D1 y) Z! F* `- Z  r/ ^. g        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


$ D0 q6 w% m6 q4 G/ q/ ^        inline bool IsAsyncFindRunning()
        {
9 k$ D$ M4 b& n2 D( g                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
9 Y6 _( s" }4 c: ^  A                {
# \" N; ]3 ?" I  S                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
' l2 f: A8 i2 y. C0 y, w3 E: Z                        m_bAsyncFindRunning = false;
- c) r# a2 q2 N' C                }
  `, W, `: E" e) S                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
3 a0 `; n5 d' J# U) @# x- Q' @5 |                }
                return m_bAsyncFindRunning;
, H8 @9 t" A. u/ x6 D5 t( y        }

2 Y6 W/ g, k$ _
  O( E+ H& ~$ ]2 A  o+ c        TRISTATE                        m_bUPnPDeviceConnected;

3 ~' @! s# T9 z
// Implementation
0 H9 T( x$ p: z* P  |( r        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
, O( n' Y3 k% F' ~1 E& `        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
- I" y8 r9 l$ f* u$ g; ~, q7 M        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
' ~- B+ H6 N' B3 a* j+ U2 K        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
6 G; h  A& o! `0 p

        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
% v1 Q( o# U& ]. `! o0 _        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
- `. w/ H5 [$ m5 O* ^. b1 J        struct FindDevice : std::unary_function< DevicePointer, bool >
; T; G1 t5 X- P' P% O; p. x, n' R        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
' n3 ]9 L5 s% ^' N- ~                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
9 m. B2 m  ~: |0 m

' m! N4 u' }6 `9 g                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


: `) x4 x! a; N9 d  a2 I' C                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
1 w* ]. m$ x/ p6 U! l7 Z* k        };
       
+ E0 p* j2 H' \3 Y! M* g        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
3 x; F1 Y0 F- \5 {        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
! n4 Z" w7 T! f! ]7 i, E9 N& V        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
  p* F: [0 t, F, b% e        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
+ e" E, ~! C' r( r: w                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

# T7 d5 ?9 }1 V$ ?9 ]3 O9 K
* N1 J; h0 @) v: c        // Utility functions
        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);
0 [5 w1 H5 S- R% H# L) ^        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
( j$ n: F5 y7 X3 ]* t* l        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
" t, _; g6 o4 N+ O3 H8 h
' G0 a3 Z8 |) C! n$ L) Z
// Private members
. m1 C  G+ S$ rprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
+ q% W% T9 Y' f( R# }/ Q. i        std::vector< DevicePointer >  m_pDevices;
5 H9 F+ K* x5 F- r        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
3 e, z+ x3 p& a# w$ Z) k$ U6 D        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
4 l7 |- j6 U4 E9 R, x

( W; v! c" a" p1 s  ^% p. F        LONG        m_nAsyncFindHandle;
! j* {  L: J& V' ?' K! c% Y        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
0 P' M3 x4 `1 H$ u2 c+ U        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
, ]' k" ?8 V) v" H7 G+ ~' C        bool        m_bInited;
# u/ r; R3 @3 \/ n) x' A) ^5 {        bool        m_bAsyncFindRunning;
0 D  c; a) ~% {- a. U        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
$ G8 `- n- V/ O! v' t" y/ \/ W        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
, ?2 x, e4 J" ~6 T) |! j; s

6 ^9 a, |, i* B( z6 e};
6 `8 r/ [: R( W/ _$ `/ N

// DeviceFinder Callback
# n$ R+ d/ K( n) j5 J% tclass CDeviceFinderCallback
; ^2 a3 c! s9 O1 `        : public IUPnPDeviceFinderCallback
{
public:
# M6 f' Z( u3 g  z4 v        CDeviceFinderCallback(CUPnPImplWinServ& instance)
! S! y$ I& k/ q  O/ T- U4 r                : m_instance( instance )
        { m_lRefCount = 0; }

# ~/ G' W+ r1 L# |% j8 @! R) W
; d( Q9 o/ t& J% i) ?4 r   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
% N' E" `% o1 u& U; W
: m0 k% P! K8 ~& F
// implementation
private:
! h( R* C3 x$ r# k- R; z        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
) M3 S& v# h3 Z8 v        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
' o5 B* T) r9 t4 o9 S

private:
2 p8 A% ?# H/ P2 A* l2 r        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
& R+ D% c( r2 P5 r" }! b# e};

" v( }  Y6 }+ \6 O" B
9 r# n3 @5 u; n% Z// Service Callback
  Q! h0 j4 X! t- Gclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
% [! ?! F0 x  ~1 `" p        CServiceCallback(CUPnPImplWinServ& instance)
4 o9 x, T" H/ {# Z: u. v4 i                : m_instance( instance )
        { m_lRefCount = 0; }
9 X' g$ c0 }% P5 {   
& n5 J" ]+ ~1 f7 Z   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
% ~6 w2 A# g7 Q; S) R, t& `: L9 c   STDMETHODIMP_(ULONG) AddRef();
) v& T8 I& N, `; R$ C" C$ y$ R, K   STDMETHODIMP_(ULONG) Release();

) p/ z. p( B( W  U
1 }% p0 ?+ D' y// implementation
private:
: k! B3 Z2 t8 \- T+ m" o        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
2 J- `- n1 c  L        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


) v* E: b& s7 c; F) Gprivate:
" O7 I9 v3 H. U6 e# ]% A( }3 G        CUPnPImplWinServ& m_instance;
+ S/ M* ]4 D9 `        LONG m_lRefCount;
8 ^% T) P7 r  Y4 K. s};
5 X! g9 k3 b) e1 N5 ^- B2 O" s% s2 V

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


使用时只需要使用抽象类的接口。
/ L7 ]* J0 h/ W, A- lCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
8 A5 S8 R( |' r& f7 ZCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
+ v" o6 \# j8 h# O0 n- G% HCUPnPImpl::DeletePorts删除端口映射.