UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   4 `8 R, L8 m2 `0 r! j+ x, S
3. 
   3 v) @5 b- ]9 z- H. f) a: [. {) J
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   2 k; A, V- q6 n7 d3 _: R
9. {
   
10. public:
    
11. typedef   enum{
    
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
    7 K1 j( {) W& ~* k# f
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;
    + ~& o% |* c" B$ l1 W0 _$ y, g
18. 
    
19. typedef   enum{
    ' I" o& n: B! l7 s! K
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    * C' u0 y. P7 v7 J& _
24. typedef   struct{
    3 t. E$ F/ }! z( A3 [5 j0 v
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)
    3 l  v  w. X8 X; n% [' n
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    
30. 
    + H8 ~! J0 H, a( Q: a+ O% U. v1 g
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 t: F0 g4 J: t+ ?8 i
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    5 n; r1 J. J6 o% k9 z  }
36. void   clearNATPortMapping();
    
37. 
    8 B1 `; O- C2 C- F4 \* d
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    ! ?9 [1 C1 d# i1 }- W1 P
41. bool IsLANIP(WORD   nIP);
    7 ]  p# A0 p* X' ^
42. 
    
43. protected:
    
44. void InitLocalIP();
    / u* b6 O! W  n3 g
45. void SetLastError(CString   error);
    
46. 
    2 {( t# Q3 G/ o% }3 J9 w
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    " I' ~8 m2 t3 d( B
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
      x  w% L& I5 U. ]* x6 n5 s
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    " Q% w, @3 }% ?
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ! T. ^% A. a+ q, x
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    ; q' S# M1 I# @; G5 z9 D5 R
60. CString m_devicename;
    * E+ e' e* J+ \7 m5 ~; g
61. CString m_name;
    
62. CString m_description;
    + k, v9 v5 s- t% `/ s$ R# B
63. CString m_baseurl;
    
64. CString m_controlurl;
    ! P2 L: Z7 }+ W, \) F9 ?
65. CString m_friendlyname;
    
66. CString m_modelname;
    % N. r$ C% x5 @9 \1 r) o
67. int m_version;
    
68. 
    . n; X/ A0 z# @
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    + H  G3 }. |# {3 F/ ]$ K0 |0 `
72. CString m_slocalIP;
    6 Z; ^( `8 b4 C* {
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    8 e, b4 E2 L9 X2 ^- s
75. 
    1 B" Y9 S/ k# F$ r
76. bool isSearched;
    & e+ {) H6 \1 U" N2 ^
77. };
    # b3 `' e$ K& d: o) V
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   1 p/ G9 D6 c  B( K3 d6 Z- D5 y
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   3 h% K) Y( u  s8 u9 N" Q
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   : L3 d* V3 `9 N* ]: G1 o' h
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   9 v: o& s! i* S) s# X( G
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    * ~) {5 K9 w& F% h! [4 ^& W5 X3 G3 L
11. 
    / f* e+ ~( V& V
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    2 l- y! j4 O& @9 Z( q" A
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    # u) C' d$ j  y# z" r4 ?' i% `& R4 b
15. 
    ' }* z- B6 b2 F4 _; v) {8 I
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    
19. 
    . k+ V% G5 b* f* N/ ?' i) v; k
20. s.Format(_T( "%d "),   i);
    
21. 
    4 I$ Q6 I" A1 e' j- z; N
22. return   s;
    
23. }
    , d; g: o6 w# ]# f
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    - n1 }4 E. x. N: \9 [# U
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    8 f" o& i) J& e0 b$ d6 i
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    " Z  g6 {; P& X% K' L/ Z
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    0 s1 ~  h; n* A- v8 Q* B
37. char   buffer[10240];
    . T% i7 z9 G, v
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    ; R; E  z& J) `$ g
41. strcpy(buffer,   (const   char*)sa);
    9 x, b+ a2 `+ B5 g% Z; P, W3 O
42. 
    ( a: k" q6 p1 d3 j
43. uint32   ip   =   inet_addr(CStringA(addr));
    . g5 \- d# f% _; j2 H
44. struct   sockaddr_in   sockaddr;
    0 {4 X! Q4 X% ]) Y( s6 \0 o
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    & D- O6 P# E' ^+ `/ ~1 I
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ; }8 Z' g% T% u# }
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    7 F7 q2 N$ r( L) Y) [
51. ioctlsocket(s,   FIONBIO,   &lv);
    ) L) @0 d1 }! p$ [* q7 B
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    % s! b9 I9 b6 y# B3 E; o$ C
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * K/ J* U3 _) i% _0 p
57. closesocket(s);
    $ A5 m' x& ^3 o3 v
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ; B/ s9 N+ d7 U: y
59. if   (!rlen)   return   false;
    
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    : b3 V" m0 d7 x, p& E. W8 i
63. return   true;
    1 t4 I  ^- a3 u1 q, R
