UPnP

天云上人

/*uPnP.h*/

1. 
   4 {- ]3 U1 C. h. {+ k- m. }
2. #ifndef   MYUPNP_H_
   
3. 
   " t. e( S1 o) l( q
4. #pragma   once
   - K' o8 \) B! A: A
5. 
   + p, L) _9 S$ X
6. typedef   unsigned   long   ulong;
   ! ]3 W# s5 H& B0 ^7 v5 D
7. 
   
8. class   MyUPnP
   ' J: j2 _0 K9 a
9. {
   $ k# f" f5 g" z
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    ( x+ i/ B( D& r9 R' q% E! U
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
    # d; C) k* Q; ^3 c" y
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    & v5 ^3 r$ o+ n' l8 _3 B( M
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    + J7 a4 d. A$ A9 c, k! ^# `. u
20. UNAT_TCP, //   TCP   Protocol
    - p5 u* t; U% c: c. \# o& j
21. UNAT_UDP //   UDP   Protocol
    $ P. y, w% c; E7 k7 k$ W- \
22. }   UPNPNAT_PROTOCOL;
    5 T2 E2 ]  o5 @% m; A
23. 
    ! Q0 n- H- Y5 t
24. typedef   struct{
    
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
    : x0 {3 @5 R* ?0 }6 A
29. }   UPNPNAT_MAPPING;
    
30. 
    7 h" A8 C5 z8 i& K  B
31. MyUPnP();
    
32. ~MyUPnP();
    ' w! L+ W6 ~% H/ @* c8 A4 E3 f
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    , x" Q% q1 a# S/ f  z: ?! l# R' ~
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    . \5 q5 U1 L) M
40. WORD GetLocalIP();
    9 e7 @* p! N$ M6 g( _2 M* J
41. bool IsLANIP(WORD   nIP);
    ( E0 v* L5 Z/ [4 F
42. 
    
43. protected:
    ) l2 k2 R1 O' l- i  p' X7 j. y
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    + C) |+ y, [/ j
46. 
    6 L5 n4 L( s% R6 O) r5 f
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    9 _/ j- Z7 W. o  r9 L2 t
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    5 L+ D+ x8 r0 f: y4 Y
60. CString m_devicename;
    
61. CString m_name;
    
62. CString m_description;
    1 y8 Q/ t$ Y( X. E4 r
63. CString m_baseurl;
    
64. CString m_controlurl;
    , E. n$ @/ F/ \
65. CString m_friendlyname;
    
66. CString m_modelname;
    0 K& }2 d6 `4 K9 i* U; P9 c
67. int m_version;
    
68. 
    . }2 s$ h7 S8 B2 Z  ~$ l8 Y$ z
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    ( W; Y% x  Q/ T7 q, j
77. };
    " S9 Z! w- a6 _0 h/ H, L4 K4 Y! |; ]
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   ; @: v* o& C  a5 Z, E4 [
3. 
   / Z/ s! {( Y/ |% t% z8 }- j
4. #include   "upnp.h "
   
5. 
   ; \; q* z0 e# d/ ~" C
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   9 @% M8 |* O+ J( M
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
     Y% B) F$ [3 x# t  q
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    8 [' p6 C- T% B! R6 @( w" H
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    9 c+ i) Q2 H3 k1 |6 d
13. static   const   int UPNPPORT   =   1900;
    6 {4 U% I% W7 X! Q6 O9 d. l9 O
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    : r+ e# _8 T8 _# e
15. 
    9 X# A. G3 g+ z- R4 @/ ^
16. const   CString   getString(int   i)
    
17. {
    ; ]' N: ~' I  T
18. CString   s;
    ( u  M, I. i9 U; _
19. 
    ' X; r" _/ k) {- C0 M7 B
20. s.Format(_T( "%d "),   i);
    , c+ z+ O- t% `3 H* T. |
21. 
    
22. return   s;
    & U  H* w# e. O+ ~% ]0 N
23. }
    : v/ y8 {1 s/ X" M& L& }, @; c
24. 
    9 A9 [- H7 G' L6 C& l4 D$ S, u
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    9 W: G  Q) r9 R8 Q- t6 I2 a
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    * ?3 \8 @5 S  D+ T3 h6 l4 X3 a
28. }
    2 B" d1 j9 e! n) l
29. 
    2 a% @" V: I6 k0 j+ h3 B
30. const   CString   GetArgString(const   CString&   name,   int   value)
    & _% M/ O5 O4 G, H9 D! f
31. {
    7 d) m. Q6 ]# {# E; O
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    + d/ C! w/ y- O1 _6 r& f
33. }
    ; s5 S# X' [5 T5 d
