UPnP

天云上人

/*uPnP.h*/

1. 
   & M) v, K  O- v2 K  i+ [+ u
2. #ifndef   MYUPNP_H_
   ) @4 T* h! t6 C$ r# o& h
3. 
   , O% P' c2 h& |
4. #pragma   once
   
5. 
   0 _2 r/ _. F; n0 I
6. typedef   unsigned   long   ulong;
   3 {# C* Z& w, X7 e3 D
7. 
   * V1 ^6 P# @3 z+ S2 n& f
8. class   MyUPnP
   0 [7 h+ e6 L% [8 r% b6 I, \
9. {
   9 W0 L' s( K5 b1 ^, Z& D; A
10. public:
    / n9 l* H' f+ v' v5 N3 `
11. typedef   enum{
    * R9 ?0 H7 q1 o$ k5 @5 z9 _
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    6 b. Q5 W! @+ ]) R( [8 s1 U
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
      G+ W  x: z9 q( T* u0 ?, }& I. E$ w
17. }   UPNPNAT_RETURN;
    : h8 T% U" B/ `& C; r9 ?4 M2 ?
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    + w+ D- [6 I3 B* B' ^1 g
23. 
    1 M% N7 C/ M1 n/ c
24. typedef   struct{
    , q5 j, E! H+ H; n
25. WORD   internalPort; //   Port   mapping   internal   port
    
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();
    7 c2 }: s1 E; U( f6 {, U/ V
33. 
    2 ]  v, b% W7 V! V# v8 }
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    8 q- I! E1 l" k$ t7 F
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    0 t8 T6 V+ m+ W' z1 i" L/ D9 A' N
42. 
    ; H, f& \6 Z% m# q. _
43. protected:
    
44. void InitLocalIP();
    ; ~! l/ ~, }( e" j
45. void SetLastError(CString   error);
    3 C. `! p  J) |7 m! p2 _, x, H
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ' @" ?- A" b) H: q
48.       const   CString&   descri,   const   CString&   type);
    0 d0 T- b* ^9 J" w8 d0 h* h
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ( u1 [& U% L& b* i% c9 e
52. 
    0 [2 E% J. @: K9 X; t8 `2 l/ U3 s
53. bool Search(int   version=1);
    
54. bool GetDescription();
    & J) n8 p5 r% `. b, |4 B! c
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    : s6 V3 b* z6 p9 q1 k. v' n  G
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    4 X% y- f7 b& A. k9 a( d
60. CString m_devicename;
    . ?, S; \! U) [. @/ @) L) a
61. CString m_name;
    
62. CString m_description;
    
63. CString m_baseurl;
    ' M' n- ^, D, e( k* ]+ W
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    ' Q" h* K5 \: ?* S. v6 C
67. int m_version;
    ' O- G3 |0 Y6 m% ]; B
68. 
    7 d/ e: v( Z; N: B- s
69. private:
    9 V- S" ~# }1 Q; {5 n& C9 j
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    ) a8 B# S, O& _7 x9 c5 ]+ P" o
74. WORD m_uLocalIP;
    / F+ [8 n7 X6 D! g  @3 l: Y8 W
75. 
    ( E. D& T" ~6 h+ A
76. bool isSearched;
    - X4 ?6 m6 c6 e* h& w2 q* J! F
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   - j% q6 y6 w4 u3 H( g, J" j5 F* _) R7 c
3. 
   . y  k5 S( Y- Z8 |! K
4. #include   "upnp.h "
   1 y% H' r9 A+ o4 ]  c& {
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
     n! c/ |2 f$ B
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   + q2 |& _7 z* r8 E
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    - a4 }+ I1 s4 r# k5 a
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    - Z, s$ `# Q$ K6 v
15. 
    
16. const   CString   getString(int   i)
    $ i  c$ J7 ?' I8 l7 w
17. {
    
18. CString   s;
    
19. 
    9 G" l6 U  b* @
20. s.Format(_T( "%d "),   i);
    
21. 
    & s* z0 V# `% |  ~( M' Y0 c
22. return   s;
    ) `7 Q) h8 t0 r5 }$ @- w) z' p/ U
23. }
    1 j* f# n  c* w$ Z
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    # c4 y- i# `* M+ q: l0 G
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    ) ^$ |- v5 X/ l
28. }
    
29. 
    2 a2 k% b: [- |6 K5 R! R
30. const   CString   GetArgString(const   CString&   name,   int   value)
    8 o  C" b+ e& K/ f
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    3 q# C+ W6 @0 r
33. }
    4 p, F3 ?* X5 _
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    6 _, ^- ~4 \- E1 |9 ]
37. char   buffer[10240];
    
38. 
    9 D/ w' H) a! e8 J- M, V6 f1 F
39. const   CStringA   sa(request);
    6 X9 q$ d$ R' }: O
40. int   length   =   sa.GetLength();
    7 k1 Q, r2 b5 P4 t
41. strcpy(buffer,   (const   char*)sa);
    4 _  o+ g/ V) |# w+ l
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    3 T3 R. o. F/ y/ t0 o
44. struct   sockaddr_in   sockaddr;
    ) q. E% m5 P, q! O( o+ `* b
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    ; W& |' G* |; H7 e3 B2 U
46. sockaddr.sin_family   =   AF_INET;
    ' t& l; V' |8 e7 |! e% O  |6 p! J
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    # o- d% W3 E  F" G8 N% ]9 S' i: W
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    + i. ^3 v- _: ^& o8 l
50. u_long   lv   =   1;
    & [! s. b# j* C) _% P
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    3 j$ X, h, h& J" T5 s; T+ v
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ( `) b0 D* F0 _! s' p1 H
57. closesocket(s);
    . c& {, D' ^; T+ u$ Q# D
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    , l- r8 a: A3 o6 _: P/ p" M
60. 
    " h8 a5 l, E3 v( R% ^0 a
61. response   =   CString(CStringA(buffer,   rlen));
    ! X- m9 [6 E. N. E; q% L' s1 S
62. 
    
63. return   true;
    
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    
69. 
    & W" K, b- W% _  ], `- c/ J( T3 Z+ m
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    1 X4 R* z, i* H( _
73. 
    ; p; f; [$ a  f8 ]- ~
74. struct   sockaddr_in   sockaddr;
      S1 N9 a) ~" E1 `
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    - Y( M7 K6 U: @. {4 b: A
77. sockaddr.sin_port   =   htons(port);
    % @" L4 R% }! g' n+ U$ U
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    5 ?2 H) u5 B! }! `# @
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    6 P1 ~6 \0 S$ D5 b4 U" e( H
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    5 \% u: h; ]+ V9 ]1 z# y9 P: F2 r
88. 
    . m1 M* w0 j6 }- \
89. result   =   response;
    , T# i" v5 s, A6 A
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    8 q- G  B9 V% J5 q2 ?- f# |! ~
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    % s9 V1 r! V; n- D
97. }
    2 I, I1 v" \% O
98. 
    + n4 P" o$ R2 L* Y
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     # ?- ^: b1 q5 p9 ]
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ) k/ P$ F  e: ~, j% a  `' B1 m
103. CString   property;
     : `$ P2 w2 m3 b' Z2 R) w
104. 
     # \( d7 f! R  z2 {& l& q
105. int   posStart   =   all.Find(startTag);
     $ M; v1 v* p3 g
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     : f$ n8 a! B2 k
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     6 W9 V/ a% S) S1 a! g
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     % u, D1 e: h2 A
113. 
     8 p; n0 K! `1 @- [* p' q: N- p; W
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     1 G4 t& `) O  U% g- G- a
118. isSearched   =   false;
     2 E3 s' q: t/ F# L! ^7 G
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     1 K2 v- `' Z- V$ L. r" Z
122. {
     * V& r- x9 d0 S* y' ~
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     7 ~- R1 N3 d0 W7 \* }) Q
126. search   =   m_Mappings.GetNext(pos);
     0 w1 B1 @7 N, o* Y1 Y2 ~, F
127. RemoveNATPortMapping(search,   false);
     
128. }
     # h; i+ X' f5 W( ~
129. 
     
130. m_Mappings.RemoveAll();
     / J8 g9 y+ G* i1 x/ k/ v; a* U% ^
131. }
     