64. }
    & T& H' W9 u, r6 N3 \
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    7 }; ?% D/ j( |' c7 [+ V9 K
67. {
    - S; W; I' C) V# Z. T8 @( h6 R& ~
68. char   buffer[10240];
    - ^+ O# Z0 T/ H! i' l( \
69. 
    ; d6 t4 W8 z8 ^) b
70. const   CStringA   sa(request);
    " {$ r2 P1 x1 F) n/ `7 h$ K
71. int   length   =   sa.GetLength();
    . _  e+ m- @$ I2 ^
72. strcpy(buffer,   (const   char*)sa);
    
73. 
      g9 q6 c$ H5 ]+ u- R  l
74. struct   sockaddr_in   sockaddr;
    1 e; A' q/ O) v
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    5 Y  j( Y, V; F# \
77. sockaddr.sin_port   =   htons(port);
    # F, l/ ]7 i' j/ }% Q8 [
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    $ C# @- a- M' r) o
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 ?- c. o# y% Q3 S+ _; \
81. }
    
82. 
    : _. M5 K9 M0 o! D7 ^
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    6 X: o/ z* F; t
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    8 P: j' {! X6 p: B
88. 
      c  }$ m; P  Z( S- t! U' e
89. result   =   response;
    # V4 G7 \$ d  j- q
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    9 i3 U7 Z5 ?# ^* D0 \3 k" H
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    ! _: ~4 y6 B3 B# L% V  K3 E
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     : R( D$ j2 m6 E# [$ K0 g
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     ( y9 C5 H# N8 q+ y' y1 D' U' D8 g
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     . h+ ]: k: P' |
107. 
     . z5 H7 H) |+ \1 ^4 a
108. int   posEnd   =   all.Find(endTag,   posStart);
     : ^( G4 |" R& Y2 e4 n" u( K/ w- \
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     4 A% M$ \% X. `. d
112. }
     4 _0 ~1 H7 d- c& O) g4 i  J
113. 
     # E1 z2 f( _' e6 }- r8 c
114. MyUPnP::MyUPnP()
     ' k' {3 y4 C& v) U) s# `- |
115. :   m_version(1)
     * u9 i# ?! z4 Z/ p; d7 q
116. {
     
117. m_uLocalIP   =   0;
     , l; O- `7 }  J$ N
118. isSearched   =   false;
     
119. }
     
120. 
     7 d1 v8 j8 l, Y" n
121. MyUPnP::~MyUPnP()
     ; W: w- r& p  B7 _
122. {
     
123. UPNPNAT_MAPPING   search;
     - l( U; V8 O" D& y; C
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     3 L+ A9 S0 u8 m/ C2 H( T
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     
131. }
     1 {4 `) U0 u/ ^6 r/ E8 _. m
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     + @& F, `) p) a2 b3 ]
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     : |, ^3 A( h6 z
138. 
     * t9 L( h0 s7 Y; ?1 X
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     1 y9 K4 T  g( P  e/ Z# `6 u8 Z% R# A* Z
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     + |: o* l$ z( ]6 `2 b6 m
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     9 K' F2 ]  B; V1 O
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     : L2 F( n% v3 x+ |! q
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     * _/ I5 M: v7 E& L" |* |
150. int   rlen   =   0;
     1 ]3 s: j4 k' h! B0 r7 u
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     & Y/ q. {2 A' x9 i
152. if   (!(i%100))   {
     - j* o! s/ ~$ M% y4 S( d/ u
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , H* j% p3 P. h6 f8 _
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     / n9 E: g- e+ v; `: C
155. CString   request;
     & ?% l6 H* w# T6 m
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     # f5 N# R  m, P) b, [0 p! ]% Y  [
160. }
     1 h: K; ]2 v( \
161. 
     
162. Sleep(10);
     
163. 
     ! J  a7 o+ }, c; C! i7 {- x
164. char   buffer[10240];
     ( i. s) q4 t2 j0 a, w9 q3 j
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     0 \2 D8 b  I. I2 K6 v
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     % o% a2 }+ U# Q8 @
171. if   (!parseHTTPResponse(response,   result))   return   false;
       m2 G# ^5 G6 L/ |; t/ @
172. 
     9 s9 _* g7 y" r% N
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)   {
     4 ~! J. X: l) H/ h. _