34. 
    4 Y# X. h1 y; F+ b) H; c
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    / I. Y5 C3 D3 A# A( b, V; K
36. {
    
37. char   buffer[10240];
    + e0 z. m% Q; H7 J3 j
38. 
    
39. const   CStringA   sa(request);
    2 n# t5 w. l! ?/ v! F
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    3 Q/ p) z) s+ Y2 x
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' F! x0 z' |! p3 P6 N
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    * t; X- J; r' n' z
50. u_long   lv   =   1;
    2 n6 h. v. w; j& Q6 c# V
51. ioctlsocket(s,   FIONBIO,   &lv);
    9 L. l) X$ O4 T# Y8 m
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ( r0 N- P- W! K, D5 T0 t. x) q8 L
53. Sleep(20);
    6 }, i1 I4 x# l
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    $ t( y& o9 u5 f. N% b
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    5 Z! |0 o2 t! W2 i% Q
59. if   (!rlen)   return   false;
    
60. 
    5 z# b2 _& }# D6 ^) o
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    9 X! v# \0 @8 ~3 o
64. }
    2 Y/ y0 w% z0 L8 |" l
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    & W$ ^3 B+ p. V" E" l' K, d0 [
68. char   buffer[10240];
    4 J1 \$ d: s$ {
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
      L& Z7 P# O- o1 f' q- l
74. struct   sockaddr_in   sockaddr;
    # I3 V# O1 {$ Y* G
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    0 n# B+ y, h# i4 A8 F" e3 S
77. sockaddr.sin_port   =   htons(port);
      f5 u* ~- y& _2 m3 r6 ^$ ?' r, G
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ' e; }/ _; ^; t1 |
79. 
    6 N2 Z' O4 T7 l8 N
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    8 q( G5 |8 n+ \- w8 h( s6 o
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    . n7 a3 j2 h! z. R/ {  V3 A
84. {
    
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    9 u5 u# D, r: t, D2 o- S; W( Z$ G
88. 
    
89. result   =   response;
    ' Q3 x6 J) h: A! `8 @; B# _. g" s0 a
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    0 g: ~, O. u! J3 r5 i/ x+ i1 [) p$ A4 q
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    9 R0 k% g9 \* z: z; W8 B
97. }
    
98. 
    + z1 b; D( B! g4 t- U7 s# R
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    7 T% P7 M* ~% \" L
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     , U0 Q2 L0 U. ?5 s3 C
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     ; a9 m* Q; ]4 J7 K" G  j. O
107. 
     * [5 @/ z$ c5 D/ A9 e  ~1 l
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     & \, [* L( E  F" [9 M- B
110. 
     " u  E- E- }! }2 n. s4 j9 [, E( k
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     % V1 L0 W$ N4 _3 a4 x0 y4 {
112. }
     - ~$ i/ t: m$ \- K4 L: z
113. 
     , I0 ?: W& |% t' ]7 a
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     5 ]% |! [! T3 s9 o8 u  K
118. isSearched   =   false;
     
119. }
     ' b# m! `7 H8 _. t; G; n% }
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     ( D* M" Y9 |* \3 F. y! h
123. UPNPNAT_MAPPING   search;
     ; p/ y" t( T" l7 O) A
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     . U" j' ~3 V% `. J# Y3 y" g1 P# c3 m
127. RemoveNATPortMapping(search,   false);
     ; h( y4 ?" C0 g# J. c# R: [
128. }
       t+ n$ m8 r( H
129. 
     ' X) I% ?' l: Q: C9 X1 [
130. m_Mappings.RemoveAll();
     % I0 Z* r9 L6 C2 x
131. }
     
132. 
     0 F1 e: j; a: Y5 y) D  n
133. 
     ) B$ r8 ^1 G. a
134. bool   MyUPnP::InternalSearch(int   version)
     9 G; Q. s; ?6 Q1 ?: ^& Q
135. {
     * M1 U  c- F; [; f
136. if(version <=0)version   =   1;
     " w, O: j+ y# z5 @2 p9 S
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     4 I6 G! d! u6 Y7 P1 a
144. };
     ' ?) e2 O0 Q( f( C
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     ' `( H1 r8 H- k8 E; z/ v
148. ioctlsocket(s,   FIONBIO,   &lv);
     1 \$ d/ c! C" L' {: |5 j: F
149. 
     / y5 Q2 _# J  Z  g
150. int   rlen   =   0;
     5 K' L& y8 {8 M
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     / a; x9 |0 p" ~, B* _. {
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);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     , o+ l  d& a5 L& G2 J2 Y
159. }
     ; M) n& B2 O+ Z; D, S6 M; s' r, h# p
160. }
     
161. 
     
162. Sleep(10);
     # K8 _# ~2 I# k: S, Q$ }2 |; f
163. 
     - O( I2 n8 S) H" e
164. char   buffer[10240];
     : {+ L2 c: P) n. @
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     0 H5 b! o5 d7 B" W+ }
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     & u9 j5 R% z' b& A# o) R" u( {
168. 
     6 Q! ?# p* C( I6 n  ^2 z1 B
169. CString   response   =   CString(CStringA(buffer,   rlen));
     , A  h" E, |. `3 c& c+ V7 n, p
170. CString   result;
     , ?/ R8 \: u( u
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
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)   {
     5 z1 E- T, J8 B& R. k
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     * S  O2 X' N4 s! c# C
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     % R6 k* B5 \7 U5 F" ]  N2 Y
184. m_description.Trim();
     
185. return   GetDescription();
     / T' h* R  [! u
186. }
     
187. }
       b% _! I' H9 ~7 R
188. }
     
