UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   , D: M. c) u$ Y- D% E. Y) j7 m
3. 
   
4. #pragma   once
   
5. 
   / p. L+ h( W2 {0 h  g4 j
6. typedef   unsigned   long   ulong;
   8 n' @0 r  u- _# a5 }
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    ; Q" B) \. n9 E" D) L6 G  I
11. typedef   enum{
    ( b; |6 D6 G9 R: g! U# J& t7 D
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    8 d" p2 ~& t2 A2 L$ A! q) y5 J' S7 d7 {
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. 
    ( V! T9 G1 g) Q9 F# ]& [* h
19. typedef   enum{
    , p# W. m1 q6 O0 f! y
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    , Z3 X2 h6 X; w; @4 n2 L
23. 
    , s3 C% O. Q" w2 ^: m* |
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    ) d8 F" s7 A. B
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    9 s# r; y3 B3 K" k5 J. k9 K0 j
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    + e. a# |% t# t% R1 T( W0 o
30. 
    
31. MyUPnP();
    1 \+ E1 d+ n( Q+ Z7 b$ q) _; L4 I% L
32. ~MyUPnP();
    9 c- V) H, z. ~. o5 Z. ^6 Z& ]
33. 
    ( L0 G7 Y9 U( x% L$ `
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    1 }6 Z! }/ D3 M
36. void   clearNATPortMapping();
    , K7 s( v; Q7 c  x
37. 
    , C" m0 O: j- m' u
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    5 w/ E& ?. C2 I! K6 Y7 p; {
41. bool IsLANIP(WORD   nIP);
    % E4 ]. v) P! t) t) u. G  w4 n
42. 
    
43. protected:
    
44. void InitLocalIP();
    3 `+ n) d: k8 N* F' g7 h; M* X, m
45. void SetLastError(CString   error);
    
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    1 j4 s: O3 W% Z5 X6 p
48.       const   CString&   descri,   const   CString&   type);
    # M8 z2 @( A0 J# u, m6 m. f
49. bool   deletePortmap(int   eport,   const   CString&   type);
    ; B, K% ]8 F  {7 w% s( y
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    8 o5 F. i' X; N$ f
53. bool Search(int   version=1);
    $ U' L8 ?" z. r1 o6 Y/ j/ k; o) I3 C
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    ( {7 o1 ^, v; i  ^& V
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    $ n6 o3 m. e5 h( e& B
57. 
    . l* a' M$ |/ R+ X; g( b- U- N
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
      |5 F9 s+ O& F* i7 E. c3 n
60. CString m_devicename;
    
61. CString m_name;
    : J5 O( y- P. b, k! O
62. CString m_description;
    
63. CString m_baseurl;
    7 R2 ?5 j6 Q: I- U
64. CString m_controlurl;
    2 e; b' K* z4 C" F6 h  r
65. CString m_friendlyname;
    # Y9 ^! J) F4 s
66. CString m_modelname;
    
67. int m_version;
    8 P1 @* K& e) y# ?8 I" w' m
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    8 L) L0 W- L0 G! ~, |4 w
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    $ D6 b* @' F$ ?: W
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    # H% Q/ |" X% I/ `% j9 M
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   * _" `8 }( e9 ~4 `9 L  w; D+ t
3. 
   , D" _$ J1 e9 j" B
4. #include   "upnp.h "
   + \2 A( k) e) e* I
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
     H" J( n; b3 P) @
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   9 k2 s3 i# A4 v* P% `0 M
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   % S) D% K& j) q, n2 I$ Q  v
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    / u: i' v' p' r/ f+ d0 @7 {* }* ]
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    0 j( n, w' E5 u/ k6 w4 l
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    8 ~. P, K* w1 N% k0 w* m9 ]2 R
15. 
    ; ]8 U* d% a6 q: B/ m* D2 N
16. const   CString   getString(int   i)
    
17. {
    8 L, h9 g3 s  P" j5 c- U2 t& A6 m9 t. J
18. CString   s;
    2 ?4 Y) z$ H3 T3 F
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    + ?0 k: s5 @4 m5 r  K  v
22. return   s;
    + _8 J) i+ N. f7 p3 `
23. }
    . |) i* C# t( E( v4 w& x, Y
24. 
    * _6 n3 ]0 _! {$ Q; U
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    4 H8 w  D. Z% o% c+ N
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    0 S& M+ P' X3 s
29. 
    ; _1 v" R9 V6 F6 D
30. const   CString   GetArgString(const   CString&   name,   int   value)
    6 a' w+ z/ U, ]
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
      y$ @# L4 x3 e  Y
34. 
    / T: _+ p3 A8 `+ i
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    2 m, U% O0 N) }! g2 R  \' w
37. char   buffer[10240];
    