132. 
     / F* }, o7 p# V' z5 n
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     5 B) P" T' C  o8 j* F
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     * r+ I( w& z, A* T
138. 
     + ]5 C8 |- R, A% H! }
139. #define   NUMBEROFDEVICES 2
     " w3 g7 b4 U+ c
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     3 _( p$ w6 {$ m) V
142. {UPNPPORTMAP0,   _T( "service ")},
     ; z" U, s2 Q0 b/ w# ^
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     % ^5 F) ]& L! X+ e$ [6 ]1 W: I
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     1 X# l. Y8 Y1 D7 p
147. u_long   lv   =   1;
     0 G+ D# B% c" _7 k0 K# F' I- r' E
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     2 r# Z' r, R- G7 u- r9 |
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     6 }0 G9 u9 U! Y& y+ ?3 y5 D4 u/ j: m
152. if   (!(i%100))   {
     ' X; ^1 ?6 l* s! R
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);
     7 P$ {6 g! C7 I4 c/ ~
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 "),
     ( J5 N4 p3 ?# Q( ^& i  e
157. 6,   m_name);
     4 k& k$ k- |! E: c6 j$ ]
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
       S* v: g1 \# h: }" R
159. }
     
160. }
     
161. 
     6 C, v$ l# I' N: U: M0 H2 c