189. }
     # D5 {3 w7 A$ `9 x2 e4 ^
190. }
     
191. closesocket(s);
     $ C  \8 Z& _1 Z9 R! w1 z
192. 
     $ Q3 n5 P9 u3 i/ W3 B: T
193. return   false;
     
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
$ L  J( F$ N- a# P9 Q) J- Q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
0 |( S' g8 G; F2 Q0 P7 E
- I1 }1 j# y" z
) m* k5 D) P8 B& L) v#pragma once
' E0 e) Y  s# B% K9 n$ V9 n#include <exception>
: J: `; Q5 O$ G! @3 z( q- Y' z
$ {# _$ v0 s/ C: A0 b
  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
  u. a; w0 M) I};


enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
" W3 u& y! z# n/ ], z        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};

- i5 e: H+ i. t9 v& P- ~
- X' l) T, u" s- ^1 G

+ n. v4 s# L' S- X  c7 Y1 gclass CUPnPImpl
{
public:
: {6 {3 }, T. f) P! T6 q        CUPnPImpl();
( C/ s8 N/ q1 ]! D! M3 O        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
  u' L6 N- E) L) l; [1 ?+ c                UPNP_OK,
( R, v* n; C, J                UPNP_FAILED,
                UPNP_TIMEOUT
        };
. \2 M/ ?  W0 p; o# o5 u" B5 z+ S

        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
% {9 H+ d: O- i        virtual bool        CheckAndRefresh() = 0;
' s9 E4 i- T+ G" d# V/ |- m        virtual void        StopAsyncFind() = 0;
) d. F0 e% |) e, K$ f4 p        virtual void        DeletePorts() = 0;
* Q7 r! \' ^/ t2 n. m! h: e        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
  h7 D/ P7 X; c$ B: p       
- V4 g- d# y- p8 D1 v        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
, b0 W: o0 d2 H* a# C2 R        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
& F* u1 r6 Y! o. U( n/ E' Z        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       


1 K* v7 J% s+ t/ m- v// Implementation
/ m0 M, ?3 }9 \' m* Y1 Uprotected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

* S/ i# W' A. z( S% [* A/ ]) `# F
private:
5 `7 W& W: E6 w$ H0 W' n  J. n, `0 o        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
: S/ E, I9 x, l6 z

0 R0 n  c- I+ W0 Y4 W};


