UPnP

天云上人

/*uPnP.h*/

1. 
   0 u6 l: |8 x' M( b( D
2. #ifndef   MYUPNP_H_
   
3. 
   $ n: Q% e. M# ]& C/ g
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   4 v; x, a* r" ?
8. class   MyUPnP
   6 w+ g# i6 i+ `
9. {
   & C) u! g  G: W5 C. X
10. public:
    & ?% L3 {3 R) K+ A
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    9 l4 c3 F! ^  G1 Q0 ]1 |% g8 F
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    1 Q( @9 R0 L# D! d) K$ ]9 F
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    4 l' ^1 L/ F. q7 \& s
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    $ r  h+ M: x2 t; @$ M
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    ; a$ K+ V+ a2 O' W
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    # Z3 f( B  p9 C0 j1 A/ d
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    2 I. O. b+ d# D' p( _
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    9 I' V) f0 ~- F4 F0 J, ^& \
28. CString   description; //   Port   mapping   description
    % v- h  N$ |6 U& @  W2 e
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    
32. ~MyUPnP();
    * ~8 b7 o% g6 _
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    6 [4 M# H8 h4 A& a+ F1 I, Y1 U
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    # _" q* y0 I2 A" v. \- h
36. void   clearNATPortMapping();
    
37. 
    0 B  [2 n  ~, S# t- K; g
38. CString GetLastError();
    ; A, b* u* {3 d; h
39. CString GetLocalIPStr();
    5 L4 E( N( G3 B7 e  p& E
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    4 c- D3 ^" v; u+ m; s% \, G2 o
42. 
    # y% b# A5 _- W) {8 F/ u
43. protected:
    / O" k$ l* W- _+ x$ f: ?" T
44. void InitLocalIP();
    , _4 Q& e( T8 u; A0 f/ {
45. void SetLastError(CString   error);
    / C7 L( ]$ d1 f5 F; N
46. 
    
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    $ D8 v. ?% g. j" g" D/ ?' D
48.       const   CString&   descri,   const   CString&   type);
    / n6 X- i3 k" T+ j
49. bool   deletePortmap(int   eport,   const   CString&   type);
    . R. m! d4 J* n7 {
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ( H- V6 S/ B) }: Q" U
52. 
    
53. bool Search(int   version=1);
    8 z: Q1 Q# a+ S' L  \) q
54. bool GetDescription();
    ; b% l0 q9 i/ O7 B7 d# X
55. CString GetProperty(const   CString&   name,   CString&   response);
    8 }, f6 C3 N7 ~' C. Y" \$ N3 x/ z
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    6 E7 k# G" G% u7 {
60. CString m_devicename;
      u  i$ h, a" ~2 z
61. CString m_name;
    0 ?/ \2 S3 p3 ?8 G1 e& @" `
62. CString m_description;
    
63. CString m_baseurl;
    
64. CString m_controlurl;
    
65. CString m_friendlyname;
    " H% |0 W: b5 ~( Z2 c. A
66. CString m_modelname;
    
67. int m_version;
    
68. 
    7 y  {5 _) E! e; g' _1 M6 c; Y" o$ {
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    ( V: I+ H- a$ A6 o- m9 Y, d  @
73. CString m_slastError;
      e+ ~9 H" ?# K) A" n
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    5 U  Q! b# o  r% w5 [/ A
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   ( F# m/ J5 l; F& l& P
2. #include   "stdafx.h "
   
3. 
   " T1 M/ z5 t4 T
4. #include   "upnp.h "
   " c% ^$ o; k6 P; b, c& Y" d. Q
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   , ?6 i7 t. c, H' p9 U+ e
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   $ m$ e" P) }8 ]+ u% W' C" V
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' ]7 e- y. c/ v. A' I# |
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
      G. i9 M, j: B. l* k' P
11. 
    9 V3 @6 B7 S! {% a& ^+ `; u' ^' @  h9 d
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    % r7 z8 j& k' _$ P; U8 s8 O: [
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    - J5 l' n' ]* w* T$ l; q8 z
15. 
    
16. const   CString   getString(int   i)
    ; D" e5 E. l4 I9 ~! m, B( f* w
17. {
    4 P; B# T7 Q: P6 X! i
18. CString   s;
    ' D1 o$ @# ]+ c1 r* c
19. 
    6 X# L, J4 L! Q5 H
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    . L  ^) [0 `8 z' g% w
28. }
    
29. 
    & I$ \6 B0 k. A  T4 `% @! w0 g
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    3 ]' R' N' f9 \: [
37. char   buffer[10240];
    
38. 
    / d. Z, V2 Q, p/ J* O8 w1 K. C
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ; \& [9 {' B/ `( r3 p5 X
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
      k5 Z. m0 s* i# p" j
46. sockaddr.sin_family   =   AF_INET;
    * `8 K# y/ ?- V1 j
47. sockaddr.sin_port   =   htons(port);
    . F6 V. V3 V: c, g
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    / u! O% ~8 E2 a' T2 g
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    3 }' w5 B3 G( H8 P- k# g- R$ n
50. u_long   lv   =   1;
    ; b0 y5 u! c) _. k- O, D
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    ' n* l% s- V% G& e
55. Sleep(100);
    + B. l) V. R. w: j
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    ! y! d: w+ a. W, p% M" p
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ) d8 a0 n  Z/ x! K8 {/ k( c7 B
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    # g- p* O4 X1 b
62. 
    9 P. o: S1 @6 O. a- o
63. return   true;
    
64. }
    # m3 \! u1 _1 Q' L0 L
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    8 {2 V/ {0 x, S2 T
67. {
    
68. char   buffer[10240];
    # C) e# s! h3 J7 ~) d* C  ]
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    3 ?- f9 C+ f7 X8 R
76. sockaddr.sin_family   =   AF_INET;
    $ q4 b4 P1 ?7 e0 J8 r0 `
77. sockaddr.sin_port   =   htons(port);
    9 r1 i4 T8 L1 }! n0 m) ^
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    # ?6 N) P/ N1 b
84. {
    + `5 B& n! d7 d& W, Z  A( K
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    
89. result   =   response;
    
90. result.Delete(0,   pos);
    7 s& P6 o1 c8 k! j
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    ; o# |0 \7 b4 A  x3 F4 S* T4 z
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    0 |! F" [( S, s0 C$ E% d( r; Y) t
97. }
    
98. 
    $ q& W- \' R4 ]4 E' x/ {, _+ E
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    8 b& c- N( s7 l; k9 e: H
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     , ^$ r/ q3 U8 ?
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     . _3 n, v+ a( U5 m" d: j
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
       v0 A5 P& ~0 ~, u
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     $ j- V, [2 R2 f! N1 {+ d
120. 
     $ d7 t8 s% [0 r
121. MyUPnP::~MyUPnP()
     
122. {
     # X. i& N4 d6 [: M$ h+ X1 ]
123. UPNPNAT_MAPPING   search;
     * o2 w6 ^% V% X6 g: o
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     " d4 \# o( G. |7 `+ }5 {
128. }
     7 M, @, L" ?( J$ |- O4 X" h% E
129. 
     
130. m_Mappings.RemoveAll();
     ) p8 d5 E" V! h/ ~# W% R
131. }
     
132. 
     * W, _( Y+ {7 d
133. 
     0 N- Z- g6 `' r2 @! k) l
134. bool   MyUPnP::InternalSearch(int   version)
     : l% T  d5 l+ g" y. o1 ^/ E
135. {
     3 t. C; ^' a& G  c
136. if(version <=0)version   =   1;
     % Y& D8 k% M' L
137. m_version   =   version;
     ( a8 m$ q( ~, I! C6 W
138. 
     ) X+ q4 w; p: L
139. #define   NUMBEROFDEVICES 2
     9 m* K; u9 B& C( B( o6 O
140. CString   devices[][2]   =   {
     ' x9 S, f' g; }( V7 |6 O( r* ]
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     : f) t, ?5 d0 |& I
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     # L2 `+ h4 G2 f0 [
149. 
     
150. int   rlen   =   0;
     * e7 }7 l" a$ I- f
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     , B/ W1 Y$ B  }/ F
152. if   (!(i%100))   {
     
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;
     * V0 D; w3 T- g7 p
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 k( f: T& ]2 k
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     ' \) S% i1 M( \. H  _- o9 T
160. }
     
161. 
     
162. Sleep(10);
     ( _  m4 z+ @* |; O
163. 
     $ ]  {3 g6 }* F
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     " J: l& N+ `) r
166. if   (rlen   <=   0)   continue;
     0 L8 F  B! z* b
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     7 q+ u$ o$ e! y. n* \
170. CString   result;
     
171. if   (!parseHTTPResponse(response,   result))   return   false;
     / ^1 z2 |( S- f- [/ S( F
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);
     5 w1 d/ n/ M3 f0 T3 F8 O
175. if   (result.Find(m_name)   > =   0)   {
     - L  _% u# X* B! T
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     + M2 r# h# C; p4 z; M- c
180. name.MakeUpper();
     # u5 ^! j  r, I$ x+ F1 v9 X
181. if   (name   ==   _T( "LOCATION: "))   {
     - N3 U6 B/ k  o+ S0 W
182. line.Delete(0,   9);
     - F0 u; p$ Y" E3 f" q
183. m_description   =   line;
     
184. m_description.Trim();
     . @2 k. Y, B7 C8 c3 e
185. return   GetDescription();
     
186. }
     5 O8 d9 [+ m' V' r- \
187. }
     " k: ~. _2 a% S4 y/ d
188. }
     9 V: }% `3 r7 I7 |( I
189. }
     0 l, k7 i: s* w8 i4 G" g- N
190. }
     3 G) i) x7 E  c  z8 C
191. closesocket(s);
     % e  |2 M7 I' q5 G& D# C3 l9 |
192. 
     
193. return   false;
     ! ^3 J9 N8 q: T+ \1 E
194. }
     % S1 H! S8 U6 a2 I. W( x

复制代码

天云上人

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


///////////////////////////////////////////
! z* z' Z( T8 H  p# P  D3 S//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
7 l/ G9 }2 ]6 R, s
: k$ @4 K# X$ G% ]6 v* R- @
#pragma once
, V6 W% @" J6 }1 Z  C6 V#include <exception>

$ o( H/ y8 ~7 W+ Y, z+ o
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
( u) E' @6 D. K8 v, W9 e1 v+ T# Z        TRIS_TRUE
};


enum UPNP_IMPLEMENTATION{
+ j# b8 J% a& q& n        UPNP_IMPL_WINDOWSERVICE = 0,
3 |: {2 b" v# G        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
" e. A+ N1 k: o7 G* v2 Y



class CUPnPImpl
# V( m% r& k- j: P{
public:
% ]" K- o- ?# n        CUPnPImpl();
% n. J; s( y$ G/ W$ l        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
) T6 b0 O" P5 h( t( }        enum {
                UPNP_OK,
" p: f2 T8 S2 @3 b                UPNP_FAILED,
                UPNP_TIMEOUT
        };
3 c, }, Q/ [( T% t; S" d  h

$ J; g! K, E! j* j0 d* `: N        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
& [4 C4 n8 v1 y( ^2 }        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
- Z9 w4 z+ J' ]% \6 f- x) v* h0 A        virtual int                GetImplementationID() = 0;
0 ?- l  ]( t+ W' y+ h$ C& ~: {       
- @! o  U$ P# u/ ^4 a6 H2 A2 ]/ Z        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
, {9 @) f! }! `+ ?2 v4 K8 P

        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
/ V# A; O- W( u; s5 W* h- [' o8 B        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
6 D' K4 P* ]9 a/ Z  [

; W; M# R9 G" C7 R// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
0 o+ q3 y: b! c        uint16                                m_nUDPPort;
, g8 _6 i4 `, `/ j        uint16                                m_nTCPPort;
- d7 ^$ B, Q3 p% m- Y6 ^9 r        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

/ w* c2 ^6 h# _6 ~" U$ b
/ _; b* U& E% h( {% O. bprivate:
: ^  \+ Z& R0 R        HWND        m_hResultMessageWindow;
" L7 ~' }2 R) N9 g( K/ X        UINT        m_nResultMessageID;


' @5 v- g: }% Y' G: I! q};
$ W- j1 W+ F+ N- ~

2 M1 c. @1 h, O6 c& p* b// Dummy Implementation to be used when no other implementation is available
! J3 h+ Y# a/ m( wclass CUPnPImplNone: public CUPnPImpl
! Q+ `9 M( Y4 z{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
2 `: n; ?) o5 |3 W        virtual bool        CheckAndRefresh()                                                                                { return false; }
3 a  H9 [4 }8 h+ p# z        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
! q: d* p1 s: Q4 _* a* b( N        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
2 j: F4 N- y# p$ k2 _( C7 D};
+ l2 D( V* m' ^" U
3 r* d! }: b$ T: I2 H2 E7 _. w  Q* t0 k
/////////////////////////////////////
# z8 t% @; n$ {  N( V2 n# V) n& ?//下面是使用windows操作系统自带的UPNP功能的子类
6 f' G4 R& |+ y3 b+ b

#pragma once
, \, W  S- ]. N$ j; N" L4 p#pragma warning( disable: 4355 )
% ~6 }: |7 w( w- @" t

#include "UPnPImpl.h"
1 S  i5 W, \6 H0 J$ ~* T#include <upnp.h>
#include <iphlpapi.h>
8 ^; d0 }1 L, e' V+ W7 E! w#include <comdef.h>
#include <winsvc.h>
- Y; h( r4 s+ z2 l
% A) Z  L/ F* u. S. _4 Z) G0 B
#include <vector>
#include <exception>
- E* K5 i% i5 x) S' T#include <functional>




$ E) V  d. b1 G$ a7 ptypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
- P6 @; _5 E# z* Ctypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
5 q' Y) u9 @. z+ k8 P6 y" f+ p7 ttypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
' m% `+ H8 I5 {1 S* U6 K6 qtypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
2 f: o5 z+ `; ]9 ?typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; }2 C# {3 U% C6 _7 E' _+ e
8 x+ P, E1 Z) ~, B
  F1 T9 q* O( rtypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
: P. y) h% K5 F0 I% A/ c  PDWORD pdwBestIfIndex
3 d, b, m3 I% });
, c: h/ t2 u5 X' _9 z. u! ]
- `0 r( H, b. f) V% [) X
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
; X; V, A# o& T- Z; A# d; j. P  BOOL bOrder
);


" U9 w4 h+ \# A" vtypedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
8 W: x* N& X, |, O
$ T- I+ p+ Y  i; J& }+ }% y
CString translateUPnPResult(HRESULT hr);
# M& U* Q7 W( lHRESULT UPnPMessage(HRESULT hr);

; e( e$ B- w" i( B
class CUPnPImplWinServ: public CUPnPImpl
1 x1 W+ i$ r' n- a% Q* p( e& a{
3 f' A( V7 k7 F+ x        friend class CDeviceFinderCallback;
/ b" D3 l# _1 D% A8 A! r- U, u        friend class CServiceCallback;
* k/ G+ O8 M( b" s8 ?% X! q; h// Construction
! W! |4 O) @9 N1 c; Vpublic:
        virtual ~CUPnPImplWinServ();
$ b5 P( G' Q0 t& k. \        CUPnPImplWinServ();

/ {+ d! L- H4 p
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
4 I1 M6 K: G9 h/ `1 r
( ?" T+ X/ p7 p7 v: E( d
* ~! r* _4 e: \+ p' P4 O        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
2 [# W) |1 e! f: z        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };


' A! y0 Z& |: D: q1 X  ~, jprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
' G3 d: \8 X& _        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
0 `" D: H& b$ ?2 g& u        bool        OnSearchComplete();
        void        Init();
4 g1 V; H, l$ o6 h/ n

  N( k& B" E6 @4 {2 |  \        inline bool IsAsyncFindRunning()
        {
0 Z7 v( h0 u) W                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
  P" ]/ }+ Y3 q7 o                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
8 l, P: J/ [7 q6 W# P6 d6 J% ?; k                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
" q" }$ e/ h4 R) X3 ~                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
- B3 o  U4 g" _: f$ Z        }

( s3 V6 p4 O4 I  K% Q* l" y
        TRISTATE                        m_bUPnPDeviceConnected;

  W9 \: y) [- Y4 v4 H
5 X, j: }" U" }// Implementation
" p& w6 t3 s( L  s! f5 ~- x        // API functions
  F0 ?) s4 b4 N! S. H        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
$ l* z  Z6 i2 n0 G6 s% r        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
$ _" o$ J. T% z' M. U- Y) r$ a4 \) C$ I        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
' @* j9 f7 q% U" j/ A5 _* j* l# M  w

        TGetBestInterface                m_pfGetBestInterface;
1 U/ v, [, g: X: X9 x' v6 C        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
/ G. H" ~/ b1 P8 f) t# h' q
; c3 K$ X3 B" M# W8 H- |& k
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
0 X2 ?9 \# `. L9 e. l$ S/ q- a& V                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
( n" p( i3 c+ h3 C* B                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

  @: H/ d  w7 k; m
                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;


( `( Z* d; l9 f: O4 v$ ?                        return wcscmp( deviceName.m_str, m_udn ) == 0;
3 e, b1 I& @; ^: g4 u* K                }
9 k4 X5 O1 o, |( w                CComBSTR m_udn;
1 g" ^7 g- b. b" w2 @        };
  e: q( s+ |5 [& @, f       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
9 t8 d4 M0 K6 r: i2 a" D1 U        void        DeleteExistingPortMappings(ServicePointer pService);
+ Q6 e6 y4 O' l; k        void        CreatePortMappings(ServicePointer pService);
: T6 k0 z3 ]; Z        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
' e9 r6 l) U5 }; ]' d. t9 o        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
% E+ l4 ~  \2 C7 ~# K) k7 e        void        StopUPnPService();
/ h8 @/ q5 d+ L& Q8 k" r% V
8 A4 T# n" c5 @6 D" a3 S  B
! {8 h* F* v4 B( g! s        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
( j/ X9 H/ t# a7 G1 N; l3 p: G        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
7 f9 G; i, k# E& S& H        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
# I6 `' K, }6 d! C5 g( p        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
. |  l3 e1 X8 g* q        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
! h. d$ Z; N5 n: J) V        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
/ d: f3 B% s; ^& X6 K6 W

// Private members
! W1 t% @6 l6 s: ^6 D3 Yprivate:
3 |( D% n' ?8 j  j6 B! m# s+ v% o        DWORD        m_tLastEvent;        // When the last event was received?
1 X3 W2 ^: s8 n. L        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
# H! @& R% R& G- o0 X

# u5 J/ l+ Z) I9 ]        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
! d5 U* s- h, G, E/ F# ]4 g0 m2 Y$ J        CString m_sLocalIP;
        CString m_sExternalIP;
" R5 V: K& i+ w9 S& n2 R        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
0 h: J: k3 z! V3 d6 E0 P1 W        bool        m_bAsyncFindRunning;
( ^. b) `, r' G) J( X/ W& c        HMODULE m_hADVAPI32_DLL;
* j; \! _7 o0 G+ \        HMODULE        m_hIPHLPAPI_DLL;
  v# _1 g2 Y' T" M/ A! V( K, y. T        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
: h5 y+ \5 p, P+ {4 g        bool        m_bDisableWANIPSetup;
8 r" M  y% ?% ?1 t/ X6 M        bool        m_bDisableWANPPPSetup;

. Y7 c$ p/ q1 ]7 ^5 g9 ?
};


// DeviceFinder Callback
! y# k  S7 g# E! _class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
& {/ C: U$ e; L9 g. p7 F) a2 Ipublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
4 \. W8 B+ Q' a  b        { m_lRefCount = 0; }
( _* F) n9 C4 @& d3 \5 F  B

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
6 {/ b; f$ E- H% j# `   STDMETHODIMP_(ULONG) Release();


// implementation
4 {5 x, Y, G" u+ x) [$ v7 H1 z: xprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
# }/ J/ v' ?8 H' |
1 [3 ^. S. I2 T, m2 g: e/ U7 x
! V& }+ A% U% Z5 L# w& @private:
        CUPnPImplWinServ& m_instance;
% T$ [7 n6 ]# G; \7 V* A: c        LONG m_lRefCount;
};
  |) N/ M' R# b2 k) p, F" e
# B9 ]* _. H  N* V4 q6 h
// Service Callback
class CServiceCallback
( r9 K! Z2 D3 p% }9 X. p6 i        : public IUPnPServiceCallback
$ U2 A9 K  `0 \/ c{
# ^' T& j  R. s5 e1 |9 upublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
8 @" s6 q) c( `   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
' p4 m- n; _) C) V- M% l4 A$ ~* J
/ n3 f" h; K6 _) i. k
// implementation
, y0 [" ^' ]) d! @private:
3 m# H; \" m& P* |! I0 t+ q0 A        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


  f5 Z+ @! W) M) ?1 t8 v& q8 bprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
6 Z2 w" C! ~" b3 @# w& i9 O# ~};


8 |) `( [7 V7 \1 g; w( p/////////////////////////////////////////////////
: H6 Z  _7 i/ J. {
! f$ e" Y- u4 I; X
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
( I; x. z$ A0 T% ]0 u+ W$ LCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
1 k5 W- t4 l* |! w+ }! FCUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.