176. for   (int   pos   =   0;;)   {
     $ `+ Y# Y" |6 i1 N+ _/ q  ^' _
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     3 \+ t) H6 e7 x+ l8 q" S
178. if   (line.IsEmpty())   return   false;
     1 X6 T. C' y5 b, k) T3 X$ l! B
179. CString   name   =   line.Mid(0,   9);
     4 q$ H# e' b% D; p) R& j
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     ! R6 V( n8 d# C6 L, ^( a! k- L
184. m_description.Trim();
     3 y1 S) U! G, V4 H- ~8 Q5 l
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     7 P$ e; ]+ q# j( i7 L" _4 g) D
190. }
     
191. closesocket(s);
     
192. 
     
193. return   false;
     
194. }
     ) G* u1 w' p, b2 r3 f1 w3 |* K2 `

复制代码

天云上人

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

/ q- N0 O9 `; i& F
///////////////////////////////////////////
4 ?/ F7 B  j6 y# [//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

! v9 j' E9 H7 A" o% |: P' @
#pragma once
#include <exception>
) n8 ~0 |" ]) p! w

+ f7 y2 @3 Q9 ~  m4 ]# m  enum TRISTATE{
6 _! d8 e1 K. E. {3 c8 e        TRIS_FALSE,
        TRIS_UNKNOWN,
6 O) n- b3 i$ T+ h        TRIS_TRUE
};

* j0 x, k% X  n7 G) ?
# m7 E. f& G% p+ ~8 w7 G& z, j; Yenum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
0 B1 d0 G. A. y$ u+ B        UPNP_IMPL_MINIUPNPLIB,
# z7 g- a1 {3 Z1 U( G* r) Y        UPNP_IMPL_NONE /*last*/
1 O8 d; g/ L$ k9 r4 `, K  p};
, `4 V" X+ W$ V; b) f, q  ]
: @# a" w7 b0 G; M+ b6 ~


class CUPnPImpl
{
4 a/ S/ I2 l& h7 W5 c. Epublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
2 Y5 \5 G! x9 i& o! o2 m        struct UPnPError : std::exception {};
        enum {
2 I7 [" _- n* Y6 v9 I+ V                UPNP_OK,
; u) o$ e4 m2 k% I+ [                UPNP_FAILED,
7 j/ E4 B% {- w0 u3 ]2 C3 p( `                UPNP_TIMEOUT
        };
! E2 S2 L+ y( D) Q

2 W5 J/ _1 E/ o: I2 t        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
. w# a0 q6 d1 k7 ?5 d        virtual bool        CheckAndRefresh() = 0;
( J! j1 S# q5 i8 T+ V* o- q        virtual void        StopAsyncFind() = 0;
% r; N5 G! ~- ]( Q9 b        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
  J% f, @" O, O4 t! X( s. _  @        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
# I: \1 L6 j% f& Y; j; B3 g' Z$ b
6 T9 A6 t4 k) N! J: E& t5 B
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
  @+ I4 S9 Q0 x6 l' `/ h        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