' c6 F+ l* h6 D8 c  a// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
1 ^! p. F0 O! y' n{
, F! {7 ]7 T* M5 fpublic:
& H* j4 [. J4 D0 K        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
- f$ w' y: X/ t) m2 y* {        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
* d4 X" N: v, E7 M};
" {1 g# d. V3 `8 S( Y6 O
* h% _& j- m  I7 n
' L' T/ G: o1 ?6 K/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
. ?! W2 ?1 W" B8 L4 z# c) v: |

#pragma once
; ^/ `& H% O2 [#pragma warning( disable: 4355 )


#include "UPnPImpl.h"
" n, w( `; M2 @3 z9 Q#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
. Y. T2 Z& _- d# B3 G
% R8 h+ r' {9 n, }) P' Z
#include <vector>
#include <exception>
7 R4 c) u' ~2 V! p% W' T#include <functional>



. T) y! j1 j% u0 L; G
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;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
% f( o* Z( U2 c6 Itypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
* w* U* v4 \1 w0 P! ^typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


6 h+ z$ ~0 m8 @% etypedef DWORD (WINAPI* TGetBestInterface) (
; T3 t8 A2 |! e) L# r1 E4 N3 P  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
);


typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
- c" ~* Q! y$ ?);
5 ?) H6 W. f8 S0 @, F( r6 [# z+ h
# w! L0 {- d& I& J# U/ [1 N% q
6 B5 j1 ^- t" o; ^typedef DWORD (WINAPI* TGetIfEntry) (
- B8 U2 L, T$ x: a  PMIB_IFROW pIfRow
);
& A) B( c$ Y8 @/ @

CString translateUPnPResult(HRESULT hr);
. j' H+ j# T8 v# C! z7 \HRESULT UPnPMessage(HRESULT hr);
; V# ~+ ?; o( B- y3 ?: w
+ E/ w8 D8 B% g' P% M7 H* r7 ]
class CUPnPImplWinServ: public CUPnPImpl
{
& `& a, p7 A. `2 s        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
& q# P5 X5 n9 c' b5 Hpublic:
4 l9 Z0 j) H, E9 J9 F        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();

4 F+ i8 r: g) T. A% D& ^
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
$ T/ s: q. j2 ~. w5 Q        virtual void        DeletePorts();
        virtual bool        IsReady();
/ }( t9 A, D8 D+ Y' r, g) k        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
+ S" x- e5 S4 i( O! J- x        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
, O2 ]+ l6 ^' v0 t: m% x& u1 l0 n
+ h, U: T  X- S& I2 S& e
protected:
& J/ ?& u6 U! P0 `- I& V3 ?6 A1 U        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
0 X6 ]) N5 s: X        bool        OnSearchComplete();
        void        Init();


        inline bool IsAsyncFindRunning()
. y4 f; b3 m* _        {
( @- O% I% |" Z" B+ v' C& x+ K' E& p                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
# u* [. _5 n4 H                        m_bAsyncFindRunning = false;
3 i2 Z( J3 i4 V                }
                MSG msg;
0 n, `. |; }6 ^                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
  I- ~+ Y1 ?$ o7 t, {6 @                {
                        TranslateMessage( &msg );
' I1 a0 S! p& z, W  Y5 r* V+ [- p                        DispatchMessage( &msg );
                }
+ X* J/ N  n  Q! ^: ]" r( |                return m_bAsyncFindRunning;
; c1 y6 K& S/ M5 n/ I4 e        }

9 {; A% w' ]* m! ^
" C: ?' ^+ P, }. Z3 d        TRISTATE                        m_bUPnPDeviceConnected;
7 q; V/ A* m% H) `( n+ p  U
$ l7 o6 ?2 ]/ U+ `4 K
// Implementation
1 h5 Y5 Q7 ]4 H. y5 e9 R        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
! `1 ^. Q" b( O& [* o! }" B3 L2 k        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
0 O9 i$ ?/ N* M        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
% A* K. t  u! x1 y        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
3 J# t2 E4 n# q2 e& W        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;
, I9 f- a$ D2 c! a/ u( h
: P# o* Y. X0 N( X- x+ b
6 _" N3 N# i" {; d. w        static FinderPointer CreateFinderInstance();
/ D5 e" c7 N; F; g( `! S0 A        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
$ g0 v4 v' M" }; M: g( ?8 \                        CComBSTR deviceName;
% c  m3 X3 a" D9 w                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
% p' |2 N- b  X& B% L
( J) j2 z+ F, L6 z- L
5 y8 q: u( q+ x8 E: O* c2 T                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
: S) n6 t" c( ~, j( G9 C; w) B& e) {
" K/ A; c/ c, H# I
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
( U$ L0 e! M- q/ _( n6 j& f                }
                CComBSTR m_udn;
        };