162. Sleep(10);
     0 w3 @8 F2 g/ j: k; J
163. 
     
164. char   buffer[10240];
     ( R, r- u7 Q1 r& F0 T  ~2 S
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     5 o9 K7 c) b, |! _% {: p
167. closesocket(s);
     
168. 
     1 j3 z' X# W, m/ q* q6 y8 K) k4 D& G
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     2 e% C- Z: h  H" L! _7 j7 y4 |* r
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
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);
     0 [- U* _9 b2 a5 X
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     8 u- W/ ~# D# C" C% b$ j" j0 \" a
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     # u+ D5 c. D" a. e" M9 u6 p% o
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     . {! z$ s- D# Q' p
187. }
     
188. }
     ) o) ~: M: \# Z* E0 w' w7 Z
189. }
     
190. }
     1 h/ \6 J" P6 P
191. closesocket(s);
     
192. 
     3 \( }' H6 y2 n; B3 \
193. return   false;
     8 A9 K. u  ]4 l2 {0 a. K
194. }
     

复制代码

天云上人

以下有关upnp的接口来自emule，
- Z% P2 m3 u% b9 j: p( R

% u0 h4 L+ {( a4 q. M) z. R! v///////////////////////////////////////////
* h5 l# \; e3 a; c, M//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
9 k9 `* Y- I6 Y" {9 H( r7 a

+ Y  @$ J0 L" z& u#pragma once
#include <exception>


6 J; `, l+ D; R* q  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
3 ?" ]+ x/ ?/ h* k/ z};


enum UPNP_IMPLEMENTATION{
: f( E2 S  r; ]        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
5 a2 o4 |" W2 S# e+ f- P$ }        UPNP_IMPL_NONE /*last*/
};
4 M# h* [( w- ?. Y6 }

$ y" q1 Z* l0 r  h5 |1 f; L9 b  l
/ g" J$ g$ k3 J8 Q% h2 d3 e
class CUPnPImpl
% _) c* q( g2 g+ {6 U{
public:
$ E" X% O0 Q1 t. q6 C        CUPnPImpl();
8 _  \3 b  Y& |. {! L        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
( z8 Z8 r; V$ c' Y                UPNP_OK,
                UPNP_FAILED,
4 n7 J' V/ ^/ y8 Y2 K2 B                UPNP_TIMEOUT
1 p  h3 C% p- F        };
& A( V5 }) m8 C7 b! `' `

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
# s, q! L7 \6 L. K        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
( ?6 j0 u) H( W* a/ ?% |( p3 T/ G        virtual int                GetImplementationID() = 0;
' I$ B& l- v3 a       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
' i$ V/ v  C9 }- n        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
: l% ]0 b4 O4 G4 x0 m6 J$ \        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
, E  @; S9 a1 f% u& B# `        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
+ `6 o; `/ T* G" O7 O1 A
1 @1 }9 J7 r1 g& E  @" J% q
+ N* P' H6 l4 Y// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

) `4 b; m4 X* u% ~2 o  t9 n, v. \" n8 @
private:
7 {8 h" g" q, N, U& C" m& R6 y, M        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
/ Z3 ]" V  Y% S% c/ N/ b
( g1 Z9 a' A  \
- y1 {$ S0 g- D9 B};
5 `  _4 F8 @9 A" s% g

// Dummy Implementation to be used when no other implementation is available
) J1 C0 t1 s2 L; q; B( d! Hclass CUPnPImplNone: public CUPnPImpl
{
public:
! G1 U2 C" j( y" h        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
: k2 M$ L2 w. C        virtual bool        CheckAndRefresh()                                                                                { return false; }
4 h7 v7 m* z; v8 B4 q1 R$ d        virtual void        StopAsyncFind()                                                                                        { }
3 H. M- |, F) n) o  M        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
# W5 ], T0 W3 k: z* O//下面是使用windows操作系统自带的UPNP功能的子类
: _) \  [1 Q8 ]9 t& E
' M4 T; \- B/ a8 [1 g9 Y
#pragma once
#pragma warning( disable: 4355 )
" e  L4 c! b# h& O% y
% B4 X; ?4 A0 n6 k
#include "UPnPImpl.h"
! W3 Z0 t/ \. _4 S6 `6 J, R6 U* l, T#include <upnp.h>
5 W9 v/ c5 N/ L#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>


#include <vector>
4 V7 e+ g/ R9 K+ Y, E#include <exception>
. R- Z7 p' L' Z$ t" J% Q#include <functional>
9 r& D- D' e$ I! Y

1 C! n2 z3 m7 I! ^4 E' O. H- Y
& J) F* |; Z! Y" m3 C
1 S/ _  a2 r* k. d4 G$ Q/ @typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
9 [6 ^, N  }+ z6 B3 \* itypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
1 ^) Q7 a" F( mtypedef        _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;


typedef DWORD (WINAPI* TGetBestInterface) (
2 w7 o) G% M0 e! e" S( m- o. E  IPAddr dwDestAddr,
4 O% o9 F; i! T+ s; L  PDWORD pdwBestIfIndex
/ B; U- v0 P' e) y0 y& Q4 s% r);

$ b) G/ z0 ^' T3 y
typedef DWORD (WINAPI* TGetIpAddrTable) (
& B# W8 B7 b  ]6 D  r; z5 O1 w  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
1 J% s( o" ]1 c  u/ M  BOOL bOrder
);
' q& }8 x) h1 F- W6 u

2 G# P' S. M8 Rtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


CString translateUPnPResult(HRESULT hr);
' p" I& I, j! a2 x5 u2 }HRESULT UPnPMessage(HRESULT hr);

7 P+ j4 k) `& N* z3 ]7 i  a
9 C8 o2 Q# |( ^" h: Xclass CUPnPImplWinServ: public CUPnPImpl
% U' n: p4 O) |) V0 M, W{
4 x) i. m5 f7 g' R4 W5 S% j# j9 Q& V        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
& ^: t$ Z- J- r9 i6 x% n// Construction
public:
        virtual ~CUPnPImplWinServ();
  P2 W( W- L  }& g5 Y        CUPnPImplWinServ();