0 J! J5 q% ^/ i$ [) i        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
% J1 Y( k$ j9 Q/ [6 G# k

1 O% o) }) B0 i$ b// Implementation
% b7 V# B" _! ]) ~8 Z4 {" D8 pprotected:
$ s2 E5 I" W) m# }3 l5 I        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
. r& d* Y1 X) J% z1 }7 i0 [        bool                                m_bCheckAndRefresh;
; m1 N( Q( V% f! @5 S1 F* s$ M" ^

private:
, e' @! G3 n( h        HWND        m_hResultMessageWindow;
( o! I+ w! d) S6 @; _6 o        UINT        m_nResultMessageID;


, A4 p. W7 K. I; l2 V. V3 U2 w};

8 r9 a% g- }2 j* k8 a* V
+ ~; _4 i1 u7 Y# S6 }: I3 h: L// Dummy Implementation to be used when no other implementation is available
; n3 ]( L2 G" r& B0 j3 M# |4 G( jclass CUPnPImplNone: public CUPnPImpl
3 o% S2 B4 ^# s, @/ P{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
3 {: T+ G( C2 ~  L        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
- M* N6 O9 ]- o: Z2 I7 W3 }        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
, |' _" h& I* H, G};
( |8 K% Z. e2 O2 ]. @9 d" D2 F! T
) {3 V. l9 F- D4 f3 v
/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
/ K( l5 x; Y# \3 a9 B
8 Y  x6 C5 I, r/ Z& ^
* T& V. b2 A2 N5 J- D' T. r5 v4 Z#pragma once
) U- _/ {3 j; Z* D#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
7 ]  p7 P% p! @#include <comdef.h>
#include <winsvc.h>
, m% ?( J  W; ]0 i5 `
5 j  @" \5 W! G9 i: j) x4 n7 C
#include <vector>
#include <exception>
# e1 k* O) J5 T; f2 j#include <functional>
8 A7 i4 j& M7 u+ t
; D/ T8 a3 j7 v- P
. b* n- m( L9 A. U5 L: f

' G" A/ a& R3 s0 ]2 k$ q6 x2 ^typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
( m, A5 r, A) S( M. Ctypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
7 \/ c. ~! j0 {5 s3 xtypedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
8 \' g+ n6 c5 `) e8 s: G; btypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

4 S& W: W# R8 p
typedef DWORD (WINAPI* TGetBestInterface) (
8 K$ t) Q( U: x0 T( q  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);

* u) w+ t# l: Q
5 X9 a! D- S6 }' k( Ttypedef DWORD (WINAPI* TGetIpAddrTable) (
, _* Y# m8 r. }- I/ K  PMIB_IPADDRTABLE pIpAddrTable,
5 m, a# T. V- Y  PULONG pdwSize,
# d5 S4 q( ^3 N8 w# [  BOOL bOrder
8 Y% Q1 e# Y( L, M);


9 M7 R" m$ `' Xtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);


0 O& K; r$ a' ~CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);

; X3 v7 B  N" H
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
8 W1 o2 t, {2 r// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

, O+ M. O0 h$ K2 `3 Q8 Q% C
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
& c  j3 P0 E* {        virtual void        StopAsyncFind();
! m, d. D5 N" {6 o/ y9 k5 g3 S8 X        virtual void        DeletePorts();
        virtual bool        IsReady();
( q% d& I" x: |4 o* b  t9 \, G) A        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

9 m5 t% ~; j& I6 c. z
        // 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
& L! \% e) o2 o& Z! U5 U        virtual bool        CheckAndRefresh()                                                                                { return false; };
, g/ s4 W4 p5 G& }9 Y

protected:
6 X! Q: |/ M- G        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
" a! R/ |0 e. `! h$ B, d9 c        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
" R( O9 c7 a3 S- C1 v6 d        bool        OnSearchComplete();
        void        Init();
, e% G8 O. ]% {) z2 d* I& j
0 u3 E- q+ C# x# k( s8 `5 P
9 m2 T) u7 n; y! s        inline bool IsAsyncFindRunning()
        {
$ l1 k( ^! R# g                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
. A  o9 w6 X2 \                }
                MSG msg;
- M& u: t8 s0 p/ ]0 W                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
; ]' i" J. m9 l5 b( f) A5 R! {                        DispatchMessage( &msg );
                }
4 G1 C' f9 F; o% M; {6 _0 d                return m_bAsyncFindRunning;
        }
4 [9 A: Q  m$ g) c

+ v- I: H" s+ e2 W4 U. ]        TRISTATE                        m_bUPnPDeviceConnected;

; z) g" q" V! P! }
  r$ r4 ]$ p; a4 {/ K// Implementation
        // API functions
        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);
