UPnP

天云上人

/*uPnP.h*/

1. 
   & G: T0 z6 D1 ?/ u) q) q- x
2. #ifndef   MYUPNP_H_
   
3. 
   $ G. ]- O+ b7 ^8 B
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   ; P) L; h' C3 ~/ t3 z2 Z. P
7. 
   
8. class   MyUPnP
   9 v" [3 P# o. e/ b
9. {
   3 j% y- c; n3 j* p0 U% ~. s; E
10. public:
    ; x4 h1 n( V! {9 [
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    ) _/ t# x( k! q# |, r7 c
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
    ( q) [7 ~$ k; A' |# y
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    1 Y5 a! _7 ]. Z9 R# d% [0 I
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    6 n" e, z+ n% E+ L
22. }   UPNPNAT_PROTOCOL;
    $ `+ n) D! i% c3 V% g- e/ F5 b
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    + a2 F5 i/ N! E- ^$ p' S
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    + @( k( b1 `' O" C- k
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    . Y9 f3 g' S6 M4 t
32. ~MyUPnP();
    
33. 
    9 ^% n! ]2 E4 a! Q& Y8 v% |
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    7 H/ z) F, z$ D- \$ U1 [# n) ]
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    2 M/ G) B, |6 T5 n6 s8 i
41. bool IsLANIP(WORD   nIP);
    
42. 
    8 l0 y, K$ v1 C! ]& p) I
43. protected:
    ' t. R2 o) a% T1 j: }1 o" }
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    # m2 b1 r' Q" M6 c: h# [9 a
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
      x# Q4 g- C# Y9 q