' T7 R& \" x8 C1 _
# S7 R3 L) v( i1 f& X, h8 A$ i
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
1 m9 {0 v. D3 d4 X7 Q: {, l# p' |        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
( ]  G" P9 B* }- V% r& n; i

        // 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
: q+ v* |3 d9 o0 R* `        virtual bool        CheckAndRefresh()                                                                                { return false; };

8 \) g6 d8 @6 L6 M8 T" V
% |' V* F7 ~0 \! a. m" sprotected:
! a  J0 z& M# o" r+ w2 w" T. L1 f        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
2 z* y' r; i0 Z% E4 j7 i5 {: y        void        Init();


        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
! Y7 A8 K9 M1 r/ h. D: D                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
3 G0 G. [+ q- v& U                }
                MSG msg;
( B% z% b' u3 v7 J7 z                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
; r- I' x/ Z) q5 Y5 g9 p$ q4 U% ~                        TranslateMessage( &msg );
# B+ T3 Z" ^6 Y5 e                        DispatchMessage( &msg );
                }
$ H9 C6 K; H% {' N; G) d. y( W, Q3 V- l                return m_bAsyncFindRunning;
        }
/ `5 a- Y8 e) e
1 T" x0 y  |2 h0 ?. \% S
6 S; {7 p- v+ d6 i1 r3 M0 ^2 r- o8 N  {        TRISTATE                        m_bUPnPDeviceConnected;

2 P+ _. E% w! f: c! X4 Z
, r% c- [7 h. C# u// Implementation
        // API functions
, q" h- q  u6 M4 _3 x8 U        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
; i$ p+ N( z4 H; o/ x2 u6 O        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
8 ~# C4 {& X/ Q, N, r, `: g        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);

/ w! g3 f4 x3 x( p6 V) \/ ]
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
# o/ y* K7 B3 j, v" Q2 i  G
9 B9 B2 I4 b# ~
2 y- Y$ d' I& ^) k9 M; k  p! Z        static FinderPointer CreateFinderInstance();
* N* m" F5 E* z5 J4 a% y        struct FindDevice : std::unary_function< DevicePointer, bool >
) O  J9 ~0 {; r; r8 u/ z, a0 }        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
6 t( R6 }0 |) f5 N7 H* n                result_type operator()(argument_type device) const
6 I. P! B/ Y. g9 @" Q                {
                        CComBSTR deviceName;
4 Y, n  v2 u/ \! D6 Q                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
/ u* N4 }0 Z" U6 ~5 U* ^
& O3 r$ Z' ~7 g
$ P6 p% w& n0 A5 k3 _9 R                        if ( FAILED( hr ) )
2 Q& }# K0 n3 A, C) P, u                                return UPnPMessage( hr ), false;