5 ^0 K) s; `/ N. H# B3 O        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
7 X. @! r" }1 f) N/ Y        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
/ i& v% @5 f5 E) n) ?6 o" ?& ~
2 d- l$ s3 m" c0 l( ~5 v
+ [! }" W# ^  B; k$ K1 D9 l. |        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
( \0 F+ f" R0 ^. H3 J! w        TGetIfEntry                                m_pfGetIfEntry;


1 a3 t2 _8 @' s8 B        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
! ]' p+ o( G3 F" [                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
1 T8 Y3 N$ x: f$ m4 S9 F                result_type operator()(argument_type device) const
- T( u9 e( {4 w8 `9 {                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

- }# d- @: z# K
2 F$ u( {" {. v, }( Y; O                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
9 s, l  `. ?3 e( D, ~6 n
# E3 V- A4 A, l5 h
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
7 o; A+ [7 a/ N) s        };
; e+ R# ]# ?& a       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
) l6 v! M+ \+ x3 A7 L, T7 f        void        StartPortMapping();
9 Y6 J: r2 ]' W, z. Y0 w8 B9 v3 t        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
) [4 ?9 L6 m: ^( j& T        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
% z4 S" z8 d/ A8 y5 i! R                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();

6 e- k  u8 O! l! k- {
" X" p: A) e$ w7 u+ }        // Utility functions
# n7 ~9 A7 k9 O# I$ i        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
4 a: T  A7 ^8 C6 X        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 d: {0 F* r! g; g        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
* n/ [2 T/ ]/ \6 N) ]$ A  T        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
; Q4 Y  y4 b3 ~; \        CString        GetLocalRoutableIP(ServicePointer pService);


; p# k8 k& i" r( {6 t3 t// Private members
private:
. U4 E3 O  {+ M5 v  X3 u        DWORD        m_tLastEvent;        // When the last event was received?
6 `9 `7 l1 E/ f( H* C/ r        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
8 S2 k7 y- W: j3 d        ServiceCallback                        m_pServiceCallback;
4 U6 O9 o) F' W7 z) I
+ G- b' u8 O, v7 ~
6 q* [! P1 B- M* j5 L        LONG        m_nAsyncFindHandle;
! K% ~! V3 z1 e9 s        bool        m_bCOM;
        bool        m_bPortIsFree;
" n: [1 D. X$ w2 \! Z        CString m_sLocalIP;
        CString m_sExternalIP;
% D5 C, [: }3 b0 z, w  q) v        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
; v0 K: }% d1 s/ b" u        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
4 E, w0 F' u' ]6 x, G& b; K        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
) S# R, q& W" ]4 W& H
% x; m+ c% B- j: W
};

6 y  N6 k& k' G, V- a
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
/ O, S9 D. d8 O) X0 n        CDeviceFinderCallback(CUPnPImplWinServ& instance)
- G! b0 m0 S' @0 c. j- m! ^                : m_instance( instance )
        { m_lRefCount = 0; }
0 ^( ~2 p6 O: `7 i9 r4 b, x
5 {9 Z5 o7 g, D7 r  F1 O% d1 O
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
+ ~+ P: ^( [: i7 f3 J/ [8 ~   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
- a" ?* ]1 A" [$ r7 C, b  l) C
8 ~6 `$ e9 ^: w" S7 m# @6 n
// implementation
: ^( \! c3 ~5 L0 Sprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
/ m7 C" Y  {6 ]# A- M1 ~! L        HRESULT __stdcall SearchComplete(LONG nFindData);

: I& W' b2 q' h$ z+ F+ M/ Z
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

; O4 m- ^( n/ `) n- n- e1 N, e
5 z6 w" e- H5 A( b# n" C0 T// Service Callback
8 T) e- G5 b, P. F; Gclass CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
/ i3 `) I1 l5 k% O3 h                : m_instance( instance )
        { m_lRefCount = 0; }
   
# g! i  r* N$ D6 t! o   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# P9 ?+ @* q' G% n4 D   STDMETHODIMP_(ULONG) AddRef();
0 E2 a; z& j1 b   STDMETHODIMP_(ULONG) Release();

  Y. u" A! q0 b1 Y& l
// implementation
8 z6 ]2 h, S, gprivate:
( l2 |/ ?+ X" `  S: N& o' b' f        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
! v% X! [# W: h* z' w" {1 v        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 Y) D9 r8 q1 k( P. y
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
1 W, D; ~' P/ S; P. k};


/////////////////////////////////////////////////
' R* Q1 ^7 i! ~! h9 S

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