38. 
    & B  x" p+ o6 Z9 i7 j, d" i  t
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
      ~6 A3 e: R  R3 h3 j. ]! M- i/ W8 F( O
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      ~2 F" W: F, S9 Z4 Z
46. sockaddr.sin_family   =   AF_INET;
    " X5 P2 b( o  T% s1 v9 C' H! A& c
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    & W$ D" Z6 L1 l
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));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    & B( s  u* K7 [& G
60. 
    3 D1 J( y1 Q& y: u% I
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    ; a' J' q0 B& {  h; z6 P4 A
63. return   true;
    0 Y, R5 b3 `# v3 G
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    9 o( n( b$ D$ l5 }
67. {
    
68. char   buffer[10240];
    , u, n' R$ \4 j) e- z4 |8 _9 D
69. 
    9 e; ^4 @! Q5 f' Q
70. const   CStringA   sa(request);
    ! d& D& R: q8 z6 |
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    ( u/ b7 D$ I% e
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    / F6 o5 k# l1 J( D- G
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    * g7 ^  c) h# N$ [) ^; q, Z
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    , X0 F0 f( A2 o0 j
90. result.Delete(0,   pos);
    $ q' |( y, ^4 t1 e6 P: P: n2 J6 \
91. 
    
92. pos   =   0;
    5 o5 V$ Z3 K$ r' i! L
93. status.Tokenize(_T( "   "),   pos);
    4 Z( s0 }5 U; W) c. V, F
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ' Q8 B5 g# \3 _! T# C' j" z
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     6 {$ z/ H5 i/ j) {3 g
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     0 e8 J7 J: s+ \% R. [3 Q& a. j
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     0 D0 [1 i/ K! C
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     5 C" d8 `1 c/ s
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ! G1 L; u' a- h% P
112. }
     5 ]: `" t- {1 a& ~/ z
113. 
     
114. MyUPnP::MyUPnP()
     ; h. }; b: G1 |$ y) S
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     
120. 
     * |% G4 ^2 F* a6 }
121. MyUPnP::~MyUPnP()
     
122. {
     ! S$ o- D* C7 H$ d0 C1 }
123. UPNPNAT_MAPPING   search;
     3 ]2 L# `" \+ J$ Y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     ' g' ^- p3 b, d0 `* N. D
126. search   =   m_Mappings.GetNext(pos);
     . V/ L# w) g' C, O: D
127. RemoveNATPortMapping(search,   false);
     : [$ m" j( ^6 K8 D9 q* Z5 L
128. }
       J6 m( d4 Z/ v  R
129. 
     
130. m_Mappings.RemoveAll();
     6 p# J8 J1 p4 ]5 u
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     . y2 F1 A/ ~; i! J8 L. S  w2 t# Q! A
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     9 G7 C- S' C' J, g
138. 
     
139. #define   NUMBEROFDEVICES 2
     + t+ E9 k; p0 Z9 R9 R5 h5 i" R
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     ! ^- ~& |% l% o# v
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     5 h! M0 A$ ~* r. p! S* g
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; }, r6 C& r# E' L1 v
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     / E( C5 o" O9 x% H8 u; G  D& J9 {
149. 
     3 w1 c5 N1 h) q) f
150. int   rlen   =   0;
     & j$ F( c# l2 O
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     3 E, z$ ], C4 J" ^
152. if   (!(i%100))   {
     7 Q5 F- }( K  T) Y/ M
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , Q) w' k! v+ d: k% W# T
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     8 D5 C4 D' D8 u5 L
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     
160. }
     7 d( _8 r2 I/ m( V/ G0 ~  J
161. 
     - n' w3 U, K0 B5 X+ T0 [* }' \
162. Sleep(10);
     # I$ ~' L1 ~) E: h3 k: M/ e& |7 m
163. 
     
164. char   buffer[10240];
     , I3 ^1 j  ]: F) b9 B
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     " d- n9 e: i% L5 U
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     9 X% {# e& n( p- S) r4 b
171. if   (!parseHTTPResponse(response,   result))   return   false;
     / v, R( E2 j0 ?
172. 
     
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;;)   {
     9 j  N( b. H7 j7 ]6 W
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     / {8 C- @& ^; N- H1 y
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     . D9 h+ `$ g8 U& `, ]% |
180. name.MakeUpper();
     4 S- Z$ s( D/ t9 v5 W' J9 f
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     . S7 L" q- X' F+ T3 M
183. m_description   =   line;
     0 s1 g" T# L- x5 j( N) m/ H
184. m_description.Trim();
     5 x" L) q' I8 l/ q$ _, N4 r
185. return   GetDescription();
     6 T) a4 `7 S: ?% U
186. }
     ; M5 K2 Z8 R; u7 A$ J% g
187. }
     
188. }
     
189. }
     0 c& N" B! q  S. [7 W( E
190. }
     
191. closesocket(s);
     
192. 
     1 d6 |/ d. K1 }/ m/ i1 ?& u+ g
193. return   false;
     4 Q: J" ]! o$ o% s2 [- V7 W& l1 s
194. }
     # r1 }2 t5 M# W7 z

复制代码

天云上人

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

( q! U# S9 n" o+ b. K" G9 S1 C
' M$ w* m# ]$ y" o$ J/ _% n/ v///////////////////////////////////////////
: ]  ?4 T6 |7 N) L7 K8 x( [& I//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

$ I% w  I6 H# y9 K6 m; `5 _- [" w2 v
#pragma once
4 S  Q* ?3 V) E( Q#include <exception>
1 l$ M" u5 j! ~4 v
  ~0 m" ]3 A7 U# h' ]$ k6 c4 s
  enum TRISTATE{
        TRIS_FALSE,
# d1 d  }/ b" N- N2 y/ b+ K& P8 O        TRIS_UNKNOWN,
        TRIS_TRUE
3 @2 D8 l. a, j: A2 I};
# O& L& u7 A; M9 W

, F" D- }1 e- ^! [* r6 O6 xenum UPNP_IMPLEMENTATION{
% M) T8 F* K! s1 v* I        UPNP_IMPL_WINDOWSERVICE = 0,
2 A/ J- r5 h, \2 @0 K  e3 P        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};



$ s1 X0 i# w; z( H  k' B& v
7 |( y1 u. z" I& q( u+ gclass CUPnPImpl
( W: [8 H0 f. |) l, R  Z{
* R& |. K% r; epublic:
        CUPnPImpl();
0 N' g  j& m+ I8 e' `        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
; u* z0 ?- @3 w6 Z: K2 n4 r        enum {
                UPNP_OK,
* u' K( X  s/ Y4 d& r9 ?2 K                UPNP_FAILED,
+ q% B) t  i! x                UPNP_TIMEOUT
        };
+ a4 a) R! L1 N, I2 r$ p5 u

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
8 f6 D6 |( ^/ _9 W, C        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
3 F$ G' I0 W+ L  [       
4 m  c, z6 c( T2 y" G        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
+ o, z8 g$ N+ w: j7 H, i
% Z% b4 K- N5 q* {6 Y+ Z6 S
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
1 E0 d0 w/ y$ k        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
& g# P! E& |; ?- I$ V4 I        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

0 F8 ?8 T" j3 G2 e8 J# o$ {
. A7 B8 H9 `' x' K! h# L1 F& k' S// Implementation
5 ~' {4 y8 |/ V6 M  o# e4 Cprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
3 l, ^( s8 K9 R6 [3 u" d. ]        void                                SendResultMessage();
  u6 v4 B9 e. g5 i9 s/ s        uint16                                m_nUDPPort;
: K/ Z+ |+ u2 k( c# _9 x        uint16                                m_nTCPPort;
' u, F+ F: K* c        uint16                                m_nTCPWebPort;
1 e; F& N! S2 e        bool                                m_bCheckAndRefresh;
4 u( _; ?4 @( V
, x; \( Z2 E. j+ L: v5 T8 D
private:
        HWND        m_hResultMessageWindow;
; i8 E% ?2 n& D- ]* i$ d  ?2 Z  o        UINT        m_nResultMessageID;
6 p! v0 s# s5 Y8 @' j

};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
5 k4 t# i) P+ e) r; L0 K/ G/ H{
public:
( G1 D/ h- e% m* i        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; }
' l9 j* [  X6 Z6 F};


/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
& X% r) {$ G; Z. D' y
5 O6 ~0 S) [. ^" U4 x! K
#pragma once
2 V0 d* J# \2 S& v& e2 G#pragma warning( disable: 4355 )
, H# s- ]; b$ B3 R( |/ w! S
- n: O# h6 H9 y; Q3 C
#include "UPnPImpl.h"
) G8 B/ m' y: B9 h0 q! J7 r#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
  Q- Y% l4 d4 k' D4 @1 g

1 V  d+ B; I) R6 f7 Y* W* c#include <vector>
#include <exception>
#include <functional>
( o  O2 K& w# E' }
2 q; @* ^5 P+ ~6 `) q$ b
* `: |; d3 e) f5 x9 c1 J
4 \1 l& B6 u: r' G. z5 N* u
+ D. H5 o" e$ a' ^# Stypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
4 x. ^/ P9 L" ~& ^0 b+ \typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
' M; ~+ ~4 ^: ptypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
9 f  I! `  n+ `+ G' G! a4 s' ztypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
9 |& m# _1 M  G8 e  ttypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

( x5 e7 P9 ~6 F7 Z3 L
& y3 [& D6 ?5 B* y5 Y9 G( ztypedef DWORD (WINAPI* TGetBestInterface) (
6 h% o* [. `4 e1 i; [7 [  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
& M7 V$ l: g6 R8 F7 u);
  U+ Y* z7 Y5 g6 S% X

. D% u" w% G- o  Y; Ztypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
* d8 r( o* l1 ]: ~. p( Y  PULONG pdwSize,
  BOOL bOrder
);
. s4 s8 W6 S8 U

/ O; v% _; N+ m; jtypedef DWORD (WINAPI* TGetIfEntry) (
% j( T( F6 G' O- O# I: s% |  PMIB_IFROW pIfRow
: J6 y4 c) V1 [( C. g3 y0 H* E);


CString translateUPnPResult(HRESULT hr);
3 b( a7 ~+ J7 I7 \- LHRESULT UPnPMessage(HRESULT hr);
5 ]" T2 L5 f  {# G1 m6 E

class CUPnPImplWinServ: public CUPnPImpl
4 }5 \/ S8 i# `) C$ {( L# W% f{
  j, j& }+ }  a. H        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
; ^- J6 H6 N# W- P- Q// Construction
" ^# n+ J9 w# E4 p' i$ I3 e/ kpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
3 Z3 T! B9 y) O) w

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
) Y% p6 E/ i& o$ u0 v4 W6 V; |  u$ k        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

) f3 V, b* k- F/ A8 r' ]
8 [5 d/ X  K- L0 K6 f" b  |        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
0 I7 f, \4 e6 C6 P; h% h! [% t        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
+ y: F0 |* R% u. F! z  H2 J        virtual bool        CheckAndRefresh()                                                                                { return false; };


protected:
% D8 H8 H- j9 Q2 F1 C$ v        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
# L7 Q8 q/ q4 ?, i) C        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
        void        Init();


9 q4 k0 r7 t& r6 v0 Y        inline bool IsAsyncFindRunning()
        {
# U% G% D  M' K/ X8 ?. z3 E) }+ k5 Q$ G                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
! u' P+ E+ }, i% P! x# v                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
- C* B5 ^1 |7 u2 I                        m_bAsyncFindRunning = false;
                }
5 o% ~' m$ }! C/ ~2 a  J" ~2 U, l                MSG msg;
; p+ F* m0 U. @: [( e                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
. v7 T# E, {1 Q$ |8 }* e' F1 k, u                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
        }
) |' z' e, s: N0 L
# @7 `4 `# H& a. c5 K+ u
# _* H( e- b# i, l% z- K3 u& y/ @        TRISTATE                        m_bUPnPDeviceConnected;

8 O0 f- D+ N6 H6 o/ q
% [3 ]' \% Z& G4 _# m// Implementation
$ O8 F- N- x5 G3 j        // API functions
' m& K) W7 Q+ Z# p+ [        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);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
* {- q- s1 _9 _4 \' {0 V6 [0 t: a        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
1 c2 C& h5 {/ Z3 g8 x
/ c/ Y# a) O# P1 u; E3 r: x
- Q  v2 Z4 `/ {% A) \        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


; n; [" L4 ~4 P3 N8 |* F8 j        static FinderPointer CreateFinderInstance();
( s  L3 J& n9 J: \( P9 F' P7 N        struct FindDevice : std::unary_function< DevicePointer, bool >
9 G* A' P0 H; _3 }  z2 k+ p        {
; v8 f4 x4 I# l7 q2 e' P2 f5 l                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
4 O" }' s1 U. }# x- A                result_type operator()(argument_type device) const
3 b- @$ r& U6 j' Y; G6 g/ Q                {
- h( |5 c4 b8 G, {                        CComBSTR deviceName;
; {7 I" }5 `/ E. ?                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
' R0 ?; j, O! H8 ]) U& G
# L5 f! |' y+ j$ `4 _
                        if ( FAILED( hr ) )
- Q  B: C- p  w) J' x: w7 Q                                return UPnPMessage( hr ), false;
7 x& n5 _2 x) T+ B, P
% H  f) E- {" V/ w7 f2 Y& f3 n
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
$ l8 L, W) `$ x; g                }
1 b4 w" N3 E7 f6 p: v4 K                CComBSTR m_udn;
7 [% v+ n) M, T3 S" I( t; k        };
5 ^. t5 w1 w+ P       
* S. a: t. w4 Q5 ], R* L        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
8 f' W) H8 T; R+ ~        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
' d1 V; M8 ~) A. M3 d        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

- x8 H8 u" X1 _$ K
        // Utility functions
) q$ ?" T( C: \3 e5 s5 N! U        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
. r6 N7 Q: Z2 g$ K- D' j0 }5 b        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
2 n# _/ _2 m7 a+ [) X5 h        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
/ A0 s1 ^6 `5 `8 ?. i% W$ n9 S        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
; o% I6 m. T& B3 q" j        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
( k3 _2 K* F* o( h        CString        GetLocalRoutableIP(ServicePointer pService);

* {% H$ B  r2 w  D
// Private members
# g7 r: ?# f9 j6 _private:
        DWORD        m_tLastEvent;        // When the last event was received?
2 [0 ^1 o* a& F  w  E- \. X        std::vector< DevicePointer >  m_pDevices;
& d5 I2 q* l# U0 c/ @: A3 o        std::vector< ServicePointer > m_pServices;
5 Y  j1 H. S1 p) ?  \# p/ M1 a        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
! m6 U# V2 u0 V7 N3 k9 z        ServiceCallback                        m_pServiceCallback;
1 ^  F7 p1 p: k3 t* R
5 Q- {. b5 s. k! }" ~! D
        LONG        m_nAsyncFindHandle;
        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;
; [* _( N6 Y+ `3 m2 a        bool        m_bAsyncFindRunning;
8 V* K. Q% X7 U, B( ?8 W        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
( k) P4 u0 h0 C7 r3 @        bool        m_bSecondTry;
" x, I5 ]. ~9 D4 O" A! z        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
$ `- r! s1 f4 z. {/ F

  U/ s/ W! K1 T' n3 ~3 G& q8 O};
' d  I) u! W( o% L0 t$ {3 U
; [* O# h2 v0 H, p
- g, ]0 ~7 [% t// DeviceFinder Callback
class CDeviceFinderCallback
% R3 h4 d2 I# }+ C, M        : public IUPnPDeviceFinderCallback
{
) N  L% Q5 d! Y" \( Y2 q+ Epublic:
6 @4 H8 U4 f! d/ A1 f3 d        CDeviceFinderCallback(CUPnPImplWinServ& instance)
0 I; ~) P. f3 y9 M; @                : m_instance( instance )
9 p3 w  E2 n8 _& ?  `        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
' d- G* L9 |" {& y( A   STDMETHODIMP_(ULONG) Release();

+ w% Q7 n4 B# Q5 u. }
9 C( j3 ^4 Q) c4 m// implementation
0 [0 T+ v8 k9 D9 g& ]private:
1 w7 X! F1 }1 L; \4 w; h        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
* O$ T! y1 T8 F# v3 F4 d" i        HRESULT __stdcall SearchComplete(LONG nFindData);
7 Y* g8 Y- N! h6 ~# j& X

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
( \: m/ ]& H2 Q3 d* I* ^};
$ u6 T) i/ h+ {' @

// Service Callback
1 M0 ^" `) Z0 n+ m) O7 }9 D$ iclass CServiceCallback
        : public IUPnPServiceCallback
7 S; F3 X$ ~% p$ K- k/ X$ Y) A& S3 S{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
) G9 U" a' O) f/ ]# u                : m_instance( instance )
$ Q6 I) K$ t6 m% y0 T        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

( B$ r( i4 D* M+ s0 _
8 u' j1 m+ W* H$ o& S// implementation
5 I% f1 B: Y0 _9 T, i( o1 r3 _private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
! n. u4 g% ~9 F

2 @; F1 y9 U' q% O+ u  A1 Kprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


/////////////////////////////////////////////////
8 d" u$ y; v  T" v# j; Z

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( n0 N7 [( K$ ACUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
/ Y6 v- m! [/ r' PCUPnPImpl::DeletePorts删除端口映射.