UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   5 L6 N' y5 m4 M: |: d4 t; f# C
3. 
   
4. #pragma   once
   
5. 
   4 L, O: c5 L( _# s+ o# H
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    + V( [/ Z- @0 m; F$ h
11. typedef   enum{
    0 Y" C* F8 S3 i$ ?
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    # {3 X- @1 ]8 C: d1 F; p1 F
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;
    8 g$ A- ]& ?* U* D% `
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    ! s+ _7 a9 w' R; |& B% [
21. UNAT_UDP //   UDP   Protocol
    : r  o2 H* I* h" E  j( k, [
22. }   UPNPNAT_PROTOCOL;
    
23. 
    $ m. ?* o0 U( _6 y, U
24. typedef   struct{
    ( g; ?* p. G- L7 g; j1 d& s) R
25. WORD   internalPort; //   Port   mapping   internal   port
    , e6 a/ _/ i/ G: s: O  l) ^' g1 @
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    . N  i+ \7 ?0 v/ g7 ^! y
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    7 U! r7 Y/ M9 u# r* R
36. void   clearNATPortMapping();
    
37. 
    3 ?% Q, U  i$ c( [) ^
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    5 p# |5 w4 K( H, O3 J. ~! d
41. bool IsLANIP(WORD   nIP);
    " V3 U4 ]/ n) g' C! D
42. 
    4 q4 ?6 X- d# b6 ^
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    0 e: \9 X1 K/ H+ n5 g
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    6 ^* j& ?- h9 I9 o
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    : ]: u  n/ i. V2 D
50. 
    3 B" m: ~* X5 D% x% N1 w
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    & C4 ^& H' m. A  U  ^
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    / Q) s" a* L# T5 A
59. bool InternalSearch(int   version);
    5 b3 P8 y$ F' ?
60. CString m_devicename;
    2 C5 L/ e9 e4 f; M- [
61. CString m_name;
    - u4 L* d2 L, G$ h3 H  ~0 F& H
62. CString m_description;
    # ~( X+ U9 E: }9 o' Z
63. CString m_baseurl;
    5 ~" u) c6 {4 [
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
    $ C7 x+ k* d* z3 S+ K6 P
69. private:
    2 W7 _8 F- m6 L% o7 {$ Z( q! U
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    7 O" f& S; A( ?  ]$ X+ r* n# ^
74. WORD m_uLocalIP;
    $ s: o- y: h5 {; n$ h) F: m5 E
75. 
    
76. bool isSearched;
    
77. };
    ) ~$ g( l0 ?! A
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   7 P9 H7 d: v6 t/ f+ n
2. #include   "stdafx.h "
   , A6 K- _; \- S7 D
3. 
   
4. #include   "upnp.h "
   1 p% D6 k# a  F% j' ~9 `1 x- Y7 K
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ; R6 x- }* n: n' ?' O
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ' \( s/ S5 ~1 [' E3 e! C' m( N
11. 
    + v' V6 P' S( A6 @
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    * j; B( j  i) [
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    
16. const   CString   getString(int   i)
    6 K/ s0 Z% ?  U5 B8 k1 F. k
17. {
    , B# y; k) Y% V1 \0 p6 h4 }6 f& J  `
18. CString   s;
    8 H9 H. A6 H; g
19. 
    $ G* v4 _2 r% p( K& \1 K5 ], H
20. s.Format(_T( "%d "),   i);
    ! x( X+ h* M) n
21. 
    5 H! k8 e8 G$ D
22. return   s;
    2 ]+ I! z5 `# V  @
23. }
    ) g7 i7 B; g& p: R$ i
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ! u2 E' x( f" W/ K8 V( e& E
33. }
    
34. 
    ; `/ {! J0 O' A/ ?3 l  J
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    7 p/ p+ n2 r  g3 F9 e" q9 G3 h
36. {
    % z1 E9 T: @7 U* I; p" G( R
37. char   buffer[10240];
    
38. 
    " i3 i( Y- A3 j, _( Y" o+ u: i
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    " ~: a. g& {( L( |) [
41. strcpy(buffer,   (const   char*)sa);
    + k+ e" u: N; Y2 Z# T7 T: n
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    , @& D4 ]5 Y8 @. M( S
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    1 P2 o( y) L0 @8 h" @. g
46. sockaddr.sin_family   =   AF_INET;
    ' K) o  _) X, K! L5 P; W6 q
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    % ~' h1 _* l) s
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 o$ ~, f3 @/ H  {0 `8 v
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    7 N3 n0 ~6 T$ y$ l9 I) \
55. Sleep(100);
    $ h1 O3 T; n* w% i
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ( U' W5 J% B4 v2 E1 E! x# M
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ) ~, @5 z/ `' Q) U) J
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    8 R; m4 I2 J7 \" }" I
62. 
    
63. return   true;
    
64. }
    5 @2 E1 ]8 s/ Y9 r0 o0 b6 I  A
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    / {0 v+ F8 s0 G- Z
67. {
    
68. char   buffer[10240];
    # Q" t3 |0 z& Q7 n; O% }
69. 
    ; @* H& K- a$ `5 e! c
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    " T  u3 _$ w% Z! O6 _7 @# U! f
72. strcpy(buffer,   (const   char*)sa);
    / |4 N; s$ c5 w0 H0 V6 v
73. 
    ) o- j8 H/ C3 H! e0 }' L1 j- j
74. struct   sockaddr_in   sockaddr;
    " r* u1 [/ D% O  l' G" ~& e# w. P6 I
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    # R8 a+ ?$ X, l/ v
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ; r4 z5 P' J: e; Z
81. }
    + K  H, S& Q7 b- I. _5 J4 _
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    # y  ~# w& S. V5 T+ f- ?3 I
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    ! O% ]+ \0 a& f; I9 e& ?# d  K
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    
91. 
    * F) `' y5 N& P9 I/ ^+ X( R
92. pos   =   0;
    * d  ~6 D% I2 ~; H
93. status.Tokenize(_T( "   "),   pos);
    . ?) a* A" V7 S, ]8 K$ C+ W
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
      i4 j& q2 `0 N; H; D7 S8 a
97. }
    ' [( Z8 h; B3 C/ z. h  k, X# T# i  h
98. 
    # Q9 c6 i7 O- g* c  j
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     & _7 j, y& q) `' Y# A* y0 R
101. CString   startTag   =   ' < '   +   name   +   '> ';
     7 }/ A9 h$ v" K) h
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ; K  L8 a3 L0 G. S7 I0 H
103. CString   property;
     - F5 C& `% Q5 W$ e9 F2 w6 z# t* Z
104. 
     - D3 Y' o! e  Y1 R5 q+ d
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     5 Z/ Y" j7 y' E- h8 r! p
107. 
     % }: y& }+ s3 m) B  e
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     4 j6 V  w1 h; W- v
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     $ S: ?' Q+ P4 ]# y& H0 N
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     9 _' q4 [! c$ h  E/ B6 D. C
118. isSearched   =   false;
     - s  K& `1 a9 S2 c( J" ~
119. }
     
120. 
     ' f8 r) S* ?" h2 n, Z. S
121. MyUPnP::~MyUPnP()
     
122. {
     ! b- {8 w1 z7 o8 M1 [
123. UPNPNAT_MAPPING   search;
     ( X- ^5 N* [. f$ n, @# e
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     6 \! O6 T/ V5 B8 z
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     2 d! f5 b( O, n5 _' ]
127. RemoveNATPortMapping(search,   false);
     4 X; N' x6 b! l: v" ^% n7 H3 v
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     / m) G! J2 }" ~+ Q9 y
131. }
     6 _; c, b: f3 G' A! a
132. 
     , |' F% s& X8 v" F
133. 
     - A0 H  ?2 }' |* y3 Z7 L
134. bool   MyUPnP::InternalSearch(int   version)
     0 D5 d( N0 ^$ \: A! ~
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     . L  I0 `* k& d& G# M
139. #define   NUMBEROFDEVICES 2
     . m/ s- h$ Z6 p. w) r7 J9 A
140. CString   devices[][2]   =   {
     0 x0 m0 ^! g( z) X
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     7 W/ T" A, m- Y$ o. H
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     6 p0 Y) O! L. S; H
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
       ^4 s8 Z1 z& P& h# k! Z
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;
     : o# _7 ]2 a1 l0 f
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 "),
     . X" c0 Y! _) ?2 ~5 y
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     ( a7 K9 p  [8 R, r3 N
159. }
     9 m+ p1 T& u! N% l( e+ u3 W
160. }
     
161. 
     / }) o5 H5 ?' j4 R
162. Sleep(10);
     3 U" E  t! y  _7 e
163. 
     
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     7 M# e" S$ h3 L: q' ]; y, X
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     8 e8 ?1 ~3 p& i. A- G& w
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     . a7 n& D; y) D! E2 D5 W
172. 
     / ]3 v" |2 m1 W! l" f/ h- H
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     : G# p7 _1 C, J) z2 r% T
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     . Q' v; C" |; s+ M0 b0 C  e& `
175. if   (result.Find(m_name)   > =   0)   {
     * x: e5 s$ ^% E! [- }7 t
176. for   (int   pos   =   0;;)   {
     ' m  v/ a5 ~: L
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     1 O, A1 w0 ~- a0 n/ w
180. name.MakeUpper();
     $ u: b6 S5 @' u$ X: _* T( O5 F
181. if   (name   ==   _T( "LOCATION: "))   {
     ' P( j3 h2 `: X5 k- f: @/ J% {; y
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     # d) p2 k+ t  A3 ^. n
186. }
     6 _- u: B: \; f- `  j+ Y; w" s
187. }
     5 ^' P% R8 f1 O+ Q6 M
188. }
     
189. }
     5 x) s9 E. c, C+ V. b  l: P
190. }
     
191. closesocket(s);
     
192. 
     % j, @+ k4 @, K" X
193. return   false;
     
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
$ V7 g3 z/ Q$ I

; J! W" b6 k" B# U///////////////////////////////////////////
& `- r5 X# O! D9 E6 X0 Q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

. U+ i" W- J) B0 T! {8 A3 `5 A  a
#pragma once
: e2 {% u, K2 B6 |#include <exception>
5 s5 L" \* F! @! ~& a! [

  enum TRISTATE{
! C  G  a* I7 L/ _1 a: u: g6 ^        TRIS_FALSE,
* C6 {- d3 b8 u6 H+ W( R& O5 I1 x9 W        TRIS_UNKNOWN,
3 L( b& A. D0 _8 d! t0 i- c9 K        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
2 S5 Y3 K' R& m  A& C        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
) ^0 _& {/ ]( |2 ^% E# X8 e, A        UPNP_IMPL_NONE /*last*/
};
# g: S( u" h: H; E6 i
0 |$ ]& C. r1 {3 p- a; ], h

6 b! t" E, P& \* x! o5 |5 o  F
3 P6 V% n. S# p4 b9 ]" e3 I7 \class CUPnPImpl
) P! j, o1 ]  t* I{
public:
        CUPnPImpl();
" `9 b: f' N) d/ [& V        virtual ~CUPnPImpl();
5 }& ^# o6 U/ b( E) t1 P8 ^        struct UPnPError : std::exception {};
# f  f4 y5 E: {& I4 V, P        enum {
                UPNP_OK,
                UPNP_FAILED,
  `+ b% g, ?2 i1 q5 R* d0 ]                UPNP_TIMEOUT
: ?* u% d4 _2 ?; m+ W        };
  M8 y4 }0 {! R) H( I& B% Y
5 W. L+ [' W5 B: H: e3 H- D
. R( q6 [7 w' h! R        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
4 O# ?1 H7 O; Q( Y        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
7 S' b4 l: S! N6 R% g' Q( \. _6 V        virtual void        DeletePorts() = 0;
8 p: p+ G5 @" b: ^        virtual bool        IsReady() = 0;
4 ^; b) z1 }# r& d5 W" c6 N        virtual int                GetImplementationID() = 0;
       
6 Y! ?7 I! o7 N; e) ?; K  D        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
7 C, Q& r+ T6 h* m( w
. J! K9 A% C0 P8 ^: Z* C
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
9 s( v4 E5 \4 o$ v( p4 W+ C        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
0 z. S' r, y* O- ?9 |

// Implementation
0 w; T: Q; S' V3 C, vprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
* R. s9 X, K. D0 ~        uint16                                m_nTCPPort;
1 u& M- Y  _7 D) F3 C* u4 N; k        uint16                                m_nTCPWebPort;
3 h5 e2 b2 U( y! ]" r: e- @5 c        bool                                m_bCheckAndRefresh;
2 w* Y0 x9 v. Z0 w& K* b2 r# U0 J

# [5 E$ v9 ~5 v8 e& p; X2 i4 Aprivate:
        HWND        m_hResultMessageWindow;
1 Q% t3 R( i6 B, x3 \6 e9 S        UINT        m_nResultMessageID;


) K- j6 h1 `% U) S! {};
1 h0 i# K3 c7 P( c; C9 U$ S" r7 `

// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
' @3 H3 l8 }3 W  u; f        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
( |' A7 X6 [5 @3 X( ^

; d  l7 ~) {  j4 b/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
8 h$ j% Z" o- ]$ h: J4 j

3 K" Y: w5 D8 ^" A3 g& _3 _#pragma once
#pragma warning( disable: 4355 )


1 ~$ T$ `+ M, @$ w1 _0 I- u! B#include "UPnPImpl.h"
. l8 ^7 i) e# H9 I- v#include <upnp.h>
. ^5 g' R: J$ R( k#include <iphlpapi.h>
! g+ _6 y0 x* a8 T+ ?3 ^- L0 ~9 u#include <comdef.h>
#include <winsvc.h>
. L7 J7 b# _, \) r2 m

# a& N; w/ B7 a2 o/ ~4 k#include <vector>
#include <exception>
#include <functional>


# ~( z$ q6 m( q- `5 Y+ q

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
: }* E/ @2 E. y; |  G8 g" H: M7 vtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
' |4 A0 L# n. `+ @* i/ xtypedef        _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;
2 |- A/ W  ?+ btypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

5 f, O9 A7 i9 J9 g$ {' P
; T8 T5 W+ ]. |( ]6 b% mtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
+ l# b( ?7 a* S. A  PDWORD pdwBestIfIndex
# j9 e: h& z$ d);

* y* P, A' R. _7 p, A
& g: h$ t) j; h% k+ Mtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);

) i; I! g# x  m6 o( @. F. a7 t
typedef DWORD (WINAPI* TGetIfEntry) (
+ {3 r) P) I. [4 h  d7 {3 {8 `  PMIB_IFROW pIfRow
* |4 V; i! a! g$ a4 a9 r* g/ ]);

' d2 ^8 d7 l9 ^6 R
CString translateUPnPResult(HRESULT hr);
+ r& `5 v* ]1 u' @4 E3 W' q3 eHRESULT UPnPMessage(HRESULT hr);
# R% j9 Z) A1 M

class CUPnPImplWinServ: public CUPnPImpl
{
+ y, Q1 l( o' y* W7 L+ y# M! u        friend class CDeviceFinderCallback;
3 D( ]8 _0 J4 W& ]2 v6 M: d, N4 |        friend class CServiceCallback;
// Construction
+ t" f# I6 Y* S# b2 x1 mpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
) r& v9 V1 a. z( B. g, c- [$ \

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
/ i3 C' a1 ^; u  o: N/ L8 T        virtual void        DeletePorts();
+ J* }! `! M, q0 \4 v! X+ k8 W        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

) p7 C3 d6 Q1 d
3 P' N7 C8 I6 p) l  v        // 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; };


protected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
" j, h+ ~( g) b: b+ l" B        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
% G; o' k6 Q" D; I        bool        OnSearchComplete();
        void        Init();
  [8 J. q/ Q& i5 X2 s
, _# e% R2 u0 S4 {
        inline bool IsAsyncFindRunning()
' b! L% C. T/ }1 o) U& [: x        {
: u( A  q: K( B5 Y6 t% V' y* S' c                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
- f6 `  y( U8 z4 m2 L# f                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
* `% I! x1 G( x' e6 J& @( y                        m_bAsyncFindRunning = false;
                }
                MSG msg;
8 N6 V! n6 `" d. }  {: E6 U                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
' d: d& p" T. a0 x                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
! w+ F' x- ]% d- b" h' z5 Y                }
  e# K1 Q  E% v. t+ y: ?                return m_bAsyncFindRunning;
        }


# ?9 t5 z8 C% t' x        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
& a4 h8 C$ w% }% j        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
7 M3 W5 L/ k8 v: ]- C7 M% j+ O5 w        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! w3 |' e3 e, Q  j( s  f        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
5 L" ]* q/ X" H6 D% ~& u        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
" l1 O/ H5 d1 P" N        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

1 E7 c9 i& o: a3 u6 i
+ a" W( P4 E3 z        TGetBestInterface                m_pfGetBestInterface;
8 d2 W# b: W, w  }: d1 ^! `' ?        TGetIpAddrTable                        m_pfGetIpAddrTable;
! U2 r6 s1 p4 K7 |( e        TGetIfEntry                                m_pfGetIfEntry;
+ j- X6 I( v* Z. r) R# v$ k2 k6 c
) W9 R* N, L* |) F! Y. Y+ |' l
5 U# f- A9 Q% i, k$ f  N9 m, [, x! Y        static FinderPointer CreateFinderInstance();
+ o' Y6 K+ }( ]        struct FindDevice : std::unary_function< DevicePointer, bool >
" ?  g* A: T* F        {
/ J0 \5 u" ?0 i: S8 |, W0 i. e  h7 h+ S                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
; y; u+ P' H2 i5 O: G) O- I                        CComBSTR deviceName;
- \. e$ {4 p: _: [                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

5 S/ U9 s0 c$ P7 \# ~* c8 d8 i  G# X
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

0 f; S& D, t  q& b  H
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
$ T" I& |4 a. `, u; R* {& B  p7 O0 o                CComBSTR m_udn;
        };
# a& I  w# ]+ Z) u( x       
6 f7 r3 a( Q& G        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
+ p8 K; c% I+ J6 D) `        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
% O7 {' T+ b' Z) K# d                LPCTSTR pszInArgString, CString& strResult);
+ e$ L; x2 |! k* `7 f* j        void        StopUPnPService();
! W/ ~7 ~9 l5 O) X
! Z' b3 t* a+ z6 X
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
$ F1 o. w9 @" E0 Y% u        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
. ?! H  L8 ^& N- e, }# B# v# O        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
. _& B2 _# d" }/ C1 k! F$ J+ h' F/ }        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
# o( Q+ v# J" ^* f8 Y" G& F& @        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
- f" }2 `) i6 E6 b' t6 K  \: V        CString        GetLocalRoutableIP(ServicePointer pService);
4 H4 w4 }$ d7 \2 [0 N
3 N6 V: n3 ^0 B6 ?3 y5 O$ ]' J
// Private members
: m/ \4 o+ d$ O7 d* L1 Tprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
* ~2 ^8 @0 v% m* I* G        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
9 q1 \* @( Z3 {) q        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
4 f* q- ^/ ~! K2 n3 m4 A# j
$ @( O$ @! m5 v% \
% A3 Q$ z* c7 I) H0 M+ p& K        LONG        m_nAsyncFindHandle;
1 Q% @% G! Y) L0 r, s& a2 v        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
8 v1 ^" ]3 s1 S3 R+ M        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
% C0 D7 k7 J3 ^  u        bool        m_bSecondTry;
8 h" {$ j, ~. r5 r        bool        m_bServiceStartedByEmule;
" i" k9 l! \0 T5 |, A# m        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
1 n$ A, h! r& |- J+ @' P. Q  u
. j* Y% {, O; i9 N! h
8 z& V; B8 _+ r% Q+ w/ h};


// DeviceFinder Callback
0 L) j7 Z% l/ Zclass CDeviceFinderCallback
, w# c' t2 F0 L& n: w6 D9 S        : public IUPnPDeviceFinderCallback
{
3 L+ G4 q% {; \public:
# @9 A% X- L; A        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
; _2 K2 E5 M% @3 L& v: X        { m_lRefCount = 0; }
1 Y8 J, j0 P2 {& L; [$ j6 `

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
+ [- D. A/ J3 ~3 U6 S" l0 d& `
. B' Z* P; o' a) g3 t
1 ]4 O( q: T- j- Z8 Q. r  P// implementation
private:
9 X( R1 @, @, I8 N( ?        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
  i2 C3 K9 h- ?0 C5 c6 G4 N7 P        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);


: [5 ~4 \3 _# _3 Z4 W# l$ yprivate:
7 J! T1 I$ p& q7 q% Q6 _1 f2 ~6 o        CUPnPImplWinServ& m_instance;
+ X3 h* H1 y  M7 S% z7 z( C# ?        LONG m_lRefCount;
& N2 Q9 {% m# M1 P9 Y  v};
. a8 N) b7 n- m! |% e

% y( j( O  d, g8 K9 V! T$ I5 r1 i1 H// Service Callback
, i9 A" W( ]/ ]% Y$ G/ _0 @8 dclass CServiceCallback
        : public IUPnPServiceCallback
9 x3 w. e2 O5 I( R2 u{
1 _' K$ m9 B) S. z/ {: Apublic:
  A& D* B0 U$ _" T) q0 [        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
  |. W+ X. s+ _   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


* _* G. r$ A" q8 w0 F// implementation
private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
$ C8 P. p5 r( q1 Q7 m5 z        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
' Z+ K: d4 a6 ?

private:
1 D) G7 q, F* q$ L2 n        CUPnPImplWinServ& m_instance;
; [6 B9 ~+ X- G7 w        LONG m_lRefCount;
3 S0 z: l" Z9 t};
# \2 ?. \5 s' @+ c

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


使用时只需要使用抽象类的接口。
8 [) s: r0 ^' O: k" eCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
( z* J% l' }# o3 mCUPnPImpl::DeletePorts删除端口映射.