& l5 T  E/ N) h( j8 p7 w                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ @# \+ U: X- w0 o                }
                CComBSTR m_udn;
% }0 G& ^2 Q' c8 u) C/ N1 d        };
! }" O9 ]$ [; ^8 D( Y       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
) E: J, ^0 [' g* F5 L9 _0 t        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
: l/ x, T, U' O& r        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
1 v: N/ O8 n  p% }        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
$ u- a- h* t2 K2 I, d        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
& @& H* Q- k' O! W+ P3 d1 a        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
1 @, D5 Q9 i6 f3 ]6 y        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

7 |6 L. }- H1 U1 j
// Private members
private:
5 b6 b. X4 s9 d4 ^& j7 ^) K6 Y  M        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
$ l/ e* P5 @* E4 a        std::vector< ServicePointer > m_pServices;
6 k( Y. U4 a  H/ M. f! j6 h% y( i% t        FinderPointer                        m_pDeviceFinder;
! O* {6 ^: `. a1 Y        DeviceFinderCallback        m_pDeviceFinderCallback;
3 E  W! \2 n! O! H0 z$ u        ServiceCallback                        m_pServiceCallback;

! W" A. k2 i7 b% k2 j
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
- W1 I' n8 E+ T, F, ~        CString m_sLocalIP;
: {8 x: l$ }8 U        CString m_sExternalIP;
' g1 f2 D, s& [/ X- m& K" n' G, x        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
- s# f9 l* x6 j        bool        m_bInited;
; V8 b4 u) y* m. t* H) s: b! A% w        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
. W: w, ~; O8 c        HMODULE        m_hIPHLPAPI_DLL;
0 N8 L  ]: X$ w) }$ M+ R        bool        m_bSecondTry;
+ H0 M/ j# |6 [6 b: B7 A& |        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

$ C* [: A( r" O, r
};

" Q& G- y& N, B
// DeviceFinder Callback
' [# b- K( S- F. B4 lclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
" E! o( U$ Z/ C6 k{
: z4 \; s* e; Ypublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
0 j" Y/ J8 }+ E9 D
* a8 \+ h  q9 m. `6 T
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 Z; a/ ?4 u+ S   STDMETHODIMP_(ULONG) AddRef();
7 d7 k1 f0 a3 Z0 X& y   STDMETHODIMP_(ULONG) Release();


* K  ?* Y9 R4 @. ?0 u9 j// implementation
private:
! L" u  q. x6 S; b/ m" b        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
8 d  P2 g# j" I! q) {        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
" S4 o0 L' z4 k3 |" j
% m9 j4 H' ]! N/ u3 k! L0 u  A- m3 [
! [% K6 e' p# j' d- }/ v9 A/ jprivate:
, d! A0 [2 ?' s        CUPnPImplWinServ& m_instance;
" P7 T. u7 d( z0 k* b1 y5 P        LONG m_lRefCount;
5 F/ [3 c3 i" S$ v6 F9 U. i};
! y. D: j, J6 D9 Y: M+ c7 t, U
1 I* l* u  _: r# u& j5 s6 L0 H& o3 R
// Service Callback
class CServiceCallback
4 i4 \9 I* ]3 {% Q        : public IUPnPServiceCallback
1 ~0 [$ O5 |, T4 n! N{
public:
! f9 p6 ?5 R; e3 ^! D7 v4 O; V        CServiceCallback(CUPnPImplWinServ& instance)
! ~6 U. f  w7 @+ }" F) x0 b                : m_instance( instance )
- O2 R( z2 m1 d' e0 U/ _; A        { m_lRefCount = 0; }
' N- D. B8 G( q/ p3 t   
6 U: k. R9 ~8 |. i$ M: x   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

- n$ G2 Y! p, c- W* Q
: H* C4 q' p- Y2 j( a# w; d0 J// implementation
private:
4 C# ?. }8 D7 c& z& N1 A9 Z        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
8 ~1 L1 w: u3 Q* A. }& B6 K$ Y        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
. n# X1 G: e5 O
7 t5 z: T) H1 {
private:
1 C3 }$ R8 i; ~/ d! j        CUPnPImplWinServ& m_instance;
7 C  y& Y8 ]5 P9 k! Q5 K2 N        LONG m_lRefCount;
. L+ F1 D0 t( k7 b$ b3 }  \( \4 E* v};
' z" W& b5 M5 r2 V) \9 H9 Q
  H# N! M$ F0 ?
) u. @0 N' D2 N9 @+ L/////////////////////////////////////////////////
2 r8 G. |" s' g0 l* g

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
: e# }/ \( H: ~. O' QCUPnPImpl::DeletePorts删除端口映射.