. y2 ?" r3 d/ A2 J, e. \       
5 C  B4 @  r3 o1 m8 R0 Y! Q        void        ProcessAsyncFind(CComBSTR bsSearchType);
0 s. |$ R, c/ C# ?! _        HRESULT        GetDeviceServices(DevicePointer pDevice);
3 a: {% m5 s. W  Q7 D; f% w& ^        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
! K: q) S! p0 ^% r  |; J0 b& y        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
1 q$ s" _  {1 E2 z  F) L                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
# F  _* I+ |( Y1 b0 A
, J2 O1 s! i; ~$ J. z$ l' C! ?
2 l) |/ w/ b$ [        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
7 |- F& H5 i+ \" [5 J        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
! I6 H7 `0 P# e        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
: Z7 b4 p% G' H4 X8 ^. R        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);

- E) ^# K% U. ?" D" X. i
- q7 D% Q2 C7 V) [// Private members
+ H2 V( N* G) Zprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
' ~& B( H8 u9 ^; M5 @* c+ ]
' L: R+ F! ^. ?" ], _" {
        LONG        m_nAsyncFindHandle;
( ^* e4 j* ]! a* X9 z, ~        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?
- G3 B6 c. B& K0 ~0 a" u: C( d0 O- s        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


};
3 A) C8 ], i7 g; W- L
8 _6 [2 A1 i9 @  q1 y; \+ j: `
// DeviceFinder Callback
class CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
{
public:
& C+ n( D! A# n  e1 {1 L        CDeviceFinderCallback(CUPnPImplWinServ& instance)
) l# O& l1 \" M, L                : m_instance( instance )
2 L, B" |% a3 p2 m        { m_lRefCount = 0; }
6 [, y# X% c7 K: M1 g" _# V
* B- p" t: `+ R* j1 F1 R
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
* _. n% p% V$ m/ i/ J* a2 X9 O   STDMETHODIMP_(ULONG) Release();


" {; w1 c" B9 K( q" E! z// implementation
# Q. U" Y! g" r5 p3 Yprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
' i* q: Y" X# ~        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
6 B4 z; d6 y% r8 [% |1 {, B        HRESULT __stdcall SearchComplete(LONG nFindData);
! O3 s" @( h1 S

private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


// Service Callback
3 L% r9 z$ M( G4 k7 I3 B5 Vclass CServiceCallback
        : public IUPnPServiceCallback
7 E- v1 U8 a# j8 a2 o2 z{
public:
: X, m* B- x$ P1 l, B        CServiceCallback(CUPnPImplWinServ& instance)
. u1 p0 K' _+ c8 p                : m_instance( instance )
6 R# `( o6 d" y" |/ l0 c        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
; `$ u. y8 q/ A1 o% g  X

// implementation
private:
. l5 J2 z) n3 i6 i        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
2 ]- ?1 d5 w, ^) t; C        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

" [) Y" s/ x- Q% I7 _8 d
private:
        CUPnPImplWinServ& m_instance;
6 |7 i1 E5 t- b) H6 J' B        LONG m_lRefCount;
};
) G8 n6 Z8 _  o7 g, ^( b
3 ?$ A/ i' w1 [3 |  `& o( a4 r& \" ?
9 [8 w1 n9 }4 D( U! j/////////////////////////////////////////////////
: a! l7 m- I: [7 e
* F) X) T2 T! M8 l/ l  ?* _5 u& T9 b
$ |4 A, L3 l7 o! t+ S! w5 q使用时只需要使用抽象类的接口。
! [3 n% \4 e+ A+ t' J; vCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.