52. 
    ! u- d( H5 k* @: Q% C: Z0 ]- {
53. bool Search(int   version=1);
    + _% `, N) M9 Z( ]0 C6 B
54. bool GetDescription();
    , [0 D9 `( g+ K7 ~2 I) j
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    5 G) I( V9 |% v: [& U, ?. D1 |
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    . F4 z& f7 F8 q8 r# x; X
61. CString m_name;
    
62. CString m_description;
    * m" E( K7 F$ Q0 w
63. CString m_baseurl;
    
64. CString m_controlurl;
    1 K1 F2 l/ U- L* a$ m6 g5 @
65. CString m_friendlyname;
    
66. CString m_modelname;
    ( f8 u5 I9 P0 A; K- z' A5 R
67. int m_version;
    
68. 
    8 z0 y$ M$ _+ i4 R6 O5 Y; g
69. private:
    6 O5 d1 x- c) D6 w' O( S: A) H
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ' l; q  [: S" X. ?, z3 a
71. 
    2 T# k! C3 E, m% c/ T1 g
72. CString m_slocalIP;
    
73. CString m_slastError;
    # `  _6 I" d- T4 U. J
74. WORD m_uLocalIP;
    + P; M8 S' B. `$ G0 ~
75. 
    / c; d# ?8 B, q4 h4 ?
76. bool isSearched;
    
77. };
    9 ^; Y& ^1 \! R5 H5 R
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   " f0 o& L, w" m( a- b
4. #include   "upnp.h "
   ' O1 }: l! [8 w  s# ^4 n  _
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    7 l! A" |* f% t* @" ^2 a; P4 S+ G% m. o
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    $ m) j5 h. ]8 [7 t
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    & V0 d1 W# i; P# W
16. const   CString   getString(int   i)
    : P# j. r: w7 G5 w
17. {
    ! i* y  b* p  x
18. CString   s;
    
19. 
      k  x/ F, ?3 g
20. s.Format(_T( "%d "),   i);
    & V% G8 X/ f! x7 \7 Z% M
21. 
    0 n" O) a1 B1 }
22. return   s;
    
23. }
    
24. 
    7 T: H- J6 q8 k/ Q2 |, T. J
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    1 x) A+ _& E7 R/ a' b% e
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    # l7 ]$ u4 Y9 b8 p. Y+ K7 N7 p% x
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ) w* d. |  D* I. z4 g- H) z% q5 A8 F
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    4 l7 Z- G, M5 N8 t* A! F  h
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    ( z5 f3 K* q/ H2 c1 |$ J3 g/ O
44. struct   sockaddr_in   sockaddr;
    : r' h, w4 M+ A  @  \
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    " E( e7 }2 M( }& O, I) ]: p
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    2 g+ e# E8 v1 i- g+ v
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    $ ]$ p/ s' x6 V! O8 ?! p: U
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    ) l0 b* x1 Q- D9 l; o
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    7 ^' ~# l6 Q; ^  G" u4 O) s  j
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    % {# u( @3 p+ m) c9 j
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    7 ^9 K1 Z% n' G) H0 E
60. 
    1 E7 w0 b( k+ A" f7 F: e
61. response   =   CString(CStringA(buffer,   rlen));
    ' E! G, z" r- r$ {3 h. G! k' q
62. 
    , h& P8 W  Z( |+ Z* I2 c* d
63. return   true;
    
64. }
    
65. 
    # @) \+ c- ?0 m$ j) ~0 H
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    * r2 u2 {& Z( M8 n% f9 a( J5 R1 }4 O
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    2 D( Z5 Z; G7 q2 X4 q1 X! t- {
72. strcpy(buffer,   (const   char*)sa);
    5 M3 O' P' ^' `; z$ o
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    6 v0 _3 \/ h0 u, N( ]
76. sockaddr.sin_family   =   AF_INET;
    ( l# m. v8 a4 e$ a$ ^
77. sockaddr.sin_port   =   htons(port);
    ' f4 D+ l7 V3 x; |
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 {7 X: P$ I6 V4 t1 j$ I& r
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    9 {4 p, B( ~% i
84. {
    9 G, |" A( S7 _: Z+ o2 \! Z
85. int   pos   =   0;
    9 d0 a' ^8 O, D1 C/ J4 L! J
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    4 [( N6 R1 v: E: L7 j% `7 d& G7 ]$ w
91. 
    0 p6 H0 T& c$ S9 F* J
92. pos   =   0;
    / [1 E& [4 r+ _" @9 k, _: n
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    1 _; I. F$ _7 i
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     # k* Q1 ^$ V# E8 ^
103. CString   property;
     
104. 
       P. |' e9 j) ^% I/ S
105. int   posStart   =   all.Find(startTag);
     ' o: E$ S; F7 Z
106. if   (posStart <0)   return   CString();
     : V" J/ C1 x7 u. y1 y8 m
107. 
     4 N4 |1 P% s4 D3 l( V) x# J+ \
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     . Q0 B" G1 p- S, R* G+ P
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     . u1 B* C+ b$ G" M% g# g
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     " G4 z: \. p" s+ y0 G/ d. |
115. :   m_version(1)
     - P6 O( X9 n2 T( e3 m- F
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     ' A# ^9 a# Z2 o3 @+ W9 q
119. }
     4 Q* T* I6 a* h* s& B& w2 ?
120. 
     5 G6 ~7 L$ G/ Z- f" Z1 h* }
121. MyUPnP::~MyUPnP()
     " A- O* j% u, t
122. {
     
123. UPNPNAT_MAPPING   search;
     5 f' i7 }) e0 E) W" M4 s* `. x
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     " P; x) K2 q" M0 R% |5 T$ J
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     $ l1 K; g( [3 [* p
127. RemoveNATPortMapping(search,   false);
     3 r6 o7 S' j5 p5 V4 a5 u. C9 a
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     0 ?6 c4 j* W7 _/ J
131. }
     
132. 
     * D: _: Z. [9 O  [
133. 
     8 u- g- }- v3 }+ H" P9 i
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     " |% ]; t0 l5 @; i7 T4 T( }6 S
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     0 M0 H1 ]7 w; t* `
138. 
     + Z+ X. g, \' j. {* ?5 H- D# ?
139. #define   NUMBEROFDEVICES 2
     
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);
     # [! K) ~, |+ s7 X
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     5 @; w9 B  T. `) z6 F0 f& |$ ^; u
149. 
     - o4 |, _$ d. h6 ]
150. int   rlen   =   0;
     + Q: n3 C. x2 t, ~* ~
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ' ?) ]8 }' H  Q( W# ^. f
152. if   (!(i%100))   {
     - S! N+ s+ _1 j
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     # e/ ]0 p0 B1 ?2 L+ c! z
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 "),
     / C( V7 \9 U0 t9 {0 a7 o2 P
157. 6,   m_name);
     - @0 h4 `* q  _
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     % q- y5 \% J! N! U9 P
159. }
     ' I7 X0 M6 V. m' L0 f6 o
160. }
     ( B* M  V* {9 P
161. 
     
162. Sleep(10);
     
163. 
     - F, m' G. C3 W/ u' a
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     , z6 F* N( i5 H: u1 G* A: p- _, F
167. closesocket(s);
     
168. 
     . j8 H. f4 L# \* o  a# |
169. CString   response   =   CString(CStringA(buffer,   rlen));
     2 F  s. @' ?' W2 O
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     % n  \6 w4 S9 }# c0 S+ f
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     - x5 [3 `$ g* h6 k
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     ( W: R' y: X7 v, o/ ^$ m6 n
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     7 Y7 E* v3 X" j9 A9 p, }. g
183. m_description   =   line;
     1 q4 R2 K3 R9 _8 v. z
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
       w- n* m  X0 J+ R/ w/ Q
187. }
     ; q* K  }: K  ^* w" z! y
188. }
     
189. }
     . f+ b! \7 o5 ?; p  d
190. }
     3 n6 D& l) J& q
191. closesocket(s);
     
192. 
     
193. return   false;
     : ~5 F: ^% X4 j) G
194. }
     ' t" o# |# s: C

复制代码

天云上人

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


% W+ F( k5 l9 f5 p' z+ @+ v: l  G1 ]5 \///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
! Q8 S4 S! S( l0 j
  E: T8 }( F+ L+ E. A7 N7 a
) U/ ]# {7 r# {3 M#pragma once
#include <exception>
6 y' b: N" f/ f. q; C( |. g

  enum TRISTATE{
3 F3 O7 f. c; N        TRIS_FALSE,
        TRIS_UNKNOWN,
+ _. C3 k: H! |. Z) r9 T. I        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
+ }% Z- X3 V4 I  L6 m  i        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
& z1 [4 `* U+ w* n        UPNP_IMPL_NONE /*last*/
};
4 d$ m& N' w- c  X7 i: F' T* D
2 e" }; g* S: C4 d1 ~

- I/ x0 X% T7 a# ~, ?$ K( U
4 V. v4 r8 g3 s7 qclass CUPnPImpl
2 E2 [+ e7 f7 ?7 Q! ~" A6 L  `{
8 h7 r! c3 b6 wpublic:
        CUPnPImpl();
1 n2 `1 P+ k  N  Y        virtual ~CUPnPImpl();
( F8 z1 c/ ^3 ]1 \& _        struct UPnPError : std::exception {};
        enum {
2 I9 u/ \1 q3 l. R                UPNP_OK,
                UPNP_FAILED,
                UPNP_TIMEOUT
, a  D/ u4 ]! b6 B) n        };

/ g( s; k( j7 z. @" E3 g- ]
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
" l( ]" u2 _2 Z# i        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


9 @% G- C' l9 a- J) T+ W: M3 c7 p; _# V        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
& v2 j2 O( q) \& c$ t        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
% R! X, M; V: G
9 B, n+ U0 ^0 W. A) A
// Implementation
protected:
9 u2 _) F" Z4 f! \. f; ?$ }3 V        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
# [) t! c# }6 _        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
% ]& {! T6 Y$ h0 I        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
4 V0 ^  E+ j) ?4 W1 a

private:
        HWND        m_hResultMessageWindow;
0 g6 i, m" j$ n! ?+ h: \        UINT        m_nResultMessageID;
4 f1 N# X, ^6 e& k

# V" ^9 V8 |) C9 Z};
! O) y2 q* P3 _. U7 H* Y1 i& O
! m# F& O1 L) n. |
- Z; i/ j3 ?4 d) @0 g& E// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
! v4 x/ P  C2 c0 C5 V) {+ {public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
! S- d) F1 x* y        virtual void        StopAsyncFind()                                                                                        { }
! S; n; Q- r4 F; g1 r        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
9 s1 X' f4 p& M) ], Y: }8 X        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
- b7 c9 {, V5 Q: h! X. r+ w

$ k" L% }* [% m! ^* P/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类


#pragma once
- y  W  R/ s- K7 T9 C' _#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
  f3 p( y3 s  y3 Y8 a#include <upnp.h>
#include <iphlpapi.h>
; |5 ^/ _& ~1 P8 H( h, z0 A#include <comdef.h>
#include <winsvc.h>
- Q& Q8 o+ }9 W5 {- j
% G8 s9 m7 `' q! v
5 \" W- e3 j& c, X# T! P5 ]#include <vector>
0 |* U5 y; N4 A. N#include <exception>
& a' P4 D! Y) |# B7 m4 g#include <functional>



" a7 N" i: V4 J7 ~5 X
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
$ s3 W) x. e& r5 n% \* Q8 @typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
: T7 _4 s  Q: qtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
" ?$ W& r# y6 W, n) Vtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
$ V: Y' G) ?, }3 N2 Htypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

, q' h. _& \8 `& a- u. ?+ J& P/ q
typedef DWORD (WINAPI* TGetBestInterface) (
( S$ B" z- C: }  IPAddr dwDestAddr,
( l% K% s: a$ ]0 C: O  PDWORD pdwBestIfIndex
);
5 P* G9 `( h) U

typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
. Z8 q/ j, r8 o9 @- \- D3 p);
3 n" M* e" }0 I9 k0 ?

% \; H% P# M; w% R( U- B6 }, ytypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
( T  i+ A3 ]; t4 S2 {- c% a5 |
1 {) ], [# S- F7 q1 i* V
, D4 S7 @, b+ }3 i8 b. E1 _& dCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
' S# N% i' P' ]4 r" f) t! r
% g6 q* J" x5 a0 [
7 F+ Z7 _# b3 B# }$ Aclass CUPnPImplWinServ: public CUPnPImpl
5 p+ t9 _( P" `3 b{
        friend class CDeviceFinderCallback;
; n  S& M5 t1 ~1 i* A" Q2 J        friend class CServiceCallback;
// Construction
public:
# S8 `0 I. ?! d! N* L* C! [        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
8 f- m- ~+ n7 D2 k, n
; D( l/ L' n4 J  D7 S# {: E
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
$ d' u7 Z% Q' D* V. k# B( L        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
: {* M+ j, h8 m. V1 |- r9 H3 M        virtual bool        IsReady();
( W4 A" ~5 R  j1 H6 E2 Q        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
: s5 q1 q6 x2 ~6 U5 u

! V! T" p5 r- ~* t. n' b        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
/ p! q5 _' r$ W" `! U6 e" ^+ Z        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


& i! F" S0 L. s$ bprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
. R4 s9 e* |, O+ `' _  Z5 N& F0 t        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();
3 K8 O( b3 I; G5 H% |

; |: `8 c7 C1 t2 _% G4 ]        inline bool IsAsyncFindRunning()
        {
$ `+ B3 g" b: i2 @+ g9 Y5 A' ^                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
! o5 _1 f, V5 b, v# l                {
* G4 c. e, I: ^" a                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
* _  Y! y: [) J& r! j8 ~                        m_bAsyncFindRunning = false;
                }
# a: _7 N! M' _                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
$ p, K# k9 C4 c8 d) x5 `                return m_bAsyncFindRunning;
        }
* ]1 l+ q- p, I

        TRISTATE                        m_bUPnPDeviceConnected;


8 T  Y5 n4 I( P5 r6 M8 S// Implementation
        // API functions
) Q+ X. ~# t5 n' ?5 z% ?/ g        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
4 j8 @5 E* E4 a  w3 {; T3 {        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
5 X+ b, Z. r3 Q- t9 Y, b        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
6 O2 {. j6 Q2 Y- E5 Z        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
) ]; w2 }, T* k; E: j: I        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
5 i- n: n+ t* [
# p2 X0 Y7 G5 B, x* v% R* p
, n$ `1 i9 {7 H4 n8 ]        TGetBestInterface                m_pfGetBestInterface;
1 X! i- W, S5 e        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
, a+ l0 m9 D. H1 G

# y& s) }# T1 Z+ A        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
- V2 h+ i+ ?& w' ]) a' p! s! D" ~        {
2 r1 S; G2 H1 w1 ~: Q  O                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
. Y3 d4 G/ a. j8 R' T6 z) e* C                result_type operator()(argument_type device) const
! A4 P+ N9 T( {$ }1 `                {
                        CComBSTR deviceName;
2 e5 V% e! h1 p3 P+ r                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
3 c. Y* X5 n" p! S8 E6 i* t$ Z1 b+ R

                        if ( FAILED( hr ) )
8 n$ @' V- q4 R; O                                return UPnPMessage( hr ), false;

6 i4 k" `: @6 L6 G
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
" d8 N, I# Q2 K        };
       
& I, m) X- w) |( Q+ M8 ^+ L        void        ProcessAsyncFind(CComBSTR bsSearchType);
( j8 j; [, T0 S7 j% R0 ^0 }        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
. \9 k3 U, A2 c        HRESULT        MapPort(const ServicePointer& service);
* h; g9 [% G. E  `8 C- D        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
0 ~/ r; y% ]; K$ ], O8 J        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
6 E' R5 R$ u: ^                LPCTSTR pszInArgString, CString& strResult);
; }7 N, q* s# B/ z* g( p+ E! ^        void        StopUPnPService();


, o: N: S/ R7 a/ w: O        // Utility functions
# U) H. e- t6 K$ v( ^        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
0 g% B5 W5 Z3 I8 X4 d* t        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
! q* `; V/ B! o! ~" m( k( u% M: G        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
$ t; ?0 J) W$ j* e7 y        CString        GetLocalRoutableIP(ServicePointer pService);

/ F& K: M. X! T# y
// Private members
0 p0 a7 T- F4 M) b1 qprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
; f. I* p, M2 ^        std::vector< ServicePointer > m_pServices;
1 U7 `* l3 L# d# _4 J' }3 U        FinderPointer                        m_pDeviceFinder;
0 Q- p9 ~8 A: M: ?" T$ _8 j" @        DeviceFinderCallback        m_pDeviceFinderCallback;
6 V1 w9 D  B$ z# X  x- @* Y        ServiceCallback                        m_pServiceCallback;
8 h2 j+ e1 b, ~; O  w; N  N

4 n: Q; e# C2 ^) N! y6 j        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
. q3 n9 ~2 |- u( G; o% |        CString m_sLocalIP;
' A! ?* s; x3 O0 p" ?) \        CString m_sExternalIP;
$ ~  L- v8 d/ _* o+ I% a! k6 F        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
% L& x5 X- Z5 ^# `7 M/ b. h        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
. m( A/ t4 v8 W! ^        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
2 K" {) I. C, k/ Y1 c        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

' `) \% w' k  _, q2 E5 o
};


5 K$ q8 m( h2 N; P- _/ X- C// DeviceFinder Callback
class CDeviceFinderCallback
# @' `  x* p7 F" i* T* J        : public IUPnPDeviceFinderCallback
{
public:
1 q8 e, V( `, J: [+ A8 F        CDeviceFinderCallback(CUPnPImplWinServ& instance)
7 }0 s& U  u6 {8 E. _                : m_instance( instance )
        { m_lRefCount = 0; }
" I6 Q2 v! v# U# H+ n7 f
5 l. x- \/ o- E, k3 p) O
8 P. S9 l4 ?8 L+ p   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 T5 p1 T$ T0 F6 i# }   STDMETHODIMP_(ULONG) Release();

# i% m0 I( t: G2 m/ u- k! h
$ c  P4 ^+ a) g7 H7 R5 \// implementation
- h& @2 Z6 S+ Q3 h, f( Mprivate:
6 D9 I# G8 Q  V1 s( s9 \) f. y2 Z" d        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
& |* d( s3 H% J  J; W        HRESULT __stdcall SearchComplete(LONG nFindData);
/ S- L' }* |# B4 r/ r0 u
7 V0 ^& a! I+ ]) J7 Q, u
9 Y7 Y) C, E6 Sprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
$ X7 x$ ]+ _; t# f3 `1 o4 \- b' R};


// Service Callback
: l* Y9 H' i( g3 h! h0 lclass CServiceCallback
        : public IUPnPServiceCallback
5 L. _% S% c  N% C2 K{
% f+ m& X% k" O( ^: opublic:
        CServiceCallback(CUPnPImplWinServ& instance)
. Y7 F9 o. }: r7 u' w                : m_instance( instance )
        { m_lRefCount = 0; }
0 \9 S. c. |) P3 z% D3 \   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
3 W( @  ]6 Q7 V: m" X   STDMETHODIMP_(ULONG) Release();
7 c* S# J6 w4 j  `. b6 j' G

5 h* P: ~9 h: h// implementation
, j/ O0 Q9 B- {' Pprivate:
( U9 o3 f7 g  j7 o        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
+ r/ A" x4 m3 n2 r6 W" G3 ?        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

$ l+ e1 R# Y& v6 B( Q
0 b) E1 \: C9 eprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


# \$ L- m4 a8 [* m: R: q) u/////////////////////////////////////////////////

$ T1 _9 N, [2 u6 f8 ~
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
4 u# m: R) l9 V7 oCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.